path: root/bash_include

#!/bin/bash
#
# <lantw44>
#
# -- UTF-8 --
#

if tty -s; then interactive_shell=1; fi

[ "$interactive_shell" ] && echo "Running .bash_include"

# Internal Variables

default_tty_setting=`stty -g`

colorprompting='\[\e[1;31m\]\!\[\e[m\] [\[\e[1;33m\]\u\[\e[m\]@\[\e[1;32m\]\h\[\e[m\] \[\e[1;36m\]\w\[\e[m\]]\$ '
nocolorprompting='\! [\u@\h \w]\$ '

historycountfile="$HOME/.bash_history.count"
historybackupfile="$HOME/.bash_history.bak"
bgrunfiledir="$HOME/tmp/bgrun-`whoami`"
trashdir="$HOME/trash"

HISTSIZE=1000000
HISTFILESIZE=1000000
HISTCONTROL=ignoredups:ignorespace

REALPATH_PROGRAM="realpath"




# Environment Variables

export EDITOR=vim
export FCEDIT=vim
export PAGER=less




# Aliases

alias startcolor='PS1=$colorprompting'
alias stopcolor='PS1=$nocolorprompting'

alias ll='ls -l'
alias grep='grep --color=always'
alias rm='rm -i'
alias cp='cp -pi'
alias mv='mv -i'
alias jobs='jobs -l'
alias less='less -RS'

alias cccc='LANG=C;LC_ALL=C'
alias enus='LANG=en_US.UTF-8;LC_ALL=en_US.UTF-8'
alias big5='LANG=zh_TW.Big5;LC_ALL=zh_TW.Big5'
alias zhtw='LANG=zh_TW.UTF-8;LC_ALL=zh_TW.UTF-8'
alias utf8='LANG=zh_TW.UTF-8;LC_ALL=zh_TW.UTF-8'

alias savetty='default_tty_setting=`stty -g`'
alias resetty='stty $default_tty_setting'




# Functions

function compile_all ()
{
    local noask=0
    [ "$1" = '' ] && echo "Which file(s) do you want to compile? " && return 1
    [ "$1" = "-n" ] && noask=1
    if [ "$noask" = "0" ]; then
        read -e -p "CFLAGS: " -i "$CFLAGS" NEWCFLAGS
        read -e -p "LDFLAGS: " -i "$LDFLAGS" NEWLDFLAGS
        [ "$NEWCFLAGS" '!=' '' ] && CFLAGS=$NEWCFLAGS
        [ "$NEWLDFLAGS" '!=' '' ] && LDFLAGS=$NEWLDFLAGS
    else
        shift
    fi
    while [ "$1" '!=' '' ]
    do
        local TARGETFILE="`echo "$1" | cut -d . -f 1`"
        local SUFFIX="`echo "$1" | cut -d . -f 2`"
        if [ -f "$1" ]; then
            true
        else
            printf\
            '\e[1;33mWarning\e[0m: Non-existent file or not a regular file\n'
            shift ; continue
        fi
        [ "$TARGETFILE" = "$1" ] && shift && continue
        if [ "$SUFFIX" = "c" ]; then
            echo "[CC] $1 -> $TARGETFILE"
            gcc $CFLAGS "$1" $LDFLAGS -o "$TARGETFILE"
        elif [ "$SUFFIX" = "cpp" ]; then
            echo "[CXX] $1 -> $TARGETFILE"
            g++ $CFLAGS "$1" $LDFLAGS -o "$TARGETFILE"
        else
            printf 'Unknown suffix (\e[1;33mskipped\e[0m)\n'
        fi
        [ "$?" '!=' "0" ] && printf\
            '\e[1;31mError\e[0m while compiling file\n'
        shift
    done
    return 0
}


function convert_to_html ()
{
    while [ "$1" '!=' '' ]
    do
        for i in "$1"
        do 
            vim $i -c 'set background=dark' \
                -c 'highlight PreProc ctermfg=darkcyan' \
                -c "$BEFORE_CONVERT_TO_HTML" \
                -c "$BEFORE_CONVERT_TO_HTML1" \
                -c "$BEFORE_CONVERT_TO_HTML2" \
                -c TOhtml \
                -c :w \
                -c :qa 
        done
        shift
    done
}

function mkscreenacl ()
{
    PERMIT_COMMAND="select windowlist other meta detach reset hardcopy info redisplay lastmsg next prev xon xoff windows suspend help colon copy paste writebuf readbuf displays stuff attach"
    while [ "$1" '!=' '' ]
    do
        for i in $PERMIT_COMMAND
        do
            echo "aclchg $1 +x $i"
        done
        echo "aclchg $1 -rw \"#?\"" 
        shift
    done
}

########## Background ##########

alias bgr=bgrun
alias bgv=bgview
alias bgl=bglist
alias bgc=bgcount
alias bgls=bglist
alias bgrm=bgclean

function bgrun ()
{
    [ "$#" = "0" ] && return 1
    [ '!' -d "$bgrunfiledir" ] && mkdir -p "$bgrunfiledir"
    local current_time=`date "+%Y%m%d-%H%M%S"`
    local cmdname=`echo "$1" | sed -e 's/-/_/g' -e 's/\\//_/g' -e 's/ /_/g'`
    if [ "`echo "$cmdname" | cut -c 1`" == "_" ]
    then
        cmdname=`echo "$cmdname" | cut -c 2-`
    fi
    local filename="$bgrunfiledir/$current_time-$cmdname"
    echo "Writing to $filename"
    echo "$@" > $filename
    "$@" &>> "$filename" &
}

function bglist ()
{
    local viewtime=0
    [ "$1" = "--full" ] && viewtime=1
    {
        for i in `find "$bgrunfiledir" -maxdepth 1 -mindepth 1 | sort`
        do
            [ "$viewtime" == "1" ] && echo "$i"
            head -n 1 "$i"
        done
    } | {
        if [ "$viewtime" == "1" ]
        then
            local readstat=0
            local -i i=1
            while true
            do
                local dateandtime_long
                local cmdline
                read dateandtime_long
                read cmdline
                [ "$?" '!=' "0" ] && break
                local dateandtime=`basename "$dateandtime_long"`
                local part_year=`echo $dateandtime | cut -c 1-4`
                local part_month=`echo $dateandtime | cut -c 5-6`
                local part_date=`echo $dateandtime | cut -c 7-8`
                local part_hour=`echo $dateandtime | cut -c 10-11`
                local part_minute=`echo $dateandtime | cut -c 12-13`
                local part_second=`echo $dateandtime | cut -c 14-15`
                printf '%6d' "$i" 
                echo " $part_year-$part_month-$part_date $part_hour:$part_minute:$part_second $cmdline"
                i=$i+1
            done
        else
            cat -n
        fi
    } | $PAGER
}

function bgview ()
{
    local -i yourchoice
    if [ "$1" = "" ]
    then
        yourchoice=`bgcount`
    else
        if [ "$1" -le "0" ]
        then
            yourchoice=$((`bgcount`+$1))
        else
            yourchoice=$1
        fi
    fi
    echo "Your choice is $yourchoice."
    local realfilename=`find "$bgrunfiledir" -maxdepth 1 -mindepth 1 | sort | sed -n ${yourchoice}p`
    echo "Command Line: `head -n 1 "$realfilename"`"
    read -e -p "View '$realfilename' ? " confirm
    if [ "$confirm" = "n" ] || [ "$confirm" = "N" ]
    then
        return 1
    fi
    {
        printf "Command Line: "
        cat "$realfilename"
    } | $PAGER
}

function bgcount ()
{
    find "$bgrunfiledir" -maxdepth 1 -mindepth 1 | cut -d - -f 2,3 | wc | awk '{print $2}'
}

function bgclean ()
{
    if [ "$1" = "all" ]
    then
        echo "Removing the directory $bgrunfiledir"
        rm -rf "$bgrunfiledir" &
        return 0
    else
        split_arguments "$@"
        local -i i=0
        while [ "${arglist[$i]}" ]
        do
            arglist[$i]="-e ${arglist[$i]}p"
            i=$i+1
        done
        local oneline
        find "$bgrunfiledir" -maxdepth 1 -mindepth 1 | sort | sed -n ${arglist[*]} | {
            while read oneline
            do
                echo "Removing $oneline"
                rm -f "$oneline"
            done
        }
    fi
    unset arglist
    unset prefixlist
}

function bgdu () 
{
    du -a "$bgrunfiledir"
}

########## Background End ##########

function check_dmesg ()
{
    [ "$#" = "0" ] && return 1

    while true
    do
        PREVIOS_DMESG_BUF="$DMESG_BUF"
        DMESG_BUF="`dmesg`"
        [ "$PREVIOS_DMESG_BUF" '!=' "$DMESG_BUF" ] && [ "$FIRST_RUN" = "4" ] && echo '===> system message buffer was modified <==='
        sleep $1
        [ "$?" '!=' "0" ] && return 1
        FIRST_RUN=0
    done 
}

function prehistory_backup ()
{
    echo "Backing up current history file"
    local currentcount=`wc "$HISTFILE" | awk '{print $1}'`
    [ '!' -f "$historycountfile" ] && touch "$historycountfile"
    local -i previoushistorycount
    previoushistorycount=`cat "$historycountfile"`
    if [ "$currentcount" -lt "$previoushistorycount" ]
    then
        printf "\e[1;31mWarning\e[m: Your $HISTFILE may be MODIFIED BY OTHER PROGRAMS!\n"
        printf "Note: \e[1;33m$currentcount\e[m < $previoushistorycount\n"
        echo "Your $historycountfile and $historybackupfile will not be overwritten until this problem is fixed."
        echo " 1. Check your $HISTFILE."
        echo " 2. Edit your $HISTFILE manually if some unexpected changes are found." 
        echo "    (You may need $historybackupfile to do it) "
        echo " 3. Remove the file $historycountfile."
        echo " 4. Run the command \`prehistory_backup' again."
        return 3
    fi
    echo "$currentcount" > "$historycountfile"
    \cp -f "$HISTFILE" "$historybackupfile"
}

########## Trash Manager ##########

alias trash_put=trash_mv


function trash_mv ()
{
    [ "$#" = "0" ] && return 1
    [ '!' -d "$trashdir" ] && mkdir -p "$trashdir"
    local original_path
    local current_time
    local -i i=0
    split_arguments "$@"
    while [ "${arglist[$i]}" ]
    do
        original_path="`$REALPATH_PROGRAM "${arglist[$i]}"`"
        current_time=`date "+%Y%m%d-%H%M%S"`
        better_time=`date "+%Y-%m-%d %H:%M:%S"`
        dirname="`basename "${arglist[$i]}" | sed -e 's/-/_/g' -e 's/ /_/g'`"
        fulldirname="$trashdir/$current_time-$dirname"
        mkdir -p "$fulldirname"
        echo "Move: ${arglist[$i]} -> $fulldirname"
        "${prefixlist[@]}" mv "${arglist[$i]}" "$fulldirname"
        echo "$better_time" > "$fulldirname/information.date"
        echo "$original_path" > "$fulldirname/information.path"
        i=$i+1
        shift
    done
    unset arglist
    unset prefixlist
}

function trash_rm ()
{
    echo bb
}

function trash_ls ()
{
    echo cc
}

function trash_cd ()
{
    echo dd
}

function trash_recover ()
{
    echo yy
}

function trash_count () 
{
    echo zz
}

function trash_du ()
{
    split_arguments "$@"
    local oneline
    find "$trashdir" -maxdepth 1 -mindepth 1 | sort | {
        while read oneline
        do
            echo "'$oneline'"
        done
    } | xargs -n 10 "${prefixlist[@]}" du -s
    unset arglist
    unset prefixlist
}

########## Trash Manager End ##########

########## Help ##########

alias helpf='help_function'
alias helpm='help_myself'

function print_iconv ()
{
    [ "$1" = "$2" ] && cat && return 0
    iconv -f "$1" -t "$2" 
}

function help_function ()
{
    [ "$#" = "0" ] && {
        cat << ENDHELPMSG
 <<< Help >>>
    help_myself [arguments ...]
    help_function [functions ...]
 <<< Group: Background >>>
    bgrun command [arguments ...]
    bglist [--full]
    bgview [number]
    bgclean [all | numbers ...]
    bgcount
    bgdu
 <<< Group: Trash >>>
    trash_mv [filename ...] [-- sudo_prefix ...]
    trash_ls
    trash_cd [number]
    trash_recover [number] [-- sudo_prefix ...]
    trash_rm [all | numbers ...] [-- sudo_prefix ...]
    trash_count
    trash_du [-- sudo_prefix ...]
 <<< Group: PATH Editor >>>
    path_editor
    ldpath_editor
 x  split_path
 x  update_path 
 <<< Other >>>
    compile_all [-n] filename ...
    convert_to_html filename ...
    mkscreenacl username ...
    check_dmesg seconds
    prehistory_backup
    check_important_files
    backup_file filename ... [-- sudo_prefix ]
    keep_sudo_credential [seconds]
 x  split_arguments [arguments ...]
ENDHELPMSG
    } && return 0
    local current_charset=`echo "$LC_ALL" | cut -d . -f 2`
    local -i i
    while [ "$1" ]
    do
        case "$1" in
            "help_myself")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
help_myself
    一個測試命令列的小函式
ENDHELPMSG
                ;;
            "help_function")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
help_function
    顯示 .bash_include 提供的額外函式清單
    註：前方加上「x」符號者表示此為內部使用的函式，不宜直接使用
ENDHELPMSG
                ;;
            "bgrun")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
bgrun command [arguments ...]
    執行指令 command 並將輸出導入檔案
    註：此函式會自動以目前時間和指令名稱為檔案命名
ENDHELPMSG
                ;;
            "bglist")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
bglist [--full]
    列出所有使用 bgrun 執行的指令
    若加上 --full 選項，則可同時察看時間
ENDHELPMSG
                ;;
            "bgview")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
bgview [number]
    顯示以 bgrun 執行指令的輸出，若省略 number，表示是最近一次執行的指令
    若 number >  0，
        表示第 number 個指令 (此數值可由 bglist 函式取得)
    若 number <= 0，
        表示第「指令總數-number」個指令 (指令總數可由 bgcount 函式取得)
ENDHELPMSG
                ;;
            "bgclean")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
bgclean [all | numbers ...]
    bgclean all 可清除所有指令的輸出檔
    bgclean 3 5 7 10 表示清除第 3、5、7、10 個指令輸出檔 (編號可由 bglist 取得)
ENDHELPMSG
                ;;
            "bgcount")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
bgcount
    顯示指令輸出檔總數
ENDHELPMSG
                ;;
            "bgdu")
                cat << ENDHELPMSG | print_iconv "UTF-8" "$current_charset"
bgdu
    顯示每個指令輸出檔的檔案大小 (單位：KB)
ENDHELPMSG
                ;;
            *)
                echo "Help message for $1 is not found"
                ;;
        esac
        shift
    done
}

function help_myself () 
{
    echo "argc = $#"
    echo "argv[0] = $0"
    i=1
    while [ "$1" ]
    do
        echo "argv[$i] = $1"
        i=$(($i+1))
        shift
    done
}

########## Help End ##########

function split_arguments ()
{
    local argcount=$#
    local -i i=0
    local prefix_start=0
    while [ "$1" ]
    do
        if [ "$prefix_start" == "0" ]
        then
            if [ "$1" = "--" ]
            then
                prefix_start=1
                i=0
                shift
                continue
            else
                arglist[$i]="$1"
            fi
        else
            prefixlist[$i]="$1"
        fi
        i=$i+1
        shift
    done
}

function check_important_files ()
{
    IMPORTANT_FILES='.screenrc .vimrc'
    for i in $IMPORTANT_FILES
    do
        [ '!' -f "$i" ] && printf "\e[1;31mWarning\e[m: \e[1;33m$i\e[m does not exist.\n"
    done
}

########## PATH Editor ##########

function split_path () 
{
    local fifoname="$HOME/`uuidgen`"
    mkfifo "$fifoname"
    local -i i=0
    local oneline
    export patharr
    echo "$current_path" | sed -e 's/:/ /g' | xargs -n 1 echo > "$fifoname" &
    exec 3<"$fifoname"
    while read -u 3 patharr[$i]
    do
        i=$i+1
    done
    exec 3<&-
    unlink "$fifoname"
    echo $oneline
}

function update_path ()
{
    current_path=''
    local -i i=0
    local firsttime="yes"
    while [ "${patharr[$i]}" ]
    do
        if [ '!' "${patharr[$i]}" = ":" ]
        then
            if [ "$firsttime" ] 
            then
                firsttime=''
            else
                current_path+=':'
            fi
            current_path+="${patharr[$i]}"
        fi
        i=$i+1
    done
}

function path_editor ()
{
    export current_path="$PATH"
    local should_continue="yes"
    local command
    local command_sub
    local command_sub2
    local -i i
    while [ "$should_continue" ]
    do
        split_path
        i=0
        while [ "${patharr[$i]}" ]
        do
            echo "$i: ${patharr[$i]}"
            i=$i+1
        done
        read -e -p "(A)ppend/(D)elete/(E)dit/(M)ove/(R)eset/(Q)uit ? " command
        case "$command" in
            A|a)
                read -e -p "Type new entry: " patharr[$i]
                update_path
                ;;
            D|d)
                read -e -p "Line: " command_sub
                patharr[$command_sub]=':'
                update_path
                ;;
            E|e)
                read -e -p "Line: " command_sub
                read -e -p "Modify this entry: " -i "${patharr[$command_sub]}" patharr[$command_sub] 
                update_path
                ;;
            M|m)
                read -e -p "Exchange: " command_sub
                read -e -p "With: " command_sub2
                swaptmp="${patharr[$command_sub]}"
                patharr[$command_sub]="${patharr[$command_sub2]}"
                patharr[$command_sub2]="$swaptmp"
                unset swaptmp
                update_path
                ;;
            R|r)
                current_path="$PATH"
                ;;
            Q|q)
                PATH="$current_path"
                echo "PATH=$PATH"
                should_continue=''
                ;;
            *)
                echo " *** Unknown command *** "
                ;;
        esac
    done
    unset patharr
}

function path_editor ()
{
    export current_path="$PATH"
    local should_continue="yes"
    local command
    local command_sub
    local command_sub2
    local -i i
    while [ "$should_continue" ]
    do
        split_path
        i=0
        while [ "${patharr[$i]}" ]
        do
            echo "$i: ${patharr[$i]}"
            i=$i+1
        done
        read -e -p "(A)ppend/(D)elete/(E)dit/(M)ove/(R)eset/(Q)uit ? " command
        case "$command" in
            A|a)
                read -e -p "Type new entry: " patharr[$i]
                update_path
                ;;
            D|d)
                read -e -p "Index: " command_sub
                patharr[$command_sub]=':'
                update_path
                ;;
            E|e)
                read -e -p "Index: " command_sub
                read -e -p "Modify this entry: " -i "${patharr[$command_sub]}" patharr[$command_sub] 
                update_path
                ;;
            M|m)
                read -e -p "From: " command_sub
                read -e -p "To: " command_sub2
                swaptmp="${patharr[$command_sub]}"
                patharr[$command_sub]="${patharr[$command_sub2]}"
                patharr[$command_sub2]="$swaptmp"
                unset swaptmp
                update_path
                ;;
            R|r)
                current_path="$PATH"
                ;;
            Q|q)
                export PATH="$current_path"
                echo "PATH=$PATH"
                should_continue=''
                ;;
            *)
                echo " *** Unknown command *** "
                ;;
        esac
    done
    unset patharr
}

function ldpath_editor ()
{
    export current_path="$LD_LIBRARY_PATH"
    local should_continue="yes"
    local command
    local command_sub
    local command_sub2
    local -i i
    while [ "$should_continue" ]
    do
        split_path
        i=0
        while [ "${patharr[$i]}" ]
        do
            echo "$i: ${patharr[$i]}"
            i=$i+1
        done
        read -e -p "(A)ppend/(D)elete/(E)dit/(M)ove/(R)eset/(Q)uit ? " command
        case "$command" in
            A|a)
                read -e -p "Type new entry: " patharr[$i]
                update_path
                ;;
            D|d)
                read -e -p "Index: " command_sub
                patharr[$command_sub]=':'
                update_path
                ;;
            E|e)
                read -e -p "Index: " command_sub
                read -e -p "Modify this entry: " -i "${patharr[$command_sub]}" patharr[$command_sub] 
                update_path
                ;;
            M|m)
                read -e -p "From: " command_sub
                read -e -p "To: " command_sub2
                swaptmp="${patharr[$command_sub]}"
                patharr[$command_sub]="${patharr[$command_sub2]}"
                patharr[$command_sub2]="$swaptmp"
                unset swaptmp
                update_path
                ;;
            R|r)
                current_path="$LD_LIBRARY_PATH"
                ;;
            Q|q)
                export LD_LIBRARY_PATH="$current_path"
                echo "LD_LIBRARY_PATH=$LD_LIBRARY_PATH"
                should_continue=''
                ;;
            *)
                echo " *** Unknown command *** "
                ;;
        esac
    done
    unset patharr
}

########## PATH Editor End ##########

function backup_file ()
{
    split_arguments "$@"
    local current_time=`date +%Y%m%d`
    local rootfilename
    local -i i=0
    local -i j
    while [ "${arglist[$i]}" ]
    do
        if [ '!' -f "${arglist[$i]}" ] 
        then 
            printf "\e[1;31mError\e[m: ${arglist[$i]} does not exist or it is not a regular file.\n"
            i=$i+1
            continue
        fi
        rootfilename="${arglist[$i]}.$current_time"
        if [ -e "$rootfilename" ]
        then
            j=0
            while [ "$j" -lt "10" ]
            do
                if [ -e "$rootfilename.$j" ]
                then
                    j=$j+1
                    continue
                else
                    "${prefixlist[@]}" \cp "${arglist[$i]}" "$rootfilename.$j"
                    break
                fi
            done
            if [ '!' "$j" -lt "10" ]
            then
                printf "\e[1;31mError\e[m: Can not create a backup file for ${arglist[$i]}.\n"
                printf "\e[1;33mPlease delete some backup file because I only use 0 - 9.\e[m\n"
            fi
        else
            "${prefixlist[@]}" \cp "${arglist[$i]}" "$rootfilename"
        fi
        i=$i+1
    done
    unset arglist
    unset prefixlist
}

function keep_sudo_credential ()
{
    if [ "$1" ]
    then
        update_sudo_interval="$1"
    else
        update_sudo_interval="60"
    fi
    while true
    do
        sudo -v
        sleep "$update_sudo_interval"
    done
}

# Doing something

if [ "$interactive_shell" ] 
then
    echo "Running interactive shell configuration"
    check_important_files
    startcolor
    prehistory_backup
    bind '"\e[A":history-search-backward'
    bind '"\e[B":history-search-forward'
fi

[ "`uname`" = "Linux" ] && REALPATH_PROGRAM="readlink -f"

umask 0022

[ "$interactive_shell" ] && echo "Setting shell options"

shopt -s histappend
shopt -s checkwinsize
shopt -s checkjobs
shopt -s checkhash
shopt -s cmdhist
shopt -s mailwarn

[ "$interactive_shell" ] && echo "Done"