#!/bin/bash

MRY_PROFILE="/etc/profile.d/mry_profile.sh"
MRY_DIR=(
    "/mry"
    "/mry/bin"
    "/mry/lib"
    "/mry/sh"
    "/mry/etc"
    "/mry/resource"
    "/mry/log"
)
SH_FILE=(
    "ca.sh"
    "common.sh"
    "install.sh"
    "k8s.sh"
    "python_debug.sh"
)


function mry_profile() {
    cat <<- 'EOF' > $MRY_PROFILE
alias h="history"
alias md="mkdir"
alias rd="rmdir"
alias vi="vim"
alias u="whoami"
alias g="groups"
alias lsp="lsof -p"
alias lsi="lsof -i"
alias ports="netstat -anp"
alias psef="ps -ef"
alias psg="ps -ef | grep"
alias cpr="cp -r"
alias rmrf="rm -rf"
alias rmf="rm -f"
alias cl="clear"
alias sctl="systemctl"

HISTFILE=~/.bash_history-$$
HISTSIZE=1000
TMOUT=36000

export HISTFILE HISTSIZE TMOUT PATH="$PATH:/mry/bin:/mry/sh"
EOF
    chmod +rw $MRY_PROFILE
    source $MRY_PROFILE
}

function set_ps1() {
    echo 'export PS1='\''[\u@\h \W \[\033[01;$(($?==0?32:31))m\]$([[ $? == 0 ]] && echo "o" || echo "x")\[\033[00m\] ] '\'' '  >> $MRY_PROFILE
    source $MRY_PROFILE
}

function md_mry() {
    for dir in "${MRY_DIR[@]}"
    do
        mkdir -p $dir
    done
}

function install_env() {
    mry_profile
    set_ps1
}

function pull_opkit_shell() {
    local opkit_url="https://raw.githubusercontent.com/Oconnt/opkit/master/sh"
    for f in "${SH_FILE[@]}"
    do
        curl --connect-timeout 10 -o /mry/sh/$f ${opkit_url}/$f > /dev/null 2>&1
        if [ $? -eq 0 ];then
            echo "${f} pull done."
        else
            echo "${f} pull fail!"
        fi
    done
}

function main() {
    md_mry
    install_env
    pull_opkit_shell
}

main