.bashrc_lxplus

#!/usr/bin/env bash

#--------------------------------------------
set_vars()
{
    export SFTDIR=/afs/cern.ch/work/a/acampove/Packages/RK
    export PATH+=:$HOME/.local/bin
    export STKDIR=/afs/cern.ch/work/a/acampove/stack
    export PROMPT_COMMAND=''
    export JOBDIR=/afs/cern.ch/user/a/acampove/.grid_jobs/
    export EOSHOME=/eos/user/a/acampove/
    export EDITOR=vim
    export TERM=xterm-256color
    export DAVINCI=/afs/cern.ch/work/a/acampove/RK/DaVinci/
    export SAMFIND=/afs/cern.ch/work/a/acampove/RK/samplefinder/py/
    export PRIDIR=/afs/cern.ch/work/a/acampove/private
    export CERNBOX="/eos/user/a/acampove/"
    export INSTALL_DIR="/afs/cern.ch/work/a/acampove/Packages/"
    export KRB5_CONFIG="/etc/krb5.conf:/afs/cern.ch/user/a/acampove/.kerberos/krb5.conf"
    export KRBDIR="/tmp/"
    export CONDIR=/afs/cern.ch/work/a/acampove/condir
    export SKDR=/afs/cern.ch/work/a/acampove/stack
    export EFOP=$SKDR/DaVinci/HltEfficiencyChecker/options
    export EFSC=$SKDR/DaVinci/HltEfficiencyChecker/scripts
    export RXDR=/afs/cern.ch/work/a/acampove/Packages/RK/rx_hlt_mva
}
#--------------------------------------------
set_alias()
{
    alias vim='/afs/cern.ch/work/a/acampove/Packages/nvim/bin/nvim'
    alias setupViews="source /cvmfs/sft.cern.ch/lcg/views/setupViews.sh"
    alias getsize='du -h --max-depth=0'
    alias work='cd /afs/cern.ch/work/a/acampove'
    alias mmd='conda deactivate'
    alias mml='conda env list'
    alias mmr='conda env remove -n '
    alias ls='ls --color=auto'
    alias sdiff='sdiff -bBWs -w 250'
    alias tmuxn="tmux new -s"
    alias tmuxa="tmux attach -t"
    alias tmuxl="tmux list-sessions"
    alias condormont='watch condor_q'
    alias tmux_check='systemctl --user status tmux.service'
}
#--------------------------------------------
mma()
{
    VENV=$1
    conda activate $VENV
    if [[ "$VENV" == "rk" ]];then
        eval "$(fzf --bash)"
    fi
}
#----------------#ZENITY NOTIFICATIONS--------------
disk_usage()
{
    PATTERN=$1
    du -s *$PATTERN* | awk -F " " '{printf("%s %s\n",$1,$2)}'
}
#---------------------LHCb specific----------------
setLbEnv()
{
    SCRIPT=/cvmfs/lhcb.cern.ch/lib/LbEnv

    if [ -f $SCRIPT ]; then
        . $SCRIPT 
    fi  
}
#---------------------------------------------------
mountCERNBOX()
{
    SRC_DIR=/eos/home-a/acampove/
    TGT_DIR=

    if [[ ! -d $MNT_DIR ]];then
        echo "Cannot find $MNT_DIR"
        kill -INT $$
    fi

    sshfs -o idmap=user acampove@lxplus.cern.ch:$MNT_DIR $MNT_DIR
}
#---------------------------------------------------
setView()
{
    if [[ ! -z $1 ]] && [[ ! -z $2 ]];then
        LCG_REL=$1
        ARCH=$2
    fi

    setupViews $LCG_REL $ARCH 

    echo "Set:"
    echo "LCG:$LCG_REL"
    echo "ARCH:$ARCH"
}
#--------------------------------------------------
listBKFiles()
{
    BKPATH=$1

    if [[ -z $BKPATH ]];then
        echo "Path needed"
        kill -INT $$
    fi

    (
    setLbEnv
    lb-run LHCbDirac/prod bash -c "dirac-bookkeeping-get-files -B \"$BKPATH\"" 
    )
}
#---------------------------------------------------
mountPFS()
{
    if [[ ! -z $1 ]];then
        MACHINE=$1
    else
        MACHINE=701
    fi  

    TM=10
    SRC_DIR=/publicfs
    TRG_DIR=/afs/cern.ch/work/a/acampove/pfs_ucas

    if [[ ! -d $TRG_DIR ]];then
        echo "Cannot find $TRG_DIR"
        kill -INT $$
    fi
    
    echo "Caching every $TM seconds in $MACHINE"
    sshfs  -o cache_timeout=115200 -o attr_timeout=115200 -o cache_stat_timeout=$TM -o cache_dir_timeout=$TM -o cache_link_timeout=$TM -o auto_cache -o idmap=user  campoverde@lxslc$MACHINE.ihep.ac.cn:$SRC_DIR $TRG_DIR
}
#------------------------------
unmountPFS()
{
    MNT_DIR=/afs/cern.ch/work/a/acampove/pfs_ucas
    fusermount -u $MNT_DIR
}
#------------------------------
getLFNFiles()
{
    LIST=$1
    if [[ -z $LIST ]];then
        echo "List of LFNs needed"
        kill -INT $$
    fi

    if [[ ! -f $LIST ]];then
        echo "File list $LIST not found"
        kill -INT $$
    fi

    (
    setLbEnv
    while read FILE;do
        echo "$FILE"
        lb-run LHCbDirac/prod bash -c "dirac-dms-get-file \"$FILE\"" 
    done < $LIST
    )
}
#------------------GHOSTSCRIPT FUNCTIONS------------
gs_merge()
{
    if [[ -z $1 ]] || [[ -z $2 ]];then
        echo "Usage:"
        echo "   gs_merge output.pdf *.pdf"
        kill -INT $$
    fi

    gs -dBATCH -dNOPAUSE -q -sDEVICE=pdfwrite -sOutputFile="$1" $2 
}
#--------------------------------------------------
conda_init()
{
    __conda_setup="$('/afs/cern.ch/work/a/acampove/miniforge3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
    if [ $? -eq 0 ]; then
        eval "$__conda_setup"
    else
        if [ -f "/afs/cern.ch/work/a/acampove/miniforge3/etc/profile.d/conda.sh" ]; then
            . "/afs/cern.ch/work/a/acampove/miniforge3/etc/profile.d/conda.sh"
        else
            export PATH="/afs/cern.ch/work/a/acampove/miniforge3/bin:$PATH"
        fi
    fi
    unset __conda_setup

    if [ -f "/afs/cern.ch/work/a/acampove/miniforge3/etc/profile.d/mamba.sh" ]; then
        . "/afs/cern.ch/work/a/acampove/miniforge3/etc/profile.d/mamba.sh"
    fi
}
#--------------------------------
git_rc()
{
    FILES="$1"
    for FILE in ${FILES[@]};do
        git add -A $FILE
    done

    git commit -m "$2"
}   

git_ac()
{
    FILES="$1" 
    for FILE in ${FILES[@]};do
        if [[ ! -e $FILE ]];then
            echo "File/Directory $FILE not found"
            kill -INT $$
        else
            echo "Adding $FILE"
        fi
        git add $FILE
    done
    
    git commit -m "$2"
}

git_acp()
{      
    git_ac "$1" "$2"
    git push origin master
}

git_pull()
{
    BRANCH=$(git branch | grep "*" | awk -F " " '{print$2}')

    git pull origin $BRANCH
}
#------------------#INSTALLED SOFTWARE--------------
# Source global definitions
if [ -f /etc/bashrc ]; then
	. /etc/bashrc
fi

shopt -s direxpand
systemctl --user enable --now tmux.service
set_vars
set_alias
conda_init