merge bash-tilde to master

This commit is contained in:
Ben Harris 2018-06-12 15:57:26 -04:00
commit d3395a42b9
2 changed files with 259 additions and 65 deletions

4
list
View File

@ -1,4 +0,0 @@
#!/bin/bash
cd /tilde/bin
ls | tr '\t' '\n'

320
tilde
View File

@ -1,74 +1,272 @@
#!/usr/bin/python2
import argparse,sys,os,subprocess
#!/bin/bash
# ---------------------------------------------------------------------------
# tilde - manage user-submitted scripts and apps
class TildeContrib(object):
def __init__(self,progdir):
self.progdir=progdir
pass
# Copyright 2018, Ben Harris <ben@tilde.team>
def __getattr__(self,k):
if k in ("list"):
return object.__getattr__(self,k)
return lambda x: os.system("/tilde/bin/{} {}".format(k," ".join(x)))
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
def list(self, x, inc=True):
cs = filter(None,subprocess.check_output(['/tilde/special/list']).split("\n"))
f = " ".join(x) if len(x)>0 else ""
ret = []
for c in cs:
if f in c:
ret.append(c)
if inc:
ret.extend(["list"])
return ret
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License at <http://www.gnu.org/licenses/> for
# more details.
class TildeLauncher:
# Usage: tilde [-h|--help]
COMMANDS = ["help","contrib"]
# ---------------------------------------------------------------------------
USAGE = dict(help="\n Displays this menu",contrib=" <program>\n Your gateway to programs made by teammates,\n for teammates!")
PROGNAME=${0##*/}
VERSION="0.0.1"
def __init__(self,prog_dir):
self.progdir = prog_dir
self.tc = TildeContrib(prog_dir)
def base(self):
print """ Welcome to tilde.team :) this program is your gateway to team-specific
commands and features. Run tilde help to see the sort of things you can do."""
clean_up() { # Perform pre-exit housekeeping
return
}
def help(self,argv):
commands_to_display = []
if len(argv)==0:
commands_to_display = self.COMMANDS
else:
commands_to_display = [i for i in argv if i in self.COMMANDS]
for cmd in commands_to_display:
print "tilde {}{}".format(cmd,self.USAGE[cmd])
def contrib(self,argv):
if not argv:
argv = ["list"] # default to listing
if argv[0]=="list":
print "Commands:"
for c in self.tc.list(argv[1:]):
print " tilde contrib "+c
# elif argv[0]=="bootstrap":
# for c in self.tc.list([],False):
# print("alias {0}=\"tilde contrib {0}\"".format(c))
else:
getattr(self.tc,argv[0])(argv[1:])
error_exit() {
echo -e "${1:-"unknown Error"}" >&2
clean_up
exit 1
}
# def chat(self,argv):
# os.system("/tilde/special/chat")
if __name__=="__main__":
tl = TildeLauncher("/tilde/bin")
argv = sys.argv[1:]
if len(argv)==0:
tl.base()
else:
if not hasattr(tl,argv[0]):
print " Unknown command %s!"
tl.help([])
graceful_exit() {
clean_up
exit
}
signal_exit() { # Handle trapped signals
case $1 in
INT)
error_exit "program interrupted by user" ;;
TERM)
echo -e "\n$PROGNAME: program terminated" >&2
graceful_exit ;;
*)
error_exit "$PROGNAME: terminating on unknown signal" ;;
esac
}
prompt_confirm() {
while true; do
read -r -n 1 -p "${1:-continue?} [y/n]: " REPLY
case $REPLY in
[yY]) echo ; return 0 ;;
[nN]) echo ; return 1 ;;
*) printf " \033[31m %s \n\033[0m" "invalid input"
esac
done
}
usage() {
echo -e "\nusage: $PROGNAME [help|list|submit|about|script_name]\n"
echo " $PROGNAME list - show a list of approved userscripts"
echo " $PROGNAME submit - start the submission flow for your own script"
[[ $(id -u) == 0 ]] && {
echo " $PROGNAME approve - enter the approval queue"
echo " $PROGNAME revoke <script_name> - send a script back to the author and remove from /tilde/bin"
}
echo " $PROGNAME about <script_name> - get the description for script_name"
echo " $PROGNAME <script_name> - run script_name with all remaining args are passed to the script"
if [[ :$PATH: != *:"/tilde/bin":* ]] ; then
echo -e "\nadd /tilde/bin to your PATH to use approved scripts without this wrapper"
echo "if you're using bash, run the following to add it quickly"
echo " echo 'export PATH=\$PATH:/tilde/bin' >> ~/.bashrc && source ~/.bashrc"
fi
}
help_message() {
cat <<- _EOF_
$PROGNAME ver. $VERSION
wrapper for user-submitted scripts
supports user submission and admin approval
$(usage)
_EOF_
return
}
verify_script_name() {
[[ $1 == "" ]] && error_exit "please start over and enter the script name"
[[ $(type -P "$1") ]] &&
[[ -x $HOME/bin/$1 ]] ||
error_exit "$1 already exists. rename your script and try again."
[[ -x /tilde/bin/$1 ]] && error_exit "$1 is already taken. rename your script and try again."
case $1 in
about|description|list|ls|submit|about|help|apropos|submit|approve)
error_exit "$1 is a subcommand of tilde. rename your script and try again." ;;
*)
return ;;
esac
}
submission_checklist() {
cat <<- _EOF_
requirements for submitting a user script or program:
- placed in your ~/bin
- executable
- responds to help or --help
- no name collisions with existing scripts or $PROGNAME subcommands
_EOF_
}
mail_body() {
cat <<- _EOF_
Subject: tilde script submission from $USER
From: $USER@tilde.team
To: ben@tilde.team
tilde script submission from $USER
script name: $1
description:
-----------------------------------------------------------------------
$2
-----------------------------------------------------------------------
you'll find the script and description in: /tilde/pending-submissions/$USER/$1
run this to see the approval queue:
sudo tilde approve
_EOF_
}
# Trap signals
trap "signal_exit TERM" TERM HUP
trap "signal_exit INT" INT
# Check for root UID
# Parse command-line
case $1 in
-h | --help | help)
help_message; graceful_exit ;;
-v | --version)
echo $VERSION ;;
-* | --*)
usage
error_exit "Unknown option $1" ;;
list | ls)
echo -e "available scripts:\n"
for scr in /tilde/bin/*; do
script_name=$(basename $scr)
target=$(readlink -f "$scr")
echo "$script_name by "$(stat -c '%U' $target)
cat /tilde/descriptions/$script_name
echo ""
done ;;
about | apropos | description)
if [[ -f /tilde/descriptions/$2 ]]; then
cat /tilde/descriptions/$2
else
echo "$2 not found. try $PROGNAME list to see available user scripts."
fi
;;
submit)
echo "hello, $USER! so it's time to submit your script?"
submission_checklist
prompt_confirm "are you ready to continue?" || graceful_exit
echo -n "enter the name of your script: "
read script_name
verify_script_name $script_name
if [[ -x $HOME/bin/$script_name ]]; then
echo "cool, found your script"
[[ -x /tilde/pending-submissions/$USER/$script_name/$script_name ]] && error_exit "you've already submitted $script_name"
else
error_exit "$script_name not found in ~/bin"
fi
echo "enter a description of your script: "
read description
echo -e "\nyour script, along with your description will be sent to the admins for approval"
prompt_confirm "ready to submit?" || graceful_exit
# submit now
mkdir -p /tilde/pending-submissions/$USER/$script_name
ln -s $HOME/bin/$script_name /tilde/pending-submissions/$USER/$script_name/$script_name
echo $description > /tilde/pending-submissions/$USER/$script_name/description.txt
mail_body $script_name "$description" | sendmail ben
echo "script submitted. thanks! :)" ;;
approve)
[[ $(id -u) != 0 ]] && error_exit "re-run this as sudo to access the approval queue"
echo -e "welcome to the approval queue\n\n"
for user in /tilde/pending-submissions/*; do
for scr in $user/*; do
user=$(basename $user)
script_name=$(basename $scr)
[[ -f $scr/approved ]] && continue
script=$scr/$script_name
echo "$script_name by $user"
cat $scr/description.txt
prompt_confirm "approve?" || continue
sudo ln -s $(readlink -f $script) /tilde/bin/$script_name
sudo cp $scr/description.txt /tilde/descriptions/$script_name
sudo touch $scr/approved
sudo chmod 664 /tilde/descriptions/*
echo "your submission of $script_name has been approved and is now available at /tilde/bin/$script_name" | sendmail $user
done
done
echo "~~done for now~~" ;;
revoke)
[[ $(id -u) != 0 ]] && error_exit "re-run this as sudo to access the revoke menu"
[[ -f /tilde/bin/$2 ]] || error_exit "$2 isn't an approved script"
prompt_confirm "revoke $2?"
echo -n "please provide a reason: "
read reason
original_script=$(readlink -f /tilde/bin/$2)
author=$(stat -c '%U' $original_script)
sudo rm /tilde/{bin,descriptions}/$2
sudo rm -rf /tilde/pending-submissions/$author/$2
echo -e "your script $2 has been returned because: $reason\nfeel free to resubmit" | sendmail $author
echo "$2 revoked and returned to author" ;;
*)
if [[ -f /tilde/bin/$1 ]]; then
prog=/tilde/bin/$1
shift
$prog "$@"
graceful_exit
else
[[ $1 == "" ]] || echo -e "$1 not found. try $PROGNAME list to see what's available\n"
help_message; graceful_exit;
fi ;;
esac
# Main logic
graceful_exit
getattr(tl,argv[0])(argv[1:])