summaryrefslogtreecommitdiff
path: root/plugins/aliases/cheatsheet.py
diff options
context:
space:
mode:
authorhqingyi <huangb.qingy@gmail.com>2021-06-12 10:45:05 +0800
committerGitHub <noreply@github.com>2021-06-11 19:45:05 -0700
commite47a8e2321be9e7af49872588619c031a757d6ea (patch)
tree3e7f70f74d233ed6313fd7a660abd31f0fd377f5 /plugins/aliases/cheatsheet.py
parenta206271460ce49e842b1b410c0424b8c9a0a3d14 (diff)
downloadzsh-e47a8e2321be9e7af49872588619c031a757d6ea.tar.gz
zsh-e47a8e2321be9e7af49872588619c031a757d6ea.tar.bz2
zsh-e47a8e2321be9e7af49872588619c031a757d6ea.zip
feat(plugins): add aliases cheatsheet plugin (#4662)
* add: explore alias quickly with aliases plugin. * change: add compatibility with python2 & python3. - add compatibility. - add termcolor.py. - remove aliass(search), just use acs. - detect python.
Diffstat (limited to 'plugins/aliases/cheatsheet.py')
-rw-r--r--plugins/aliases/cheatsheet.py55
1 files changed, 55 insertions, 0 deletions
diff --git a/plugins/aliases/cheatsheet.py b/plugins/aliases/cheatsheet.py
new file mode 100644
index 000000000..d6d507b92
--- /dev/null
+++ b/plugins/aliases/cheatsheet.py
@@ -0,0 +1,55 @@
+#!/usr/bin/env python
+import sys
+import itertools
+import termcolor
+
+def parse(line):
+ left = line[0:line.find('=')].strip()
+ right = line[line.find('=')+1:].strip('\'"\n ')
+ try:
+ cmd = next(part for part in right.split() if len([char for char in '=<>' if char in part])==0)
+ except StopIteration:
+ cmd = right
+ return (left, right, cmd)
+
+def cheatsheet(lines):
+ exps = [ parse(line) for line in lines ]
+ cheatsheet = {'_default': []}
+ for key, group in itertools.groupby(exps, lambda exp:exp[2]):
+ group_list = [ item for item in group ]
+ if len(group_list)==1:
+ target_aliases = cheatsheet['_default']
+ else:
+ if key not in cheatsheet:
+ cheatsheet[key] = []
+ target_aliases = cheatsheet[key]
+ target_aliases.extend(group_list)
+ return cheatsheet
+
+def pretty_print_group(key, aliases, hightlight=None):
+ if len(aliases) == 0:
+ return
+ group_hl_formatter = lambda g, hl: termcolor.colored(hl, 'yellow').join([termcolor.colored(part, 'red') for part in ('[%s]' % g).split(hl)])
+ alias_hl_formatter = lambda alias, hl: termcolor.colored(hl, 'yellow').join([termcolor.colored(part, 'green') for part in ('\t%s = %s' % alias[0:2]).split(hl)])
+ group_formatter = lambda g: termcolor.colored('[%s]' % g, 'red')
+ alias_formatter = lambda alias: termcolor.colored('\t%s = %s' % alias[0:2], 'green')
+ if hightlight and len(hightlight)>0:
+ print (group_hl_formatter(key, hightlight))
+ print ('\n'.join([alias_hl_formatter(alias, hightlight) for alias in aliases]))
+ else:
+ print (group_formatter(key))
+ print ('\n'.join([alias_formatter(alias) for alias in aliases]))
+ print ('')
+
+def pretty_print(cheatsheet, wfilter):
+ sorted_key = sorted(cheatsheet.keys())
+ for key in sorted_key:
+ aliases = cheatsheet.get(key)
+ if not wfilter:
+ pretty_print_group(key, aliases, wfilter)
+ else:
+ pretty_print_group(key, [ alias for alias in aliases if alias[0].find(wfilter)>-1 or alias[1].find(wfilter)>-1], wfilter)
+
+if __name__ == '__main__':
+ lines = sys.stdin.readlines()
+ pretty_print(cheatsheet(lines), sys.argv[1] if len(sys.argv)>1 else None)