aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorMattias Andrée <maandree@operamail.com>2012-08-20 02:22:12 +0200
committerMattias Andrée <maandree@operamail.com>2012-08-20 02:22:12 +0200
commit639452aa16d51b1b947d03f04c9eabe2b90482b6 (patch)
treea812eb8c2c4f4a9ed97b1e49b72d59e9b2745e02
parent1e0b26af590d35e0a7a325103baa0f6d8310a58c (diff)
downloadponysay-639452aa16d51b1b947d03f04c9eabe2b90482b6.tar.gz
ponysay-639452aa16d51b1b947d03f04c9eabe2b90482b6.tar.bz2
ponysay-639452aa16d51b1b947d03f04c9eabe2b90482b6.zip
cleanup
-rwxr-xr-xconfigure23
-rwxr-xr-xponysay1099
-rwxr-xr-xponysay.py1097
-rw-r--r--testpony38
-rw-r--r--truncater.c153
5 files changed, 1104 insertions, 1306 deletions
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 <stdio.h>
-#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 <code>`tput cols || echo 0`</code> 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;
-}