You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
341 lines
11 KiB
341 lines
11 KiB
import sys
|
|
import os
|
|
import subprocess
|
|
import configparser
|
|
import magic
|
|
import json
|
|
import datetime
|
|
|
|
mime = magic.Magic(mime=True)
|
|
|
|
# supported values: fish, bash, zsh
|
|
if ( 'DO_WHAT_SHELL' in os.environ ):
|
|
shell=os.environ['DO_WHAT_SHELL']
|
|
else:
|
|
print("echo Shell type not set. To set up the environment, add the following line to your shell\\\'s init file:")
|
|
print("echo dowhat {shell} \| .")
|
|
exit()
|
|
|
|
# Defaults
|
|
print_file="cat"
|
|
pretty_print_file="less -R"
|
|
edit_file="vim"
|
|
list_directory="ls --color=auto"
|
|
list_git_directory="ls --color=auto"
|
|
change_dir="cd"
|
|
print_dir="pwd"
|
|
help_command="man"
|
|
path_locator="which"
|
|
open_file="xdg-open"
|
|
|
|
# general function for setting a shell's environment variable
|
|
def set_runtime_var(name, value, options=""):
|
|
if ( shell == "fish" ):
|
|
print("set "+name+" "+options+" \""+value+"\"")
|
|
elif ( shell == "bash" or shell == "zsh" ):
|
|
print("export "+name+"="+"\""+value+"\"")
|
|
|
|
# checks if a file is a binary file or a plaintext file
|
|
def is_binary_file(filepathname):
|
|
textchars = bytearray([7,8,9,10,12,13,27]) + bytearray(range(0x20, 0x7f)) + bytearray(range(0x80, 0x100))
|
|
is_binary_string = lambda bytes: bool(bytes.translate(None, textchars))
|
|
|
|
if is_binary_string(open(filepathname, 'rb').read(1024)):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
# check if a given program is in the path environment using which
|
|
def env_has(program):
|
|
def is_exe(fpath):
|
|
return os.path.isfile(fpath) and os.access(fpath, os.X_OK)
|
|
|
|
fpath, fname = os.path.split(program)
|
|
if fpath:
|
|
if is_exe(program):
|
|
return True
|
|
else:
|
|
for path in os.environ["PATH"].split(os.pathsep):
|
|
exe_file = os.path.join(path, program)
|
|
if is_exe(exe_file):
|
|
return True
|
|
|
|
return None
|
|
|
|
# get the number of lines in a file
|
|
def file_len(fname):
|
|
with open(fname) as f:
|
|
for i, l in enumerate(f):
|
|
pass
|
|
try:
|
|
i
|
|
except NameError:
|
|
return 0
|
|
else:
|
|
return i + 1
|
|
|
|
# bootstrap the config
|
|
ENV_HOME=os.environ['HOME']
|
|
subprocess.call(['mkdir', '-p', ENV_HOME+'/.config/do_what'])
|
|
if ( os.path.isfile(ENV_HOME+'/.config/do_what/what.config') ):
|
|
config = configparser.ConfigParser()
|
|
config.read(ENV_HOME+'/.config/do_what/what.config')
|
|
# TODO add else default configs
|
|
if ( 'DEFAULT' in config ):
|
|
if ( 'print_file' in config['DEFAULT'] ):
|
|
print_file = config['DEFAULT']['print_file']
|
|
if ( 'pretty_print_file' in config['DEFAULT'] ):
|
|
pretty_print_file = config['DEFAULT']['pretty_print_file']
|
|
if ( 'edit_file' in config['DEFAULT'] ):
|
|
edit_file = config['DEFAULT']['edit_file']
|
|
if ( 'list_directory' in config['DEFAULT'] ):
|
|
list_directory = config['DEFAULT']['list_directory']
|
|
if ( 'help_command' in config['DEFAULT'] ):
|
|
help_command = config['DEFAULT']['help_command']
|
|
if ( 'path_locator' in config['DEFAULT'] ):
|
|
path_locator = config['DEFAULT']['path_locator']
|
|
if ( 'list_git_directory' in config['DEFAULT'] ):
|
|
list_git_directory = config['DEFAULT']['list_git_directory']
|
|
if ( 'open_file' in config['DEFAULT'] ):
|
|
open_file = config['DEFAULT']['open_file']
|
|
else:
|
|
config = configparser.ConfigParser()
|
|
config['DEFAULT'] = {
|
|
'list_directory': list_directory,
|
|
'edit_file': edit_file,
|
|
'pretty_print_file': pretty_print_file,
|
|
'print_file': print_file,
|
|
'help_command': help_command,
|
|
'path_locator': path_locator,
|
|
'list_git_directory': list_git_directory,
|
|
'open_file': open_file
|
|
}
|
|
print("echo Default config loaded.")
|
|
with open(ENV_HOME+'/.config/do_what/what.config', 'w') as configfile:
|
|
config.write(configfile)
|
|
configfile.close()
|
|
|
|
# check if active mode is enabled
|
|
argiter = 0
|
|
active = False
|
|
for arg in sys.argv:
|
|
if ( str(arg) == '-' ):
|
|
active = True
|
|
break
|
|
argiter += 1
|
|
|
|
# absolutize the path for python
|
|
if ( (active and len(sys.argv) >= 3) or (not active and len(sys.argv) >= 2) ):
|
|
filearg = sys.argv[1]
|
|
path = sys.argv[1]
|
|
if ( path[0] != "/" and path[0] != "~" and path[0] != "." ):
|
|
path = os.getcwd()+"/"+path
|
|
elif ( path[0] == "." and len(path) > 1 and ( path[:2] == "./" ) ):
|
|
path = os.getcwd()+path[1:]
|
|
elif ( path[0] == "." and len(path) > 1 and ( path[:2] == ".." ) ):
|
|
path = os.getcwd()+"/"+path
|
|
elif ( path[0] == "." and len(path) == 1 ):
|
|
path = os.getcwd()
|
|
|
|
# CLIpboard (ha.)
|
|
if ( active and len(sys.argv) > argiter+1 and sys.argv[argiter+1] == 'c' ):
|
|
subprocess.call(['rm', '-rf', ENV_HOME+'/.config/do_what/clipboard'])
|
|
subprocess.call(['touch', ENV_HOME+'/.config/do_what/clipboard'])
|
|
if ( filearg == '-' ):
|
|
path = os.getcwd()
|
|
if ( os.path.isdir(path) or os.path.isfile(path) ):
|
|
with open(ENV_HOME+"/.config/do_what/clipboard", "w") as cb:
|
|
cbdata = { 'action': 'copy', 'path': path }
|
|
json.dump(cbdata, cb)
|
|
print("echo Copied: "+path)
|
|
exit()
|
|
else:
|
|
print("echo Unknown file/directory.")
|
|
exit()
|
|
elif ( active and len(sys.argv) > argiter+1 and sys.argv[argiter+1] == 'x' ):
|
|
subprocess.call(['rm', '-rf', ENV_HOME+'/.config/do_what/clipboard'])
|
|
subprocess.call(['touch', ENV_HOME+'/.config/do_what/clipboard'])
|
|
if ( filearg == '-' ):
|
|
path = os.getcwd()
|
|
if ( os.path.isdir(path) or os.path.isfile(path) ):
|
|
with open(ENV_HOME+"/.config/do_what/clipboard", "w") as cb:
|
|
cbdata = { 'action': 'cut', 'path': path }
|
|
json.dump(cbdata, cb)
|
|
print("echo Cut: "+path)
|
|
exit()
|
|
else:
|
|
print("echo Unknown file/directory.")
|
|
exit()
|
|
|
|
elif ( active and len(sys.argv) > argiter+1 and sys.argv[argiter+1] == 'p' ):
|
|
if ( os.path.isfile(ENV_HOME+"/.config/do_what/clipboard") ):
|
|
with open(ENV_HOME+"/.config/do_what/clipboard", "r") as cb:
|
|
try:
|
|
cbdata = json.load(cb)
|
|
except json.decoder.JSONDecodeError:
|
|
print("echo Invalid clipboard file.")
|
|
exit()
|
|
|
|
print("echo \""+str(cbdata)+"\"")
|
|
|
|
if ( cbdata['action'] == 'copy' ):
|
|
file_op = 'cp'
|
|
elif ( cbdata['action'] == 'cut' ):
|
|
file_op = 'mv'
|
|
|
|
if ( os.path.isdir( cbdata['path'] ) and filearg == '-' ):
|
|
print("echo Pasted directory: "+cbdata['path'])
|
|
print(file_op+" -r "+cbdata['path']+" .")
|
|
elif ( os.path.isdir( cbdata['path'] ) and os.path.isdir( path ) ):
|
|
print("echo Pasted directory "+cbdata['path']+" in "+path)
|
|
print(file_op+" -r "+cbdata['path']+" "+path)
|
|
elif ( os.path.isdir( cbdata['path'] ) and os.path.isfile( path ) ):
|
|
print("echo Cannot paste directory into regular file.")
|
|
print("echo \"("+cbdata['path']+" -> "+path+")\"")
|
|
elif ( os.path.isfile( cbdata['path'] ) and filearg == '-' ):
|
|
print("echo Pasted file: "+cbdata['path'])
|
|
print(file_op+" "+cbdata['path']+" .")
|
|
elif ( os.path.isfile( cbdata['path'] ) and os.path.isdir( path ) ):
|
|
print("echo Pasted file "+cbdata['path']+" in directory "+path)
|
|
print(file_op+" "+cbdata['path']+" "+path)
|
|
elif ( os.path.isfile( cbdata['path'] ) and os.path.isfile( path ) ):
|
|
print("echo Pasted file "+cbdata['path']+" over existing file "+path)
|
|
print(file_op+" "+cbdata['path']+" "+path)
|
|
elif ( os.path.isfile( cbdata['path'] ) and filearg != '-' ):
|
|
print("echo Pasted file "+cbdata['path']+" as new file "+path)
|
|
print(file_op+" "+cbdata['path']+" "+path)
|
|
elif ( os.path.isdir( cbdata['path'] ) and filearg != '-' ):
|
|
print("echo Pasted directory "+cbdata['path']+" as new directory "+path)
|
|
print(file_op+" "+cbdata['path']+" "+path)
|
|
else:
|
|
print("echo Invalid clipboard. Use \\\'c\\\' directive to copy a file or directory.")
|
|
else:
|
|
print("echo Empty clipboard. Use \\\'c\\\' directive to copy a file or directory.")
|
|
|
|
exit()
|
|
|
|
# File/Directory backup creator
|
|
# Capital B = Time versioned backup
|
|
elif ( active and len(sys.argv) > argiter+1 and sys.argv[argiter+1] == 'B' ):
|
|
if ( filearg == '-' ):
|
|
print("No file/directory specified to backup.")
|
|
elif ( os.path.isfile( path ) ):
|
|
bakname = ".bak-"+datetime.datetime.now().strftime("%Y-%B-%d_%I:%M%p")
|
|
print("cp "+path+" "+path+bakname)
|
|
print("echo Created file backup: "+path+bakname)
|
|
elif( os.path.isdir( path ) ):
|
|
bakname = ".bak-"+datetime.datetime.now().strftime("%Y-%B-%d_%I:%M%p")
|
|
print("cp -r "+path+" "+path+bakname)
|
|
print("echo Created directory backup: "+path+bakname)
|
|
exit()
|
|
# Lowercase b = un-versioned backup
|
|
elif ( active and len(sys.argv) > argiter+1 and sys.argv[argiter+1] == 'b' ):
|
|
if ( filearg == '-' ):
|
|
print("No file/directory specified to backup.")
|
|
elif ( os.path.isfile( path ) ):
|
|
bakname = ".bak"
|
|
print("cp "+path+" "+path+bakname)
|
|
print("echo Created file backup: "+path+bakname)
|
|
elif( os.path.isdir( path ) ):
|
|
bakname = ".bak"
|
|
print("cp -r "+path+" "+path+bakname)
|
|
print("echo Created directory backup: "+path+bakname)
|
|
exit()
|
|
# Lowercase r = restore the specified backup
|
|
elif ( active and len(sys.argv) > argiter+1 and sys.argv[argiter+1] == 'r' ):
|
|
if ( filearg == '-' ):
|
|
print("No file/directory specified to restore.")
|
|
elif ( os.path.isfile( path ) ):
|
|
if ( ".bak" not in path ):
|
|
print("echo "+path+" is not a valid backup file.")
|
|
else:
|
|
restorepath = path.split(".bak", 1)[0]
|
|
print("mv "+path+" "+restorepath)
|
|
print("echo Restored file: "+restorepath)
|
|
elif ( os.path.isdir( path ) ):
|
|
if ( ".bak" not in path ):
|
|
print("echo "+path+" is not a valid backup directory.")
|
|
else:
|
|
restorepath = path.split(".bak", 1)[0]
|
|
print("mv "+path+" "+restorepath)
|
|
print("echo Restored directory: "+restorepath)
|
|
exit()
|
|
|
|
elif ( active and len(sys.argv) > argiter+1 ):
|
|
print("echo Invalid directive: "+sys.argv[argiter+1])
|
|
exit()
|
|
|
|
# File Operations
|
|
if ( (not active and len(sys.argv) == 1) or (active and len(sys.argv) == 2) ):
|
|
# list the contents of the current directory
|
|
if ( not active ):
|
|
if ( os.path.isdir('.git') ):
|
|
print(list_git_directory)
|
|
else:
|
|
print(list_directory)
|
|
# print the working directory (active)
|
|
elif ( active ):
|
|
print(print_dir)
|
|
elif ( (not active and len(sys.argv) == 2) or (active and len(sys.argv) == 3) ):
|
|
# if directory, list its contents
|
|
if ( os.path.isdir(path) and not active ):
|
|
if ( os.path.isdir(path+"/.git") ):
|
|
print(list_git_directory+" "+path)
|
|
else:
|
|
print(list_directory+" "+path)
|
|
# if directory, change into it (active)
|
|
elif ( os.path.isdir(path) and active ):
|
|
print(change_dir+" "+path)
|
|
# if file, display its contents
|
|
elif ( os.path.isfile(path) and not active ):
|
|
# check if file is binary
|
|
if ( is_binary_file(path) ):
|
|
filemime = mime.from_file(path)
|
|
print("echo Binary file: "+filemime)
|
|
if ( filemime == "inode/symlink" ):
|
|
print("echo Symlink location: "+os.path.realpath(path))
|
|
else:
|
|
trows, tcolumns = os.popen('stty size', 'r').read().split()
|
|
# if file is taller than the terminal, pipe it to a pretty-print display
|
|
if ( int(trows) - 5 < int(file_len(path)) ):
|
|
set_runtime_var('LESSOPEN', '| /usr/share/source-highlight/src-hilite-lesspipe.sh %s')
|
|
set_runtime_var('LESS', ' -R ')
|
|
print(pretty_print_file+" "+path)
|
|
# if file is shorter than the terminal, print its contents
|
|
else:
|
|
print(print_file+" "+path)
|
|
# if file, open in editor (active)
|
|
elif ( os.path.isfile(path) and active ):
|
|
if ( is_binary_file(path) ):
|
|
print(open_file+" "+path)
|
|
else:
|
|
if ( os.access(path, os.W_OK) ):
|
|
print(edit_file+" "+path)
|
|
else:
|
|
print("sudo "+edit_file+" "+path)
|
|
|
|
else:
|
|
# check if file is a path program, then list using which/man on active
|
|
path_has = env_has(filearg)
|
|
# print the location of the program
|
|
if ( path_has and not active ):
|
|
print(path_locator+" "+filearg)
|
|
# display the help page for the program (active)
|
|
elif ( path_has and active ):
|
|
print(help_command+" "+filearg)
|
|
else:
|
|
print("echo Unknown file/directory.")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|