From 639452aa16d51b1b947d03f04c9eabe2b90482b6 Mon Sep 17 00:00:00 2001 From: Mattias Andrée Date: Mon, 20 Aug 2012 02:22:12 +0200 Subject: cleanup --- configure | 23 +- ponysay | 1099 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++- ponysay.py | 1097 ---------------------------------------------------------- testpony | 38 --- truncater.c | 153 --------- 5 files changed, 1104 insertions(+), 1306 deletions(-) delete mode 100755 ponysay.py delete mode 100644 testpony delete mode 100644 truncater.c diff --git a/configure b/configure index db6229d..15c069e 100755 --- a/configure +++ b/configure @@ -11,14 +11,14 @@ licenseFiles='COPYING' -oldInstalledFiles='bin/ponysaytruncater lib/ponysay/list.pl lib/ponysay/linklist.pl lib/ponysay/pq4ps lib/ponysay/pq4ps.pl lib/ponysay/pq4ps-list lib/ponysay/pq4ps-list.pl' -oldInstalledDirs='share/ponies share/ttyponies' -oldCompiledFiles='ponysaytruncater ponysay.install' +oldInstalledFiles='bin/ponysaytruncater lib/ponysay/truncater lib/ponysay/list.pl lib/ponysay/linklist.pl lib/ponysay/pq4ps lib/ponysay/pq4ps.pl lib/ponysay/pq4ps-list lib/ponysay/pq4ps-list.pl bin/ponysay.py bin/ponythink.py' +oldInstalledDirs='lib/ponysay share/ponies share/ttyponies' +oldCompiledFiles='truncater ponysaytruncater ponysay.install' oldCompiledDirs='' -installedFiles='bin/ponysay bin/ponythink bin/ponysay.py bin/ponythink.py doc/ponysay.pdf share/info/ponysay.info.gz share/info/ponythink.info.gz' +installedFiles='bin/ponysay bin/ponythink doc/ponysay.pdf share/info/ponysay.info.gz share/info/ponythink.info.gz' installedDirs='share/ponysay lib/ponysay' -compiledFiles='truncater ponysay.info.gz ponysay.py.install' +compiledFiles='ponysay.info.gz ponysay.py.install' compiledDirs='quotes' for man in $manFiles; do @@ -91,7 +91,7 @@ for arg in "$@"; do fi done -compileMethods='core truncater manpages ponysaycompletion ponythinkcompletion' +compileMethods='core quotes manpages ponysaycompletion ponythinkcompletion' installMethods='install-min'"$installPdf$installInfo$installMan$installManEs$installBash$installFish$installZsh" if [ ! "$installInfo" = '' ]; then compileMethods="$compileMethods infomanual" @@ -160,16 +160,11 @@ function makeMakefile() echo echo 'default: '"$compileMethods" echo - echo 'all: core truncater quotes manpages infomanual ponysaycompletion ponythinkcompletion' + echo 'all: core quotes manpages infomanual ponysaycompletion ponythinkcompletion' echo echo "core:" - correctPrefix 'ponysay.py' - echo - - echo "truncater:" - echo -en '\t' - echo '$(CC) $(CPPFLAGS) $(CFLAGS) $(LDFLAGS) -o "truncater" "truncater.c"' + correctPrefix 'ponysay' echo echo "quotes:" @@ -210,9 +205,7 @@ function makeMakefile() echo echo -en '\t' ; echo 'mkdir -p "$(INSTALLDIR)/bin/"' echo -en '\t' ; echo 'install "ponysay" "$(INSTALLDIR)/bin/ponysay"' - echo -en '\t' ; echo 'install "ponysay.py" "$(INSTALLDIR)/bin/ponysay.py"' echo -en '\t' ; echo 'ln -sf "ponysay" "$(INSTALLDIR)/bin/ponythink"' - echo -en '\t' ; echo 'ln -sf "ponysay.py" "$(INSTALLDIR)/bin/ponythink.py"' echo echo -en '\t' ; echo 'mkdir -p "$(INSTALLDIR)/lib/ponysay/"' echo -en '\t' ; echo 'install -s "truncater" "$(INSTALLDIR)/lib/ponysay/truncater"' diff --git a/ponysay b/ponysay index d7f3811..7256322 100755 --- a/ponysay +++ b/ponysay @@ -1,4 +1,1097 @@ -#!/usr/bin/env bash +#!/usr/bin/env python3 +# -*- coding: utf-8 -*- -# Run ponysay -"$0.py" "$@" +''' +ponysay.py - Ponysay, a cowsay wrapper for ponies +Copyright (C) 2012 Erkin Batu Altunbaş + +Authors: Erkin Batu Altunbaş: Project leader, helped write the first implementation + Mattias "maandree" Andrée: Major contributor of both implementions + Elis "etu" Axelsson: Major contributor of current implemention and patcher of the first implementation + Sven-Hendrik "svenstaro" Haase: Major contributor of the first implementation + Jan Alexander "heftig" Steffens: Major contributor of the first implementation + Kyah "L-four" Rindlisbacher: Patched the first implementation + +License: WTFPL +''' + +import os +import sys +import random +from subprocess import Popen, PIPE + + +''' +The version of ponysay +''' +VERSION = '2.1' + + +''' +The directory where ponysay is installed, this is modified when building with make +''' +INSTALLDIR = '/usr' + + +''' +The user's home directory +''' +HOME = os.environ['HOME'] + + +''' +Whether the program is execute in Linux VT (TTY) +''' +linuxvt = os.environ['TERM'] == 'linux' + + +''' +Whether the script is executed as ponythink +''' +isthink = (len(__file__) >= 8) and (__file__[-8:] == 'think.py') + + +''' +Whether stdin is piped +''' +pipelinein = not sys.stdin.isatty() + +''' +Whether stdout is piped +''' +pipelineout = False #not sys.stdout.isatty() # currently impossible, we need to get rid of the little shell script first + +''' +Whether stderr is piped +''' +pipelineerr = not sys.stderr.isatty() + + +''' +The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY) +''' +ponydirs = [] +if linuxvt: _ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', INSTALLDIR + '/share/ponysay/ttyponies/'] +else: _ponydirs = [HOME + '/.local/share/ponysay/ponies/', INSTALLDIR + '/share/ponysay/ponies/' ] +for ponydir in _ponydirs: + if os.path.isdir(ponydir): + ponydirs.append(ponydir) + + +''' +The directories where quotes files are stored +''' +quotedirs = [] +_quotedirs = [HOME + '/.local/share/ponysay/quotes/', INSTALLDIR + '/share/ponysay/quotes/'] +for quotedir in _quotedirs: + if os.path.isdir(quotedir): + quotedirs.append(quotedir) + + + +''' +This is the mane class of ponysay +''' +class Ponysay(): + ''' + Starts the part of the program the arguments indicate + ''' + def __init__(self, args): + if (args.argcount == 0) and not pipelinein: + args.help() + return + + if (args.opts['-l'] is not None) and pipelineout: + args.opts['--onelist'] = args.opts['-l'] + args.opts['-l'] = None + + if args.opts['-h'] is not None: args.help() + elif args.opts['--quoters'] is not None: self.quoters() + elif args.opts['--onelist'] is not None: self.onelist() + elif args.opts['-v'] is not None: self.version() + elif args.opts['-l'] is not None: self.list() + elif args.opts['-L'] is not None: self.linklist() + elif args.opts['-q'] is not None: self.quote(args) + else: self.print_pony(args) + + + ## + ## Auxiliary methods + ## + + ''' + Returns one file with full path, names is filter for names, also accepts filepaths. + ''' + def __getponypath(self, names = None): + ponies = {} + + if not names == None: + for name in names: + if os.path.isfile(name): + ponies[name] = name + + for ponydir in ponydirs: + for ponyfile in os.listdir(ponydir): + pony = ponyfile[:-5] + if pony not in ponies: + ponies[pony] = ponydir + ponyfile + + if names == None: + names = list(ponies.keys()) + + pony = names[random.randrange(0, len(names))] + if pony not in ponies: + sys.stderr.write('I have never heared of any pony named %s\n' % (pony)); + exit(1) + else: + return ponies[pony] + + + ''' + Returns a set with all ponies that have quotes and are displayable + ''' + def __quoters(self): + quotes = [] + quoteshash = set() + _quotes = [] + for quotedir in quotedirs: + _quotes += [item[:item.index('.')] for item in os.listdir(INSTALLDIR + '/share/ponysay/quotes/')] + for quote in _quotes: + if not quote == '': + if not quote in quoteshash: + quoteshash.add(quote) + quotes.append(quote) + + ponies = set() + for ponydir in ponydirs: + for pony in os.listdir(ponydir): + if not pony[0] == '.': + p = pony[:-5] # remove .pony + for quote in quotes: + if ('+' + p + '+') in ('+' + quote + '+'): + if not p in ponies: + ponies.add(p) + + return ponies + + + ''' + Returns a list with all (pony, quote file) pairs + ''' + def __quotes(self): + quotes = [] + for quotedir in quotedirs: + quotes += [quotedir + item for item in os.listdir(quotedir)] + rc = [] + + for ponydir in ponydirs: + for pony in os.listdir(ponydir): + if not pony[0] == '.': + p = pony[:-5] # remove .pony + for quote in quotes: + q = quote[quote.rindex('/') + 1:] + q = q[:q.rindex('.')] + if ('+' + p + '+') in ('+' + q + '+'): + rc.append((p, quote)) + + return rc + + + ''' + Gets the size of the terminal in (rows, columns) + ''' + def __gettermsize(self): + termsize = Popen(['stty', 'size'], stdout=PIPE, stdin=sys.stderr).communicate()[0] + termsize = termsize.decode('utf8', 'replace')[:-1].split(' ') # [:-1] removes a \n + termsize = [int(item) for item in termsize] + return termsize + + + ## + ## Listing methods + ## + + ''' + Lists the available ponies + ''' + def list(self): + termsize = self.__gettermsize() + quoters = self.__quoters() + + for ponydir in ponydirs: # Loop ponydirs + print('\033[1mponyfiles located in ' + ponydir + '\033[21m') + + ponies = os.listdir(ponydir) + ponies = [item[:-5] for item in ponies] # remove .pony from file name + ponies.sort() + + width = len(max(ponies, key = len)) + 2 # Get the longest ponyfilename lenght + 2 spaces + + x = 0 + for pony in ponies: + spacing = ' ' * (width - len(pony)) + print(('\033[1m' + pony + '\033[21m' if (pony in quoters) else pony) + spacing, end='') # Print ponyfilename + x += width + if x > (termsize[1] - width): # If too wide, make new line + print() + x = 0 + + print('\n'); + + + ''' + Lists the available ponies with alternatives inside brackets + ''' + def linklist(self): + termsize = self.__gettermsize() + quoters = self.__quoters() + + for ponydir in ponydirs: # Loop ponydirs + print('\033[1mponyfiles located in ' + ponydir + '\033[21m') + + files = os.listdir(ponydir) + files = [item[:-5] for item in files] # remove .pony from file name + files.sort() + pairs = [(item, os.readlink(ponydir + item + '.pony') if os.path.islink(ponydir + item + '.pony') else '') for item in files] + + ponymap = {} + for pair in pairs: + if pair[1] == '': + if pair[0] not in ponymap: + ponymap[pair[0]] = [] + else: + target = pair[1][:-5] + if '/' in target: + target = target[target.rindex('/') + 1:] + if target in ponymap: + ponymap[target].append(pair[0]) + else: + ponymap[target] = [pair[0]] + + width = 0 + ponies = [] + widths = [] + for pony in ponymap: + w = len(pony) + item = '\033[1m' + pony + '\033[21m' if (pony in quoters) else pony + syms = ponymap[pony] + if len(syms) > 0: + w += 2 + len(syms) + item += ' (' + first = True + for sym in syms: + w += len(sym) + if not first: + item += ' ' + else: + first = False + item += '\033[1m' + sym + '\033[21m' if (sym in quoters) else sym + item += ')' + ponies.append(item) + widths.append(w) + if width < w: + width = w + + width += 2; + x = 0 + index = 0 + for pony in ponies: + spacing = ' ' * (width - widths[index]) + index += 1 + print(pony + spacing, end='') # Print ponyfilename + x += width + if x > (termsize[1] - width): # If too wide, make new line + print() + x = 0 + + print('\n'); + + + ''' + Lists with all ponies that have quotes and are displayable + ''' + def quoters(self): + last = '' + ponies = [] + for pony in self.__quoters(): + ponies.append(pony) + ponies.sort() + for pony in ponies: + if not pony == last: + last = pony + print(pony) + + + ''' + Lists the available ponies one one column without anything bold + ''' + def onelist(self): + last = '' + ponies = [] + for ponydir in ponydirs: # Loop ponydirs + ponies += os.listdir(ponydir) + ponies = [item[:-5] for item in ponies] # remove .pony from file name + ponies.sort() + for pony in ponies: + if not pony == last: + last = pony + print(pony) + + + ## + ## Displaying methods + ## + + ''' + Prints the name of the program and the version of the program + ''' + def version(self): + print('%s %s' % ('ponysay', VERSION)) + + + ''' + Returns (the cowsay command, whether it is a custom program) + ''' + def __getcowsay(self): + if isthink: + cowthink = os.environ['PONYSAY_COWTHINK'] if 'PONYSAY_COWTHINK' in os.environ else None + return ('cowthink', False) if (cowthink is None) or (cowthink == '') else (cowthink, True) + + cowsay = os.environ['PONYSAY_COWSAY'] if 'PONYSAY_COWSAY' in os.environ else None + return ('cowsay', False) if (cowsay is None) or (cowsay == '') else (cowsay, True) + + + ''' + Print the pony with a speech or though bubble. message, pony and wrap from args are used. + ''' + def print_pony(self, args): + if args.message == None: + msg = ''.join(sys.stdin.readlines()).rstrip() + else: + msg = args.message + + + pony = self.__getponypath(args.opts['-f']) + (cowsay, customcowsay) = self.__getcowsay() + + if (len(pony) > 4) and (pony[-4:].lower() == '.png'): + pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' + pngcmd = ('img2ponysay -p -- ' if linuxvt else 'img2ponysay -- ') + pony + pngpipe = os.pipe() + Popen(pngcmd, stdout=os.fdopen(pngpipe[1], 'w'), shell=True).wait() + pony = '/proc/' + str(os.getpid()) + '/fd/' + str(pngpipe[0]) + + cmd = [cowsay, '-f', self.__kms(pony)] + if args.opts['-W'] is not None: + cmd += ['-W', args.opts['-W'][0]] + cmd.append(msg) + + if linuxvt: + print('\033[H\033[2J', end='') + + env_width = os.environ['PONYSAY_FULL_WIDTH'] if 'PONYSAY_FULL_WIDTH' in os.environ else None + if env_width is None: env_width = '' + widthtruncation = self.__gettermsize()[1] if env_width not in ('yes', 'y', '1') else None + messagewrap = int(args.opts['-W'][0]) if args.opts['-W'] is not None else None + + proc = Backend(message = msg, ponyfile = pony, wrapcolumn = messagewrap if messagewrap is not None else 40, width = widthtruncation) # Popen(cmd, stdout=PIPE, stdin=sys.stderr) + exit_value = 0 + #try: + proc.parse() + #except: + # exit_value = 1 + output = proc.output # proc.communicate()[0].decode('utf8', 'replace') + if (len(output) > 0) and (output[-1] == '\n'): + output = output[:-1] + + + env_bottom = os.environ['PONYSAY_BOTTOM'] if 'PONYSAY_BOTTOM' in os.environ else None + if env_bottom is None: env_bottom = '' + + env_height = os.environ['PONYSAY_TRUNCATE_HEIGHT'] if 'PONYSAY_TRUNCATE_HEIGHT' in os.environ else None + if env_height is None: env_height = '' + + env_lines = os.environ['PONYSAY_SHELL_LINES'] if 'PONYSAY_SHELL_LINES' in os.environ else None + if (env_lines is None) or (env_lines == ''): env_lines = '2' + + lines = self.__gettermsize()[0] - int(env_lines) + + + if not exit_value == 0: + sys.stderr.write('Unable to successfully execute' + (' custom ' if customcowsay else ' ') + 'cowsay [' + cowsay + ']\n') + else: + if linuxvt or (env_height is ('yes', 'y', '1')): + if env_bottom is ('yes', 'y', '1'): + for line in output[: -lines]: + print(line) + else: + for line in output[: lines]: + print(line) + else: + print(output); + + + ''' + Print the pony with a speech or though bubble and a self quote + ''' + def quote(self, args): + pairs = self.__quotes() + if len(args.opts['-q']) > 0: + ponyset = set(args.opts['-q']) + alts = [] + for pair in pairs: + if pair[0] in ponyset: + alts.append(pair) + pairs = alts + + if not len(pairs) == 0: + pair = pairs[random.randrange(0, len(pairs))] + qfile = None + try: + qfile = open(pair[1], 'r') + args.message = '\n'.join(qfile.readlines()).strip() + finally: + if qfile is not None: + qfile.close() + args.opts['-f'] = [pair[0]] + elif len(args.opts['-q']) == 0: + sys.stderr.write('All the ponies are mute! Call the Princess!\n') + exit(1) + else: + args.opts['-f'] = [args.opts['-q'][random.randrange(0, len(args.opts['-q']))]] + args.message = 'I got nuthin\' good to say :(' + + self.print_pony(args) + + + ''' + Returns the file name of the input pony converted to a KMS pony, or if KMS is not used, the input pony itself + ''' + def __kms(self, pony): + if not linuxvt: + return pony + + env_kms = os.environ['PONYSAY_KMS_PALETTE'] if 'PONYSAY_KMS_PALETTE' in os.environ else None + if env_kms is None: env_kms = '' + + env_kms_cmd = os.environ['PONYSAY_KMS_PALETTE_CMD'] if 'PONYSAY_KMS_PALETTE_CMD' in os.environ else None + if (env_kms_cmd is not None) and (not env_kms_cmd == ''): + env_kms = Popen(shlex.split(env_kms_cmd), stdout=PIPE, stdin=sys.stderr).communicate()[0].decode('utf8', 'replace') + if env_kms[-1] == '\n': + env_kms = env_kms[:-1] + + if env_kms == '': + return pony + + palette = env_kms + palettefile = env_kms.replace('\033]P', '') + + kmsponies = '/var/cache/ponysay/kmsponies/' + palettefile + kmspony = (kmsponies + pony).replace('//', '/') + + if not os.path.isfile(kmspony): + protokmsponies = '/var/cache/ponysay/protokmsponies/' + protokmspony = (protokmsponies + pony).replace('//', '/') + + protokmsponydir = protokmspony[:protokmspony.rindex('/')] + kmsponydir = kmspony[: kmspony.rindex('/')] + + _protokmspony = '\'' + protokmspony.replace('\'', '\'\\\'\'') + '\'' + _kmspony = '\'' + kmspony.replace('\'', '\'\\\'\'') + '\'' + _pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' + + if not os.path.isfile(protokmspony): + if not os.path.isdir(protokmsponydir): + os.makedirs(protokmsponydir) + if not os.system('ponysay2ttyponysay < ' + _pony + ' > ' + _protokmspony) == 0: + sys.stderr.write('Unable to run ponysay2ttyponysay successfully, you need util-say for KMS support\n') + exit(1) + + if not os.path.isdir(kmsponydir): + os.makedirs(kmsponydir) + if not os.system('tty2colourfultty -e -p ' + palette + ' < ' + _protokmspony + ' > ' + _kmspony) == 0: + sys.stderr.write('Unable to run tty2colourfultty successfully, you need util-say for KMS support\n') + exit(1) + + return kmspony + + + +ARGUMENTLESS = 0 +ARGUMENTED = 1 +VARIADIC = 2 +''' +Simple argument parser +''' +class ArgParser(): + ''' + Constructor. + The short description is printed on same line as the program name + ''' + def __init__(self, program, description, usage, longdescription = None): + self.__program = program + self.__description = description + self.__usage = usage + self.__longdescription = longdescription + self.__arguments = [] + self.opts = {} + self.optmap = {} + + + ''' + Add option that takes no arguments + ''' + def add_argumentless(self, alternatives, help = None): + ARGUMENTLESS + self.__arguments.append((ARGUMENTLESS, alternatives, None, help)) + stdalt = alternatives[0] + self.opts[stdalt] = None + for alt in alternatives: + self.optmap[alt] = (stdalt, ARGUMENTLESS) + + ''' + Add option that takes one argument + ''' + def add_argumented(self, alternatives, arg, help = None): + self.__arguments.append((ARGUMENTED, alternatives, arg, help)) + stdalt = alternatives[0] + self.opts[stdalt] = None + for alt in alternatives: + self.optmap[alt] = (stdalt, ARGUMENTED) + + ''' + Add option that takes all following argument + ''' + def add_variadic(self, alternatives, arg, help = None): + self.__arguments.append((VARIADIC, alternatives, arg, help)) + stdalt = alternatives[0] + self.opts[stdalt] = None + for alt in alternatives: + self.optmap[alt] = (stdalt, VARIADIC) + + + ''' + Parse arguments + ''' + def parse(self, argv = sys.argv): + self.argcount = len(argv) - 1 + self.files = [] + + argqueue = [] + optqueue = [] + deque = [] + for arg in argv[1:]: + deque.append(arg) + + dashed = False + tmpdashed = False + get = 0 + dontget = 0 + + def unrecognised(arg): + sys.stderr.write('%s: warning: unrecognised option %s\n' % (self.__program, arg)) + + while len(deque) != 0: + arg = deque[0] + deque = deque[1:] + if (get > 0) and (dontget == 0): + get -= 1 + argqueue.append(arg) + elif tmpdashed: + self.files.append(arg) + tmpdashed = False + elif dashed: self.files.append(arg) + elif arg == '++': tmpdashed = True + elif arg == '--': dashed = True + elif (len(arg) > 1) and ((arg[0] == '-') or (arg[0] == '+')): + if (len(arg) > 2) and ((arg[:2] == '--') or (arg[:2] == '++')): + if dontget > 0: + dontget -= 1 + elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTLESS): + optqueue.append(arg) + argqueue.append(None) + elif '=' in arg: + arg_opt = arg[:arg.index('=')] + if (arg_opt in self.optmap) and (self.optmap[arg_opt][1] >= ARGUMENTED): + optqueue.append(arg_opt) + argqueue.append(arg[arg.index('=') + 1:]) + if self.optmap[arg_opt][1] == VARIADIC: + dashed = True + else: + unrecognised(arg) + elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTED): + optqueue.append(arg) + get += 1 + elif (arg in self.optmap) and (self.optmap[arg][1] == VARIADIC): + optqueue.append(arg) + argqueue.append(None) + dashed = True + else: + unrecognised(arg) + else: + sign = arg[0] + i = 1 + n = len(arg) + while i < n: + narg = sign + arg[i] + i += 1 + if (narg in self.optmap): + if self.optmap[narg][1] == ARGUMENTLESS: + optqueue.append(narg) + argqueue.append(None) + elif self.optmap[narg][1] == ARGUMENTED: + optqueue.append(narg) + nargarg = arg[i:] + if len(nargarg) == 0: + get += 1 + else: + argqueue.append(nargarg) + break + elif self.optmap[narg][1] == VARIADIC: + optqueue.append(narg) + nargarg = arg[i:] + argqueue.append(nargarg if len(nargarg) > 0 else None) + dashed = True + break + else: + unrecognised(arg) + else: + self.files.append(arg) + + i = 0 + n = len(optqueue) + while i < n: + opt = optqueue[i] + arg = argqueue[i] + i += 1 + opt = self.optmap[opt][0] + if (opt not in self.opts) or (self.opts[opt] is None): + self.opts[opt] = [] + self.opts[opt].append(arg) + + for arg in self.__arguments: + if (arg[0] == VARIADIC): + varopt = self.opts[arg[1][0]] + if varopt is not None: + additional = ','.join(self.files).split(',') if len(self.files) > 0 else [] + if varopt[0] is None: + self.opts[arg[1][0]] = additional + else: + self.opts[arg[1][0]] = varopt[0].split(',') + additional + self.files = [] + break + + self.message = ' '.join(self.files) if len(self.files) > 0 else None + + + ''' + Prints a colourful help message + ''' + def help(self): + print('\033[1m%s\033[21m %s %s' % (self.__program, '-' if linuxvt else '—', self.__description)) + print() + if self.__longdescription is not None: + print(self.__longdescription) + print() + print() + + print('\033[1mUSAGE:\033[21m', end='') + first = True + for line in self.__usage.split('\n'): + if first: + first = False + else: + print(' or', end="") + print('\t%s' % (line)) + print() + + print('\033[1mSYNOPSIS:\033[21m') + print() + for opt in self.__arguments: + opt_type = opt[0] + opt_alts = opt[1] + opt_arg = opt[2] + opt_help = opt[3] + if opt_help is None: + continue + for opt_alt in opt_alts: + if opt_alt is opt_alts[-1]: + print('\t' + opt_alt, end='') + if opt_type == ARGUMENTED: print(' \033[4m%s\033[24m' % (opt_arg)) + elif opt_type == VARIADIC: print(' [\033[4m%s\033[24m...]' % (opt_arg)) + else: print() + else: + print('\t\033[2m' + opt_alt + '\033[22m') + first = True + for line in opt_help.split('\n'): + if first: + first = False + print('\t\t\033[32;1m%s\033[21;39m' % (line)) + else: + print('\t\t%s' % (line)) + print() + + print() + + + +usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m' +usage_wrap = '--wrap \033[4mCOLUMN\033[24m' +usage_listhelp = '(--list | ---altlist | --version | --help)' +usage_file = '[--pony \033[4mPONY\033[24m]... ([--] \033[4mmessage\033[24m | <<<\033[4mmessage\033[24m)' +usage_quote = '--quote [\033[4mPONY\033[24m...]' + +usage = '%s %s\n%s [%s] %s\n%s [%s] %s' % (usage_saythink, usage_listhelp, + usage_saythink, usage_wrap, usage_file, + usage_saythink, usage_wrap, usage_quote) + +usage = usage.replace('\033[', '\0') +for sym in ('[', ']', '(', ')', '|', '...'): + usage = usage.replace(sym, '\033[2m' + sym + '\033[22m') +usage = usage.replace('\0', '\033[') + +''' +Argument parsing +''' +opts = ArgParser(program = 'ponythink' if isthink else 'ponysay', + description = 'cowsay wrapper for ponies', + usage = usage, + longdescription = +'''Ponysay displays an image of a pony saying some text provided by the user. +If \033[4mmessage\033[24m is not provided, it accepts standard input. For an extensive +documentation run `info ponysay`, or for just a little more help than this +run `man ponysay`. Ponysay has so much more to offer than described here.''') + +opts.add_argumentless(['--quoters']) +opts.add_argumentless(['--onelist']) + +opts.add_argumentless(['-h', '--help'], help = 'Print this help message.') +opts.add_argumentless(['-v', '--version'], help = 'Print the version of the program.') +opts.add_argumentless(['-l', '--list'], help = 'List pony files.') +opts.add_argumentless(['-L', '--altlist'], help = 'List pony files with alternatives.') +opts.add_argumented( ['-W', '--wrap'], arg = "COLUMN", help = 'Specify the column when the message should be wrapped.') +opts.add_argumented( ['-f', '--pony'], arg = "PONY", help = 'Select a pony.\nEither a file name or a pony name.') +opts.add_variadic( ['-q', '--quote'], arg = "PONY", help = 'Select a ponies which will quote themself.') + +opts.parse() + + + +''' +Replacement for cowsay +''' +class Backend(): + ''' + Constructor + Takes message [string], ponyfile [filename string], wrapcolumn [None or an int] and width [None or an int] + ''' + def __init__(self, message, ponyfile, wrapcolumn, width): + self.message = message + self.ponyfile = ponyfile + self.wrapcolumn = wrapcolumn + self.width = width + + self.link = {'\\' : '\\', '/' : '/'} if not isthink else {'\\' : 'o', '/' : 'o'} + + self.output = None + self.pony = None + + + ''' + Process all data + ''' + def parse(self): + self.__expandMessage() + self.__loadFile() + self.__processPony() + self.__truncate() + + + def __expandMessage(self): + lines = self.message.split('\n') + buf = '' + for line in lines: + (i, n, x) = (0, len(line), 0) + while i < n: + c = line[i] + i += 1 + if c == '\033': + colour = self.__getcolour(line, i - 1) + i += len(colour) - 1 + buf += colour + elif c == '\t': + nx = 8 - (x & 7) + buf += ' ' * nx + x += nx + else: + buf += c + x += 1 + buf += '\n' + self.message = buf[:-1] + + + def __loadFile(self): + ponystream = None + try: + ponystream = open(self.ponyfile, 'r') + self.pony = ''.join(ponystream.readlines()) + finally: + if ponystream is not None: + ponystream.close() + + + def __truncate(self): + if self.width is None: + return + lines = self.output.split('\n') + self.output = '' + for line in lines: + (i, n, x) = (0, len(line), 0) + while i < n: + c = line[i] + i += 1 + if c == '\033': + colour = self.__getcolour(line, i - 1) + i += len(colour) - 1 + self.output += colour + else: + if x < self.width: + self.output += c + x += 1 + self.output += '\n' + self.output = self.output[:-1] + + + def __processPony(self): + self.output = '' + + variables = {'' : '$'} + for key in self.link: + variables[key] = self.link[key] + + indent = 0 + dollar = None + + (i, n) = (0, len(self.pony)) + while i < n: + c = self.pony[i] + if c == '\t': + n += 8 - (indent & 7) + ed = ' ' * (7 - (indent & 7)) + c = ' ' + self.pony = self.pony[:i] + ed + self.pony[i:] + i += 1 + if c == '$': + if dollar is not None: + if '=' in dollar: + name = dollar[:find('=')] + value = dollar[find('=') + 1:] + variables[name] = value + elif (len(dollar) < 7) or not (dollar[:7] == 'balloon'): + if dollar in ('\\', '/'): + i += len(variables[dollar]) - 1 + lines = variables[dollar].split('\n') + firstvarline = True + for line in lines: + if firstvarline: + firstvarline = False + else: + indent = 0 + self.output = '\n' + self.output += line + indent += len(line) + else: + (w, h) = (0, 0) + props = dollar[7:] + if len(props) == 0: + if ',' in props: + if props[0] is not ',': + w = int(props[:props.index(',')]) + h = int(props[props.index(',') + 1:]) + else: + w = int(props) + balloon = self.__getballoon(w, h, indent).split('\n') + balloonpre = '\n' + (' ' * indent) + self.output += balloon[0] + for line in balloon[1:]: + self.output += balloonpre; + self.output += line; + indent = 0 + dollar = None + else: + dollar = '' + elif dollar is not None: + if c == '\033': + c = self.pony[i] + i += 1 + dollar += c + elif c == '\033': + colour = self.__getcolour(self.pony, i - 1) + self.output += colour + i += len(colour) - 1 + elif c == '\n': + self.output += c + indent = 0 + else: + self.output += c + indent += 1 + + + def __getcolour(self, input, offset): + (i, n) = (offset, len(input)) + rc = input[i] + i += 1 + if i == n: return rc + c = input[i] + i += 1 + rc += c + + if c == ']': + if i == n: return rc + c = input[i] + i += 1 + rc += c + if c == 'P': + di = 0 + while (di < 7) and (i < n): + c = input[i] + i += 1 + di += 1 + rc += c + elif c == '[': + while i < n: + c = input[i] + i += 1 + rc += c + if (c == '~') or (('a' <= c) and (c <= 'z')) or (('A' <= c) and (c <= 'Z')): + break + + return rc + + + def __getballoon(self, width, height, left): + wrap = None + if self.wrapcolumn is not None: + wrap = self.wrapcolumn - left + + msg = self.message + if wrap is not None: + msg = self.__wrapMessage(msg, wrap) + lines = msg.split('\n') + h = 4 + len(lines) + w = 6 + len(max(lines, key = len)) + if w < width: w = width + if h < height: h = height + + rc = '/' + '-' * (w - 2) + '\\\n' + rc += '|' + ' ' * (w - 2) + '|\n' + for line in lines: + rc += '| ' + line + ' ' * (w - len(line) - 6) + ' |\n' + rc += '|' + ' ' * (w - 2) + '|\n' + rc += '\\' + '-' * (w - 2) + '/' + + return rc + + + def __wrapMessage(self, message, wrap): + lines = message.split('\n') + buf = '' + for line in lines: + b = [None] * len(line) + map = {} + (bi, cols, w) = (0, 0, wrap) + (indent, indentc) = (-1, 0) + + (i, n) = (0, len(line)) + while i <= n: + d = None + if i != n: + d = line[i] + i += 1 + if d == '\033': + b[bi] = d + bi += 1 + b[bi] = line[i] + d = line[i] + bi += 1 + i += 1 + if d == '[': + while True: + b[bi] = line[i] + d = line[i] + bi += 1 + i += 1 + if (('a' <= d) and (d <= 'z')) or (('A' <= d) and (d <= 'Z')) or (d == '~'): + break + elif d == ']': + b[bi] = line[i] + d = line[i] + bi += 1 + i += 1 + if d == 'P': + for j in range(0, 7): + b[bi] = line[i] + bi += 1 + i += 1 + elif (d is not None) and (d != ' '): + if indent == -1: + indent = i - 1 + for j in range(0, indent): + if line[j] == ' ': + indentc += 1 + b[bi] = d + bi += 1 + cols += 1 + map[cols] = bi + else: + mm = 0 + while (w > 8) and (cols > w + 3): + mm += w - 1 + m = map[mm] + for bb in b[:m]: + buf += bb + buf += '-\n' + cols -= w - 1 + m += w -1 + bi -= m + bb = b[m:] + for j in range(0, bi): + b[j] = bb[j] + w = wrap + if indent != -1: + buf += line[:indent] + w -= indentc + if cols > w: + buf += '\n' + w = wrap + if indent != -1: + buf += line[:indent] + w -= indentc + for bb in b[:bi]: + buf += bb + w -= cols + cols = 0 + bi = 0 + if d == -1: + i += 1 + else: + if w > 0: + buf += ' ' + w -= 1 + else: + buf += '\n' + w = wrap + if indent != -1: + buf + line[:indent] + w -= indentc + + buf += '\n' + return buf[:-1] + + + +''' +Start the program from ponysay.__init__ if this is the executed file +''' +if __name__ == '__main__': + Ponysay(opts) diff --git a/ponysay.py b/ponysay.py deleted file mode 100755 index 7256322..0000000 --- a/ponysay.py +++ /dev/null @@ -1,1097 +0,0 @@ -#!/usr/bin/env python3 -# -*- coding: utf-8 -*- - -''' -ponysay.py - Ponysay, a cowsay wrapper for ponies -Copyright (C) 2012 Erkin Batu Altunbaş - -Authors: Erkin Batu Altunbaş: Project leader, helped write the first implementation - Mattias "maandree" Andrée: Major contributor of both implementions - Elis "etu" Axelsson: Major contributor of current implemention and patcher of the first implementation - Sven-Hendrik "svenstaro" Haase: Major contributor of the first implementation - Jan Alexander "heftig" Steffens: Major contributor of the first implementation - Kyah "L-four" Rindlisbacher: Patched the first implementation - -License: WTFPL -''' - -import os -import sys -import random -from subprocess import Popen, PIPE - - -''' -The version of ponysay -''' -VERSION = '2.1' - - -''' -The directory where ponysay is installed, this is modified when building with make -''' -INSTALLDIR = '/usr' - - -''' -The user's home directory -''' -HOME = os.environ['HOME'] - - -''' -Whether the program is execute in Linux VT (TTY) -''' -linuxvt = os.environ['TERM'] == 'linux' - - -''' -Whether the script is executed as ponythink -''' -isthink = (len(__file__) >= 8) and (__file__[-8:] == 'think.py') - - -''' -Whether stdin is piped -''' -pipelinein = not sys.stdin.isatty() - -''' -Whether stdout is piped -''' -pipelineout = False #not sys.stdout.isatty() # currently impossible, we need to get rid of the little shell script first - -''' -Whether stderr is piped -''' -pipelineerr = not sys.stderr.isatty() - - -''' -The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY) -''' -ponydirs = [] -if linuxvt: _ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', INSTALLDIR + '/share/ponysay/ttyponies/'] -else: _ponydirs = [HOME + '/.local/share/ponysay/ponies/', INSTALLDIR + '/share/ponysay/ponies/' ] -for ponydir in _ponydirs: - if os.path.isdir(ponydir): - ponydirs.append(ponydir) - - -''' -The directories where quotes files are stored -''' -quotedirs = [] -_quotedirs = [HOME + '/.local/share/ponysay/quotes/', INSTALLDIR + '/share/ponysay/quotes/'] -for quotedir in _quotedirs: - if os.path.isdir(quotedir): - quotedirs.append(quotedir) - - - -''' -This is the mane class of ponysay -''' -class Ponysay(): - ''' - Starts the part of the program the arguments indicate - ''' - def __init__(self, args): - if (args.argcount == 0) and not pipelinein: - args.help() - return - - if (args.opts['-l'] is not None) and pipelineout: - args.opts['--onelist'] = args.opts['-l'] - args.opts['-l'] = None - - if args.opts['-h'] is not None: args.help() - elif args.opts['--quoters'] is not None: self.quoters() - elif args.opts['--onelist'] is not None: self.onelist() - elif args.opts['-v'] is not None: self.version() - elif args.opts['-l'] is not None: self.list() - elif args.opts['-L'] is not None: self.linklist() - elif args.opts['-q'] is not None: self.quote(args) - else: self.print_pony(args) - - - ## - ## Auxiliary methods - ## - - ''' - Returns one file with full path, names is filter for names, also accepts filepaths. - ''' - def __getponypath(self, names = None): - ponies = {} - - if not names == None: - for name in names: - if os.path.isfile(name): - ponies[name] = name - - for ponydir in ponydirs: - for ponyfile in os.listdir(ponydir): - pony = ponyfile[:-5] - if pony not in ponies: - ponies[pony] = ponydir + ponyfile - - if names == None: - names = list(ponies.keys()) - - pony = names[random.randrange(0, len(names))] - if pony not in ponies: - sys.stderr.write('I have never heared of any pony named %s\n' % (pony)); - exit(1) - else: - return ponies[pony] - - - ''' - Returns a set with all ponies that have quotes and are displayable - ''' - def __quoters(self): - quotes = [] - quoteshash = set() - _quotes = [] - for quotedir in quotedirs: - _quotes += [item[:item.index('.')] for item in os.listdir(INSTALLDIR + '/share/ponysay/quotes/')] - for quote in _quotes: - if not quote == '': - if not quote in quoteshash: - quoteshash.add(quote) - quotes.append(quote) - - ponies = set() - for ponydir in ponydirs: - for pony in os.listdir(ponydir): - if not pony[0] == '.': - p = pony[:-5] # remove .pony - for quote in quotes: - if ('+' + p + '+') in ('+' + quote + '+'): - if not p in ponies: - ponies.add(p) - - return ponies - - - ''' - Returns a list with all (pony, quote file) pairs - ''' - def __quotes(self): - quotes = [] - for quotedir in quotedirs: - quotes += [quotedir + item for item in os.listdir(quotedir)] - rc = [] - - for ponydir in ponydirs: - for pony in os.listdir(ponydir): - if not pony[0] == '.': - p = pony[:-5] # remove .pony - for quote in quotes: - q = quote[quote.rindex('/') + 1:] - q = q[:q.rindex('.')] - if ('+' + p + '+') in ('+' + q + '+'): - rc.append((p, quote)) - - return rc - - - ''' - Gets the size of the terminal in (rows, columns) - ''' - def __gettermsize(self): - termsize = Popen(['stty', 'size'], stdout=PIPE, stdin=sys.stderr).communicate()[0] - termsize = termsize.decode('utf8', 'replace')[:-1].split(' ') # [:-1] removes a \n - termsize = [int(item) for item in termsize] - return termsize - - - ## - ## Listing methods - ## - - ''' - Lists the available ponies - ''' - def list(self): - termsize = self.__gettermsize() - quoters = self.__quoters() - - for ponydir in ponydirs: # Loop ponydirs - print('\033[1mponyfiles located in ' + ponydir + '\033[21m') - - ponies = os.listdir(ponydir) - ponies = [item[:-5] for item in ponies] # remove .pony from file name - ponies.sort() - - width = len(max(ponies, key = len)) + 2 # Get the longest ponyfilename lenght + 2 spaces - - x = 0 - for pony in ponies: - spacing = ' ' * (width - len(pony)) - print(('\033[1m' + pony + '\033[21m' if (pony in quoters) else pony) + spacing, end='') # Print ponyfilename - x += width - if x > (termsize[1] - width): # If too wide, make new line - print() - x = 0 - - print('\n'); - - - ''' - Lists the available ponies with alternatives inside brackets - ''' - def linklist(self): - termsize = self.__gettermsize() - quoters = self.__quoters() - - for ponydir in ponydirs: # Loop ponydirs - print('\033[1mponyfiles located in ' + ponydir + '\033[21m') - - files = os.listdir(ponydir) - files = [item[:-5] for item in files] # remove .pony from file name - files.sort() - pairs = [(item, os.readlink(ponydir + item + '.pony') if os.path.islink(ponydir + item + '.pony') else '') for item in files] - - ponymap = {} - for pair in pairs: - if pair[1] == '': - if pair[0] not in ponymap: - ponymap[pair[0]] = [] - else: - target = pair[1][:-5] - if '/' in target: - target = target[target.rindex('/') + 1:] - if target in ponymap: - ponymap[target].append(pair[0]) - else: - ponymap[target] = [pair[0]] - - width = 0 - ponies = [] - widths = [] - for pony in ponymap: - w = len(pony) - item = '\033[1m' + pony + '\033[21m' if (pony in quoters) else pony - syms = ponymap[pony] - if len(syms) > 0: - w += 2 + len(syms) - item += ' (' - first = True - for sym in syms: - w += len(sym) - if not first: - item += ' ' - else: - first = False - item += '\033[1m' + sym + '\033[21m' if (sym in quoters) else sym - item += ')' - ponies.append(item) - widths.append(w) - if width < w: - width = w - - width += 2; - x = 0 - index = 0 - for pony in ponies: - spacing = ' ' * (width - widths[index]) - index += 1 - print(pony + spacing, end='') # Print ponyfilename - x += width - if x > (termsize[1] - width): # If too wide, make new line - print() - x = 0 - - print('\n'); - - - ''' - Lists with all ponies that have quotes and are displayable - ''' - def quoters(self): - last = '' - ponies = [] - for pony in self.__quoters(): - ponies.append(pony) - ponies.sort() - for pony in ponies: - if not pony == last: - last = pony - print(pony) - - - ''' - Lists the available ponies one one column without anything bold - ''' - def onelist(self): - last = '' - ponies = [] - for ponydir in ponydirs: # Loop ponydirs - ponies += os.listdir(ponydir) - ponies = [item[:-5] for item in ponies] # remove .pony from file name - ponies.sort() - for pony in ponies: - if not pony == last: - last = pony - print(pony) - - - ## - ## Displaying methods - ## - - ''' - Prints the name of the program and the version of the program - ''' - def version(self): - print('%s %s' % ('ponysay', VERSION)) - - - ''' - Returns (the cowsay command, whether it is a custom program) - ''' - def __getcowsay(self): - if isthink: - cowthink = os.environ['PONYSAY_COWTHINK'] if 'PONYSAY_COWTHINK' in os.environ else None - return ('cowthink', False) if (cowthink is None) or (cowthink == '') else (cowthink, True) - - cowsay = os.environ['PONYSAY_COWSAY'] if 'PONYSAY_COWSAY' in os.environ else None - return ('cowsay', False) if (cowsay is None) or (cowsay == '') else (cowsay, True) - - - ''' - Print the pony with a speech or though bubble. message, pony and wrap from args are used. - ''' - def print_pony(self, args): - if args.message == None: - msg = ''.join(sys.stdin.readlines()).rstrip() - else: - msg = args.message - - - pony = self.__getponypath(args.opts['-f']) - (cowsay, customcowsay) = self.__getcowsay() - - if (len(pony) > 4) and (pony[-4:].lower() == '.png'): - pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' - pngcmd = ('img2ponysay -p -- ' if linuxvt else 'img2ponysay -- ') + pony - pngpipe = os.pipe() - Popen(pngcmd, stdout=os.fdopen(pngpipe[1], 'w'), shell=True).wait() - pony = '/proc/' + str(os.getpid()) + '/fd/' + str(pngpipe[0]) - - cmd = [cowsay, '-f', self.__kms(pony)] - if args.opts['-W'] is not None: - cmd += ['-W', args.opts['-W'][0]] - cmd.append(msg) - - if linuxvt: - print('\033[H\033[2J', end='') - - env_width = os.environ['PONYSAY_FULL_WIDTH'] if 'PONYSAY_FULL_WIDTH' in os.environ else None - if env_width is None: env_width = '' - widthtruncation = self.__gettermsize()[1] if env_width not in ('yes', 'y', '1') else None - messagewrap = int(args.opts['-W'][0]) if args.opts['-W'] is not None else None - - proc = Backend(message = msg, ponyfile = pony, wrapcolumn = messagewrap if messagewrap is not None else 40, width = widthtruncation) # Popen(cmd, stdout=PIPE, stdin=sys.stderr) - exit_value = 0 - #try: - proc.parse() - #except: - # exit_value = 1 - output = proc.output # proc.communicate()[0].decode('utf8', 'replace') - if (len(output) > 0) and (output[-1] == '\n'): - output = output[:-1] - - - env_bottom = os.environ['PONYSAY_BOTTOM'] if 'PONYSAY_BOTTOM' in os.environ else None - if env_bottom is None: env_bottom = '' - - env_height = os.environ['PONYSAY_TRUNCATE_HEIGHT'] if 'PONYSAY_TRUNCATE_HEIGHT' in os.environ else None - if env_height is None: env_height = '' - - env_lines = os.environ['PONYSAY_SHELL_LINES'] if 'PONYSAY_SHELL_LINES' in os.environ else None - if (env_lines is None) or (env_lines == ''): env_lines = '2' - - lines = self.__gettermsize()[0] - int(env_lines) - - - if not exit_value == 0: - sys.stderr.write('Unable to successfully execute' + (' custom ' if customcowsay else ' ') + 'cowsay [' + cowsay + ']\n') - else: - if linuxvt or (env_height is ('yes', 'y', '1')): - if env_bottom is ('yes', 'y', '1'): - for line in output[: -lines]: - print(line) - else: - for line in output[: lines]: - print(line) - else: - print(output); - - - ''' - Print the pony with a speech or though bubble and a self quote - ''' - def quote(self, args): - pairs = self.__quotes() - if len(args.opts['-q']) > 0: - ponyset = set(args.opts['-q']) - alts = [] - for pair in pairs: - if pair[0] in ponyset: - alts.append(pair) - pairs = alts - - if not len(pairs) == 0: - pair = pairs[random.randrange(0, len(pairs))] - qfile = None - try: - qfile = open(pair[1], 'r') - args.message = '\n'.join(qfile.readlines()).strip() - finally: - if qfile is not None: - qfile.close() - args.opts['-f'] = [pair[0]] - elif len(args.opts['-q']) == 0: - sys.stderr.write('All the ponies are mute! Call the Princess!\n') - exit(1) - else: - args.opts['-f'] = [args.opts['-q'][random.randrange(0, len(args.opts['-q']))]] - args.message = 'I got nuthin\' good to say :(' - - self.print_pony(args) - - - ''' - Returns the file name of the input pony converted to a KMS pony, or if KMS is not used, the input pony itself - ''' - def __kms(self, pony): - if not linuxvt: - return pony - - env_kms = os.environ['PONYSAY_KMS_PALETTE'] if 'PONYSAY_KMS_PALETTE' in os.environ else None - if env_kms is None: env_kms = '' - - env_kms_cmd = os.environ['PONYSAY_KMS_PALETTE_CMD'] if 'PONYSAY_KMS_PALETTE_CMD' in os.environ else None - if (env_kms_cmd is not None) and (not env_kms_cmd == ''): - env_kms = Popen(shlex.split(env_kms_cmd), stdout=PIPE, stdin=sys.stderr).communicate()[0].decode('utf8', 'replace') - if env_kms[-1] == '\n': - env_kms = env_kms[:-1] - - if env_kms == '': - return pony - - palette = env_kms - palettefile = env_kms.replace('\033]P', '') - - kmsponies = '/var/cache/ponysay/kmsponies/' + palettefile - kmspony = (kmsponies + pony).replace('//', '/') - - if not os.path.isfile(kmspony): - protokmsponies = '/var/cache/ponysay/protokmsponies/' - protokmspony = (protokmsponies + pony).replace('//', '/') - - protokmsponydir = protokmspony[:protokmspony.rindex('/')] - kmsponydir = kmspony[: kmspony.rindex('/')] - - _protokmspony = '\'' + protokmspony.replace('\'', '\'\\\'\'') + '\'' - _kmspony = '\'' + kmspony.replace('\'', '\'\\\'\'') + '\'' - _pony = '\'' + pony.replace('\'', '\'\\\'\'') + '\'' - - if not os.path.isfile(protokmspony): - if not os.path.isdir(protokmsponydir): - os.makedirs(protokmsponydir) - if not os.system('ponysay2ttyponysay < ' + _pony + ' > ' + _protokmspony) == 0: - sys.stderr.write('Unable to run ponysay2ttyponysay successfully, you need util-say for KMS support\n') - exit(1) - - if not os.path.isdir(kmsponydir): - os.makedirs(kmsponydir) - if not os.system('tty2colourfultty -e -p ' + palette + ' < ' + _protokmspony + ' > ' + _kmspony) == 0: - sys.stderr.write('Unable to run tty2colourfultty successfully, you need util-say for KMS support\n') - exit(1) - - return kmspony - - - -ARGUMENTLESS = 0 -ARGUMENTED = 1 -VARIADIC = 2 -''' -Simple argument parser -''' -class ArgParser(): - ''' - Constructor. - The short description is printed on same line as the program name - ''' - def __init__(self, program, description, usage, longdescription = None): - self.__program = program - self.__description = description - self.__usage = usage - self.__longdescription = longdescription - self.__arguments = [] - self.opts = {} - self.optmap = {} - - - ''' - Add option that takes no arguments - ''' - def add_argumentless(self, alternatives, help = None): - ARGUMENTLESS - self.__arguments.append((ARGUMENTLESS, alternatives, None, help)) - stdalt = alternatives[0] - self.opts[stdalt] = None - for alt in alternatives: - self.optmap[alt] = (stdalt, ARGUMENTLESS) - - ''' - Add option that takes one argument - ''' - def add_argumented(self, alternatives, arg, help = None): - self.__arguments.append((ARGUMENTED, alternatives, arg, help)) - stdalt = alternatives[0] - self.opts[stdalt] = None - for alt in alternatives: - self.optmap[alt] = (stdalt, ARGUMENTED) - - ''' - Add option that takes all following argument - ''' - def add_variadic(self, alternatives, arg, help = None): - self.__arguments.append((VARIADIC, alternatives, arg, help)) - stdalt = alternatives[0] - self.opts[stdalt] = None - for alt in alternatives: - self.optmap[alt] = (stdalt, VARIADIC) - - - ''' - Parse arguments - ''' - def parse(self, argv = sys.argv): - self.argcount = len(argv) - 1 - self.files = [] - - argqueue = [] - optqueue = [] - deque = [] - for arg in argv[1:]: - deque.append(arg) - - dashed = False - tmpdashed = False - get = 0 - dontget = 0 - - def unrecognised(arg): - sys.stderr.write('%s: warning: unrecognised option %s\n' % (self.__program, arg)) - - while len(deque) != 0: - arg = deque[0] - deque = deque[1:] - if (get > 0) and (dontget == 0): - get -= 1 - argqueue.append(arg) - elif tmpdashed: - self.files.append(arg) - tmpdashed = False - elif dashed: self.files.append(arg) - elif arg == '++': tmpdashed = True - elif arg == '--': dashed = True - elif (len(arg) > 1) and ((arg[0] == '-') or (arg[0] == '+')): - if (len(arg) > 2) and ((arg[:2] == '--') or (arg[:2] == '++')): - if dontget > 0: - dontget -= 1 - elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTLESS): - optqueue.append(arg) - argqueue.append(None) - elif '=' in arg: - arg_opt = arg[:arg.index('=')] - if (arg_opt in self.optmap) and (self.optmap[arg_opt][1] >= ARGUMENTED): - optqueue.append(arg_opt) - argqueue.append(arg[arg.index('=') + 1:]) - if self.optmap[arg_opt][1] == VARIADIC: - dashed = True - else: - unrecognised(arg) - elif (arg in self.optmap) and (self.optmap[arg][1] == ARGUMENTED): - optqueue.append(arg) - get += 1 - elif (arg in self.optmap) and (self.optmap[arg][1] == VARIADIC): - optqueue.append(arg) - argqueue.append(None) - dashed = True - else: - unrecognised(arg) - else: - sign = arg[0] - i = 1 - n = len(arg) - while i < n: - narg = sign + arg[i] - i += 1 - if (narg in self.optmap): - if self.optmap[narg][1] == ARGUMENTLESS: - optqueue.append(narg) - argqueue.append(None) - elif self.optmap[narg][1] == ARGUMENTED: - optqueue.append(narg) - nargarg = arg[i:] - if len(nargarg) == 0: - get += 1 - else: - argqueue.append(nargarg) - break - elif self.optmap[narg][1] == VARIADIC: - optqueue.append(narg) - nargarg = arg[i:] - argqueue.append(nargarg if len(nargarg) > 0 else None) - dashed = True - break - else: - unrecognised(arg) - else: - self.files.append(arg) - - i = 0 - n = len(optqueue) - while i < n: - opt = optqueue[i] - arg = argqueue[i] - i += 1 - opt = self.optmap[opt][0] - if (opt not in self.opts) or (self.opts[opt] is None): - self.opts[opt] = [] - self.opts[opt].append(arg) - - for arg in self.__arguments: - if (arg[0] == VARIADIC): - varopt = self.opts[arg[1][0]] - if varopt is not None: - additional = ','.join(self.files).split(',') if len(self.files) > 0 else [] - if varopt[0] is None: - self.opts[arg[1][0]] = additional - else: - self.opts[arg[1][0]] = varopt[0].split(',') + additional - self.files = [] - break - - self.message = ' '.join(self.files) if len(self.files) > 0 else None - - - ''' - Prints a colourful help message - ''' - def help(self): - print('\033[1m%s\033[21m %s %s' % (self.__program, '-' if linuxvt else '—', self.__description)) - print() - if self.__longdescription is not None: - print(self.__longdescription) - print() - print() - - print('\033[1mUSAGE:\033[21m', end='') - first = True - for line in self.__usage.split('\n'): - if first: - first = False - else: - print(' or', end="") - print('\t%s' % (line)) - print() - - print('\033[1mSYNOPSIS:\033[21m') - print() - for opt in self.__arguments: - opt_type = opt[0] - opt_alts = opt[1] - opt_arg = opt[2] - opt_help = opt[3] - if opt_help is None: - continue - for opt_alt in opt_alts: - if opt_alt is opt_alts[-1]: - print('\t' + opt_alt, end='') - if opt_type == ARGUMENTED: print(' \033[4m%s\033[24m' % (opt_arg)) - elif opt_type == VARIADIC: print(' [\033[4m%s\033[24m...]' % (opt_arg)) - else: print() - else: - print('\t\033[2m' + opt_alt + '\033[22m') - first = True - for line in opt_help.split('\n'): - if first: - first = False - print('\t\t\033[32;1m%s\033[21;39m' % (line)) - else: - print('\t\t%s' % (line)) - print() - - print() - - - -usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m' -usage_wrap = '--wrap \033[4mCOLUMN\033[24m' -usage_listhelp = '(--list | ---altlist | --version | --help)' -usage_file = '[--pony \033[4mPONY\033[24m]... ([--] \033[4mmessage\033[24m | <<<\033[4mmessage\033[24m)' -usage_quote = '--quote [\033[4mPONY\033[24m...]' - -usage = '%s %s\n%s [%s] %s\n%s [%s] %s' % (usage_saythink, usage_listhelp, - usage_saythink, usage_wrap, usage_file, - usage_saythink, usage_wrap, usage_quote) - -usage = usage.replace('\033[', '\0') -for sym in ('[', ']', '(', ')', '|', '...'): - usage = usage.replace(sym, '\033[2m' + sym + '\033[22m') -usage = usage.replace('\0', '\033[') - -''' -Argument parsing -''' -opts = ArgParser(program = 'ponythink' if isthink else 'ponysay', - description = 'cowsay wrapper for ponies', - usage = usage, - longdescription = -'''Ponysay displays an image of a pony saying some text provided by the user. -If \033[4mmessage\033[24m is not provided, it accepts standard input. For an extensive -documentation run `info ponysay`, or for just a little more help than this -run `man ponysay`. Ponysay has so much more to offer than described here.''') - -opts.add_argumentless(['--quoters']) -opts.add_argumentless(['--onelist']) - -opts.add_argumentless(['-h', '--help'], help = 'Print this help message.') -opts.add_argumentless(['-v', '--version'], help = 'Print the version of the program.') -opts.add_argumentless(['-l', '--list'], help = 'List pony files.') -opts.add_argumentless(['-L', '--altlist'], help = 'List pony files with alternatives.') -opts.add_argumented( ['-W', '--wrap'], arg = "COLUMN", help = 'Specify the column when the message should be wrapped.') -opts.add_argumented( ['-f', '--pony'], arg = "PONY", help = 'Select a pony.\nEither a file name or a pony name.') -opts.add_variadic( ['-q', '--quote'], arg = "PONY", help = 'Select a ponies which will quote themself.') - -opts.parse() - - - -''' -Replacement for cowsay -''' -class Backend(): - ''' - Constructor - Takes message [string], ponyfile [filename string], wrapcolumn [None or an int] and width [None or an int] - ''' - def __init__(self, message, ponyfile, wrapcolumn, width): - self.message = message - self.ponyfile = ponyfile - self.wrapcolumn = wrapcolumn - self.width = width - - self.link = {'\\' : '\\', '/' : '/'} if not isthink else {'\\' : 'o', '/' : 'o'} - - self.output = None - self.pony = None - - - ''' - Process all data - ''' - def parse(self): - self.__expandMessage() - self.__loadFile() - self.__processPony() - self.__truncate() - - - def __expandMessage(self): - lines = self.message.split('\n') - buf = '' - for line in lines: - (i, n, x) = (0, len(line), 0) - while i < n: - c = line[i] - i += 1 - if c == '\033': - colour = self.__getcolour(line, i - 1) - i += len(colour) - 1 - buf += colour - elif c == '\t': - nx = 8 - (x & 7) - buf += ' ' * nx - x += nx - else: - buf += c - x += 1 - buf += '\n' - self.message = buf[:-1] - - - def __loadFile(self): - ponystream = None - try: - ponystream = open(self.ponyfile, 'r') - self.pony = ''.join(ponystream.readlines()) - finally: - if ponystream is not None: - ponystream.close() - - - def __truncate(self): - if self.width is None: - return - lines = self.output.split('\n') - self.output = '' - for line in lines: - (i, n, x) = (0, len(line), 0) - while i < n: - c = line[i] - i += 1 - if c == '\033': - colour = self.__getcolour(line, i - 1) - i += len(colour) - 1 - self.output += colour - else: - if x < self.width: - self.output += c - x += 1 - self.output += '\n' - self.output = self.output[:-1] - - - def __processPony(self): - self.output = '' - - variables = {'' : '$'} - for key in self.link: - variables[key] = self.link[key] - - indent = 0 - dollar = None - - (i, n) = (0, len(self.pony)) - while i < n: - c = self.pony[i] - if c == '\t': - n += 8 - (indent & 7) - ed = ' ' * (7 - (indent & 7)) - c = ' ' - self.pony = self.pony[:i] + ed + self.pony[i:] - i += 1 - if c == '$': - if dollar is not None: - if '=' in dollar: - name = dollar[:find('=')] - value = dollar[find('=') + 1:] - variables[name] = value - elif (len(dollar) < 7) or not (dollar[:7] == 'balloon'): - if dollar in ('\\', '/'): - i += len(variables[dollar]) - 1 - lines = variables[dollar].split('\n') - firstvarline = True - for line in lines: - if firstvarline: - firstvarline = False - else: - indent = 0 - self.output = '\n' - self.output += line - indent += len(line) - else: - (w, h) = (0, 0) - props = dollar[7:] - if len(props) == 0: - if ',' in props: - if props[0] is not ',': - w = int(props[:props.index(',')]) - h = int(props[props.index(',') + 1:]) - else: - w = int(props) - balloon = self.__getballoon(w, h, indent).split('\n') - balloonpre = '\n' + (' ' * indent) - self.output += balloon[0] - for line in balloon[1:]: - self.output += balloonpre; - self.output += line; - indent = 0 - dollar = None - else: - dollar = '' - elif dollar is not None: - if c == '\033': - c = self.pony[i] - i += 1 - dollar += c - elif c == '\033': - colour = self.__getcolour(self.pony, i - 1) - self.output += colour - i += len(colour) - 1 - elif c == '\n': - self.output += c - indent = 0 - else: - self.output += c - indent += 1 - - - def __getcolour(self, input, offset): - (i, n) = (offset, len(input)) - rc = input[i] - i += 1 - if i == n: return rc - c = input[i] - i += 1 - rc += c - - if c == ']': - if i == n: return rc - c = input[i] - i += 1 - rc += c - if c == 'P': - di = 0 - while (di < 7) and (i < n): - c = input[i] - i += 1 - di += 1 - rc += c - elif c == '[': - while i < n: - c = input[i] - i += 1 - rc += c - if (c == '~') or (('a' <= c) and (c <= 'z')) or (('A' <= c) and (c <= 'Z')): - break - - return rc - - - def __getballoon(self, width, height, left): - wrap = None - if self.wrapcolumn is not None: - wrap = self.wrapcolumn - left - - msg = self.message - if wrap is not None: - msg = self.__wrapMessage(msg, wrap) - lines = msg.split('\n') - h = 4 + len(lines) - w = 6 + len(max(lines, key = len)) - if w < width: w = width - if h < height: h = height - - rc = '/' + '-' * (w - 2) + '\\\n' - rc += '|' + ' ' * (w - 2) + '|\n' - for line in lines: - rc += '| ' + line + ' ' * (w - len(line) - 6) + ' |\n' - rc += '|' + ' ' * (w - 2) + '|\n' - rc += '\\' + '-' * (w - 2) + '/' - - return rc - - - def __wrapMessage(self, message, wrap): - lines = message.split('\n') - buf = '' - for line in lines: - b = [None] * len(line) - map = {} - (bi, cols, w) = (0, 0, wrap) - (indent, indentc) = (-1, 0) - - (i, n) = (0, len(line)) - while i <= n: - d = None - if i != n: - d = line[i] - i += 1 - if d == '\033': - b[bi] = d - bi += 1 - b[bi] = line[i] - d = line[i] - bi += 1 - i += 1 - if d == '[': - while True: - b[bi] = line[i] - d = line[i] - bi += 1 - i += 1 - if (('a' <= d) and (d <= 'z')) or (('A' <= d) and (d <= 'Z')) or (d == '~'): - break - elif d == ']': - b[bi] = line[i] - d = line[i] - bi += 1 - i += 1 - if d == 'P': - for j in range(0, 7): - b[bi] = line[i] - bi += 1 - i += 1 - elif (d is not None) and (d != ' '): - if indent == -1: - indent = i - 1 - for j in range(0, indent): - if line[j] == ' ': - indentc += 1 - b[bi] = d - bi += 1 - cols += 1 - map[cols] = bi - else: - mm = 0 - while (w > 8) and (cols > w + 3): - mm += w - 1 - m = map[mm] - for bb in b[:m]: - buf += bb - buf += '-\n' - cols -= w - 1 - m += w -1 - bi -= m - bb = b[m:] - for j in range(0, bi): - b[j] = bb[j] - w = wrap - if indent != -1: - buf += line[:indent] - w -= indentc - if cols > w: - buf += '\n' - w = wrap - if indent != -1: - buf += line[:indent] - w -= indentc - for bb in b[:bi]: - buf += bb - w -= cols - cols = 0 - bi = 0 - if d == -1: - i += 1 - else: - if w > 0: - buf += ' ' - w -= 1 - else: - buf += '\n' - w = wrap - if indent != -1: - buf + line[:indent] - w -= indentc - - buf += '\n' - return buf[:-1] - - - -''' -Start the program from ponysay.__init__ if this is the executed file -''' -if __name__ == '__main__': - Ponysay(opts) diff --git a/testpony b/testpony deleted file mode 100644 index 68e385e..0000000 --- a/testpony +++ /dev/null @@ -1,38 +0,0 @@ -indent $balloon15$ - $\$ - $\$ - $\$ - $\$ - $\$ ▄▄ ▄▄ ▄▄ - ▀ ▄ ▄         -   ▄▄▄ ▄▄▄ ▄   ▄▄ ▄▄   - ▄▄ ▄▄▄▄▄▄▄▄▄   ▄▄ ▄▀ ▄▄▄▄ ▄▀ ▄▄ -   ▄▄▄ ▄▄▄ ▄  ▄▄ ▄ ▄▄ ▄▄ ▄ ▄▄▄   - ▄▄  ▄▄▄▄  ▄   ▄ ▄▄ ▄ ▄▀ - ▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄▄ ▄▄▄▄▄   ▄ ▄▄▄ ▄▄▀▀ - ▄▄▄ ▄▄▄▄▄▄ ▄▄▄▄▄▄  ▄▄▄ ▄   ▄▄ ▄ ▄▄▄▄ -  ▄ ▄▄     ▄▄  ▄       ▄▄▄ ▄▄▄▄▄▄▀ - ▀▄▄ ▄ ▄ ▄▄ ▄▄▄▄▄     ▄▄ ▄ ▄▄ - ▀▄▄▄▄ ▄▄▄▄▄▄▄ ▄▄ ▄▄ ▄▄▄      ▄  ▀▀▄▄▄▀ -    ▄▄ ▄▄ ▄▄ ▄    ▄▄▄ ▄▄ - ▄▄ ▄▄ ▄ ▄▄ ▄▄▄▄▄▄▄▄▄▄ ▄▄ ▄▄▄▄▄▄  - ▄▄▄▄▄▄ ▄▄▄▄▄▄▄▄▄▄▄▄ ▄▄▄▄▄▄▄▄▄▄▄▄▄ -  ▄▄ ▄▄ ▄▄▄▄ ▄▄▄▄ ▄▄▄ ▄▄▄▄▄▄▄ ▄ ▄▄▄▄▄▄▄▄▄▄▄▄ - ▀▄  ▄ ▄   ▄▄    ▄  ▄▄ ▄ ▀▄ ▄▄▄▄▄▄ ▄  - ▀▄▄ ▄ ▄ ▄▄ ▄▀▄▄▄▄▄ ▄ ▄▄▄▀  ▄▄ ▄ ▄ ▄  -    ▄▄▄▄▄▄▀  ▄▄  ▄▄▄▄▀ ▄▄▄   ▄▄ ▄ ▄ ▄ ▄  - ▄▄▄ ▄▄▄ ▄▄▄        ▄▄      ▄    - ▄▄ ▄▄▄▄▄▀▀▀▀▀       ▄▄  ▄  ▄ ▄▄ ▄    -  ▄ ▄▄▀        ▀▄   ▀▀▀▀▀▀▀▄     -  ▄           ▀▄   ▄▄▄▄ ▄▄▄▄▄ - ▀▄ ▄    ▄▄     ▀▄   ▄▄▄▄ ▄ ▄▄▄▄▄ -    ▄▄     ▀▄ ▄▄    ▄ ▄ ▄ ▄▄  - ▄▄  ▄▄ ▄▄▄   ▀▄▄ ▄▄ ▀▀▀▄ ▄ ▄ ▄    - ▄▄▄▄▄▄▄▀ ▄▄ ▄▄  ▄▄▄▄ ▀▄▄ ▄ ▄▄▄▄▄▀ -  ▄▄▄▄▀ ▀▀▀▀▀  ▄▄▄▄▀  ▄ ▄ ▄▄▄▄▀ -    ▄▄▄  -       - ▄▄ ▄▀ - ▄▄▄▄▄ ▄▄▄▀ - ▀▀▀▀▀▀▀▀▀▀ - diff --git a/truncater.c b/truncater.c deleted file mode 100644 index d703406..0000000 --- a/truncater.c +++ /dev/null @@ -1,153 +0,0 @@ -/* ponysaytruncater - * Output truncater used by ponysay to stop - * large ponies from being printed badly. - * - * Licensed under WTFPL - * See COPYING for details - */ -#include -#define String char* -#define boolean char -#define true 1 -#define false 0 - -/* Stdin file descriptor ID */ -#define STDIN 0 - -/* The number of columns on the current line */ -static int col = 0; - -/* Escape sequence state */ -static int esc = 0; - -/* Last bytes as written */ -static boolean ok = true; - -void write(char b, int width); -int toInt(String string); - - -/* Mane method! - * The only argument, in addition to the executed file, - * should be the width of the terminal which you get by - * adding `tput cols || echo 0` as an argument. - * - * @param argc The number of startup arguments - * @param argv The startup arguments, the first is the file itself - * - * @author Mattias Andrée, maandree@kth.se - */ -void main(int argc, String* argv) -{ - int width = 0; - if (argc > 1) - width = toInt(*(argv + 1)); - - char b = 0; - - if (width > 15) /* sanity */ - while (read(STDIN, &b, 1)) - write(b, width); - else - while (read(STDIN, &b, 1)) - printf("%c", b); -} - -/* Writes a character to stdout, iff it fits within the terminal - * - * @param b The character (byte) to write - * @param width The width of the terminal - */ -void write(char b, int width) -{ - int i; - char tabstop; - - if (esc == 0) - { - if (b == '\n') - { - if (col >= width) - { - /* Reset background colour */ - write('\e', width); - write('[', width); - write('4', width); - write('9', width); - write('m', width); - } - col = -1; - } - else if (b == '\t') - { - /* Tab to next pos ≡₈ 0 */ - tabstop = 8 - (col & 7); - for (i = 0; i < tabstop; i++) - write(' ', width); - return; /* (!) */ - } - else if (b == '\e') - esc = 1; - } - else if (esc == 1) - { - if (b == '[') esc = 2; /* CSI: CSI ends with a letter, m is for colour */ - else if (b == ']') esc = 3; /* OSI: OSI P is for palette editing in Linux VT */ - else esc = 10; /* Nothing to see here, move along */ - } - else if (esc == 2) - { - if ((('a' <= b) && (b <= 'z')) || (('A' <= b) && (b <= 'Z'))) - esc = 10; - } - else if ((esc == 3) && (b == 'P')) - { - esc = ~0; - } - else if (esc < 0) - { - esc--; - if (esc == ~7) - esc = 10; - } - else - esc = 10; - - if ( - /* Can be printed: - within bounds ∨ - ∨ escape sequence ∨ - ∨ last with printed ∧ not first byte in character */ - (col < width) || - (esc != 0) || - (ok && ((b & 0xC0) == 0x80))) - { - printf("%c", b); - if ((esc == 0) && ((b & 0xC0) != 0x80)) - /* Count up columns of not in escape sequnce and */ - col++; /* the byte is not the first byte in the character */ - ok = true; - } - else - ok = false; - - if (esc == 10) - esc = 0; -} - -/* Converts a string to an integer - * - * @param string The string to convert - * @return The integer represented by the string - */ -int toInt(String string) -{ - int rc = 0; - String str = string; - char c = 0; - - while ((c = *str++) != 0) - rc = (rc << 1) + (rc << 3) - (c & 15); - - return -rc; -} -- cgit