pythonify man script
This commit is contained in:
parent
ddb7a69f7e
commit
56d1705abd
GPG key ID:
01E71F18AA4398E5
4 changed files with 129 additions and 219 deletions
2
.gitignore
vendored
Normal file → Executable file
2
.gitignore
vendored
Normal file → Executable file
|
|
@ -1,3 +1,5 @@
|
||||||
*.tar
|
*.tar
|
||||||
*.tar.gz
|
*.tar.gz
|
||||||
.idea
|
.idea
|
||||||
|
.vscode
|
||||||
|
__pycharm__
|
||||||
219
man
219
man
|
|
@ -1,220 +1,3 @@
|
||||||
#!/bin/sh
|
#!/bin/sh
|
||||||
|
|
||||||
PROJECT_NAME=${PROJECT_NAME:-neo}
|
python utils/manager/main.py $@
|
||||||
LIST=docker-compose.*.yml
|
|
||||||
SAVEFILE=$HOME/.local/faulty/$PROJECT_NAME.save
|
|
||||||
PROFILES=""
|
|
||||||
|
|
||||||
alias dc="docker-compose -p $PROJECT_NAME"
|
|
||||||
# alias dc="echo docker-compose -p $PROJECT_NAME"
|
|
||||||
|
|
||||||
alias _p="printf \"%s\""
|
|
||||||
default() {
|
|
||||||
fopts=""
|
|
||||||
for f in $LIST; do
|
|
||||||
fopts="$fopts -f $f"
|
|
||||||
done
|
|
||||||
profopts=""
|
|
||||||
for p in $PROFILES; do
|
|
||||||
profopts="$profopts --profile $p"
|
|
||||||
done
|
|
||||||
dc $fopts $profopts $*
|
|
||||||
}
|
|
||||||
|
|
||||||
safe_env_load() {
|
|
||||||
# $1 = file
|
|
||||||
# $2 = var
|
|
||||||
# get env var from regex instead of inclusion
|
|
||||||
[ ! -f $1 ] && fatal "file doesn't exist"
|
|
||||||
[ ! -r $1 ] && fatal "file not readable"
|
|
||||||
res=$(grep -E "^$2=" $1 | sed -E "s/^$2=\"(.*)\"/\1/")
|
|
||||||
[ -z "$res" ] && fatal "var not found"
|
|
||||||
_p $res
|
|
||||||
}
|
|
||||||
|
|
||||||
write_savefile() {
|
|
||||||
[ -d $(dirname $1) ] || mkdir -p $(dirname $1})
|
|
||||||
echo -n "LIST=\"" > $1
|
|
||||||
for li in $LIST; do
|
|
||||||
echo -n " $li" >> $1
|
|
||||||
done
|
|
||||||
echo "\"" >> $1
|
|
||||||
echo -n "PROFILES=\"" >> $1
|
|
||||||
for p in $PROFILES; do
|
|
||||||
echo -n " $p" >> $1
|
|
||||||
done
|
|
||||||
echo "\"" >> $1
|
|
||||||
}
|
|
||||||
|
|
||||||
load_savefile() {
|
|
||||||
[ "$1" == "/dev/null" ] && return
|
|
||||||
[ ! -f $1 ] && return
|
|
||||||
LIST=$(safe_env_load $1 LIST)
|
|
||||||
PROFILES=$(save_env_load $1 PROFILES)
|
|
||||||
}
|
|
||||||
|
|
||||||
fatal() {
|
|
||||||
echo "FATAL: $@"
|
|
||||||
exit 1
|
|
||||||
}
|
|
||||||
|
|
||||||
includechain=0
|
|
||||||
handleFallthrough() {
|
|
||||||
cmd=${1:-}; shift
|
|
||||||
case $cmd in
|
|
||||||
delete-project-files|dpf)
|
|
||||||
[ -n "$1" ] || fatal "no directory specified"
|
|
||||||
[ -d /srv/$1 ] || fatal "directory doesn't exist"
|
|
||||||
ls /srv/$1
|
|
||||||
printf "Are you sure? (y/n) [n]: "
|
|
||||||
read answer
|
|
||||||
[ "$answer" = "y" ] && sudo rm -rf /srv/$1 || fatal "cancelled"
|
|
||||||
;;
|
|
||||||
|
|
||||||
-p) # add profile
|
|
||||||
PROFILES="$PROFILES $1"; shift
|
|
||||||
handleFallthrough $@
|
|
||||||
;;
|
|
||||||
|
|
||||||
-i) #include
|
|
||||||
if [ $includechain -eq 0 ]; then
|
|
||||||
LIST="docker-compose.$1.yml"
|
|
||||||
includechain=1
|
|
||||||
else
|
|
||||||
LIST="$LIST docker-compose.$1.yml"
|
|
||||||
fi
|
|
||||||
shift
|
|
||||||
handleFallthrough $@
|
|
||||||
;;
|
|
||||||
|
|
||||||
-x) #exclude
|
|
||||||
LIST=$(echo $LIST | sed "s/docker-compose.$1.yml/ /")
|
|
||||||
shift
|
|
||||||
handleFallthrough $@
|
|
||||||
;;
|
|
||||||
|
|
||||||
make:*)
|
|
||||||
WHAT=`echo $cmd | cut -c6-`
|
|
||||||
case $WHAT in
|
|
||||||
dc|docker-compose)
|
|
||||||
[ -n "$1" ] || fatal "no name specified"
|
|
||||||
FILE="docker-compose.$1.yml"
|
|
||||||
[ -f $FILE ] && fatal "file exists"
|
|
||||||
echo -e "version: '2.2'\n" > $FILE
|
|
||||||
echo -e "services: \n" >> $FILE
|
|
||||||
;;
|
|
||||||
backup)
|
|
||||||
[ -n "$1" ] || fatal "no name specified"
|
|
||||||
FILE="$1.tar.gz"
|
|
||||||
[ -f $FILE ] && fatal "file exists"
|
|
||||||
tar cvf $FILE /srv .
|
|
||||||
;;
|
|
||||||
*) fatal "not supported" ;;
|
|
||||||
esac
|
|
||||||
;;
|
|
||||||
|
|
||||||
setup:*|s:*)
|
|
||||||
WHAT=`echo $cmd | cut -c8-`
|
|
||||||
case $WHAT in
|
|
||||||
ports) sudo ./utils/setup-ports.sh up ./utils/rulelist.rules ;;
|
|
||||||
def) sudo ./utils/setup.sh ;;
|
|
||||||
all|a)
|
|
||||||
sudo ./utils/setup.sh
|
|
||||||
sudo ./utils/setup-ports.sh up ./utils/rulelist.rules
|
|
||||||
;;
|
|
||||||
*) fatal "not supported" ;;
|
|
||||||
esac
|
|
||||||
;;
|
|
||||||
|
|
||||||
--save) # save behaviour
|
|
||||||
handleFallthrough $@
|
|
||||||
write_savefile $SAVEFILE
|
|
||||||
;;
|
|
||||||
|
|
||||||
# preference
|
|
||||||
logs) default logs --tail=20 -f $@ ;;
|
|
||||||
up) default up -d $@ ;;
|
|
||||||
upr) default up -d --remove-orphans $@ ;;
|
|
||||||
|
|
||||||
--help|-h)
|
|
||||||
__() {
|
|
||||||
colsep="$1"
|
|
||||||
cmd="$2"
|
|
||||||
desc="$3"
|
|
||||||
aliases="$4"
|
|
||||||
printf "%s %-20s %s %-64s %s %-40s %s \n" "$colsep" "$cmd" "$colsep" "$desc" "$colsep" "$aliases" "$colsep"
|
|
||||||
}
|
|
||||||
_hr() {
|
|
||||||
repchar() {
|
|
||||||
printf "%$1s" | tr " " "$2"
|
|
||||||
}
|
|
||||||
__ "*" ${1:-$(repchar 20 -)} ${2:-$(repchar 64 -)} ${3:-$(repchar 40 -)}
|
|
||||||
}
|
|
||||||
_r() {
|
|
||||||
__ "|" "$1" "$2" "$3"
|
|
||||||
}
|
|
||||||
_hr
|
|
||||||
_r $0 Description Aliases
|
|
||||||
_hr
|
|
||||||
_hr "Flags" " " " "
|
|
||||||
_hr
|
|
||||||
_r "-x" "Excludes a docker-compose file from the list." " "
|
|
||||||
_r "-i" "Includes a docker-compose file from the list." " "
|
|
||||||
_hr
|
|
||||||
_hr "Commands" " " " "
|
|
||||||
_hr
|
|
||||||
_r "dpf" "Delete project files." "delete-project-files <name>"
|
|
||||||
_r "setup:ports" "Setup ports." "setup:ports"
|
|
||||||
_r "setup:def" "Setup default." "setup:def"
|
|
||||||
_r "setup:all" "Setup all." "setup:all"
|
|
||||||
_r "make:dc <name>" "Make docker-compose file." "make:docker-compose <name>"
|
|
||||||
_r "make:backup <name>" "Make backup file." "tar cvf <name>.tar.gz /srv"
|
|
||||||
_hr
|
|
||||||
_hr "Commands/aliases" "(shortcuts)" " "
|
|
||||||
_hr
|
|
||||||
_r "up" "Bring up services. (-d)" "$0 default up -d"
|
|
||||||
_r "upr" "Bring up services and remove orphans." "$0 default up -d --remove-orphans"
|
|
||||||
_r "logs" "View logs and follow with a tail of 20." "$0 default logs --tail=20 -f"
|
|
||||||
_hr
|
|
||||||
_r "default" "fallback to the main docker-compose command with fileopts." "docker-compose"
|
|
||||||
_hr
|
|
||||||
;;
|
|
||||||
|
|
||||||
debug)
|
|
||||||
load_savefile $SAVEFILE
|
|
||||||
_() {
|
|
||||||
eval "echo -n \"$1=\"; echo \$$1"
|
|
||||||
}
|
|
||||||
_ SAVEFILE
|
|
||||||
_ LIST
|
|
||||||
for i in $LIST; do
|
|
||||||
echo $i
|
|
||||||
done
|
|
||||||
return
|
|
||||||
;;
|
|
||||||
default) default $@ ;;
|
|
||||||
*|'') default $cmd $@ ;;
|
|
||||||
esac
|
|
||||||
}
|
|
||||||
|
|
||||||
main() {
|
|
||||||
case $1 in
|
|
||||||
--ignore-save)
|
|
||||||
SAVEFILE=/dev/null
|
|
||||||
shift 1
|
|
||||||
;;
|
|
||||||
--set-save)
|
|
||||||
SAVEFILE=$2
|
|
||||||
shift 2
|
|
||||||
main $@; exit 0
|
|
||||||
;;
|
|
||||||
--del-save)
|
|
||||||
rm $SAVEFILE
|
|
||||||
shift
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
load_savefile $SAVEFILE
|
|
||||||
handleFallthrough $@
|
|
||||||
}
|
|
||||||
|
|
||||||
main $@
|
|
||||||
97
utils/manager/main.py
Normal file
97
utils/manager/main.py
Normal file
|
|
@ -0,0 +1,97 @@
|
||||||
|
import glob
|
||||||
|
import os
|
||||||
|
import argparse
|
||||||
|
from savefile import SaveFile
|
||||||
|
|
||||||
|
# Spaghetti code to replace the shell script
|
||||||
|
|
||||||
|
PROJECT_NAME = os.environ.get("PROJECT_NAME", "neo")
|
||||||
|
|
||||||
|
files = glob.glob("docker-compose.*.yml")
|
||||||
|
|
||||||
|
def setup_handler(what: str):
|
||||||
|
types = {
|
||||||
|
"def": lambda: os.system("sh ./utils/setup.sh"),
|
||||||
|
"ports": lambda: os.system("sh ./utils/setup-ports.sh ./utils/rulelist.rules")
|
||||||
|
}.get(what, lambda: print("Invalid setup type"))
|
||||||
|
|
||||||
|
|
||||||
|
def dc(cmd: str):
|
||||||
|
fopts = ""
|
||||||
|
for f in files:
|
||||||
|
fopts += f"-f {f} "
|
||||||
|
# print(f"docker-compose {fopts}{cmd}")
|
||||||
|
os.system(f"docker-compose -p {PROJECT_NAME} {fopts} {cmd}")
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
global files
|
||||||
|
ap = argparse.ArgumentParser()
|
||||||
|
|
||||||
|
ap.add_argument("--savefile", help="The savefile to use",
|
||||||
|
default=os.path.expanduser("~/.local/faulty/neo.save"))
|
||||||
|
ap.add_argument("--del-savefile", help="Delete the savefile",
|
||||||
|
action="store_true")
|
||||||
|
ap.add_argument("--ignore-savefile", help="Ignore the savefile",
|
||||||
|
action="store_true")
|
||||||
|
|
||||||
|
# -x, --exclude: e.g. -x this -x that
|
||||||
|
ap.add_argument("-x", "--exclude", help="Exclude a file from the list",
|
||||||
|
action="append", default=[])
|
||||||
|
# -i, --include: e.g. -i this -i that
|
||||||
|
ap.add_argument("-i", "--include", help="Include a file from the list",
|
||||||
|
action="append", default=[])
|
||||||
|
|
||||||
|
ap.add_argument('--save', help="Save the list of files to the savefile", action="store_true")
|
||||||
|
|
||||||
|
ap.add_argument('action', help="The action to perform on the files",
|
||||||
|
nargs="?", default="up")
|
||||||
|
ap.add_argument('args', help="Extra arguments for the action",
|
||||||
|
nargs=argparse.REMAINDER)
|
||||||
|
|
||||||
|
args = ap.parse_args()
|
||||||
|
|
||||||
|
if args.del_savefile:
|
||||||
|
os.remove(args.savefile)
|
||||||
|
return
|
||||||
|
|
||||||
|
if not args.ignore_savefile:
|
||||||
|
savefile = SaveFile(args.savefile)
|
||||||
|
else:
|
||||||
|
savefile = SaveFile(os.path.expanduser("~/.local/faulty/neo.save"))
|
||||||
|
|
||||||
|
# load the savefile
|
||||||
|
if not args.ignore_savefile:
|
||||||
|
files = savefile.get("files", files)
|
||||||
|
|
||||||
|
if len(args.include) > 0:
|
||||||
|
files = []
|
||||||
|
for i in args.include:
|
||||||
|
files.append(f"docker-compose.{i}.yml")
|
||||||
|
|
||||||
|
if len(args.exclude) > 0:
|
||||||
|
for i in args.exclude:
|
||||||
|
files.remove(f"docker-compose.{i}.yml")
|
||||||
|
|
||||||
|
if args.save:
|
||||||
|
savefile["files"] = files
|
||||||
|
return
|
||||||
|
|
||||||
|
def handle_aliases(fnlist: dict, name: str):
|
||||||
|
return fnlist.get(name, lambda: print("does not exist"))
|
||||||
|
|
||||||
|
fnlist = {
|
||||||
|
"up": lambda: dc(f"up -d {' '.join(args.args)}"),
|
||||||
|
"upr": lambda: dc(f"up -d --build {' '.join(args.args)}"),
|
||||||
|
"logs": lambda: dc(f"logs --tail=20 -f {' '.join(args.args)}"),
|
||||||
|
"setup": lambda: setup_handler(args.args[0]),
|
||||||
|
"delete-project-files": lambda: os.system(f"sudo rm -rf /srv/{args.args[0]}"),
|
||||||
|
|
||||||
|
"dpf": lambda: handle_aliases(fnlist, "delete-project-files")()
|
||||||
|
}
|
||||||
|
|
||||||
|
fnlist.get(args.action, lambda: dc(f"{args.action} {' '.join(args.args)}"))()
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
28
utils/manager/savefile.py
Normal file
28
utils/manager/savefile.py
Normal file
|
|
@ -0,0 +1,28 @@
|
||||||
|
import json
|
||||||
|
import os
|
||||||
|
|
||||||
|
|
||||||
|
class SaveFile(dict):
|
||||||
|
def __init__(self, file: str) -> None:
|
||||||
|
super().__init__()
|
||||||
|
self.path: str = file
|
||||||
|
_dir = os.path.dirname(self.path)
|
||||||
|
if not os.path.exists(_dir):
|
||||||
|
os.makedirs(_dir)
|
||||||
|
if os.path.exists(self.path):
|
||||||
|
with open(self.path) as f:
|
||||||
|
self.update(json.loads(f.read()))
|
||||||
|
else:
|
||||||
|
self.save()
|
||||||
|
|
||||||
|
def save(self):
|
||||||
|
with open(self.path, 'a+') as f:
|
||||||
|
f.write(json.dumps(self, indent=1))
|
||||||
|
|
||||||
|
def __getitem__(self, key):
|
||||||
|
return super().__getitem__(key)
|
||||||
|
|
||||||
|
def __setitem__(self, key, value):
|
||||||
|
super().__setitem__(key, value)
|
||||||
|
self.save()
|
||||||
|
|
||||||
Reference in a new issue