aboutsummaryrefslogtreecommitdiff
path: root/ponysay.py
diff options
context:
space:
mode:
authorMattias Andrée <maandree@operamail.com>2012-09-29 22:55:59 +0200
committerMattias Andrée <maandree@operamail.com>2012-09-29 22:55:59 +0200
commit04aae4a3aa354485be4b2a148ac2e2ed7bbf7208 (patch)
tree381cecaddcd1f3bc052e060b3308422ae9db5854 /ponysay.py
parent46544c3822418304295f83c3cb66bcaccaf0d723 (diff)
downloadponysay-04aae4a3aa354485be4b2a148ac2e2ed7bbf7208.tar.gz
ponysay-04aae4a3aa354485be4b2a148ac2e2ed7bbf7208.tar.bz2
ponysay-04aae4a3aa354485be4b2a148ac2e2ed7bbf7208.zip
rename file ponysay to ponysay.py
Diffstat (limited to 'ponysay.py')
-rwxr-xr-xponysay.py1779
1 files changed, 1779 insertions, 0 deletions
diff --git a/ponysay.py b/ponysay.py
new file mode 100755
index 0000000..d1571a7
--- /dev/null
+++ b/ponysay.py
@@ -0,0 +1,1779 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+'''
+ponysay.py - Ponysay, cowsay reimplementation for ponies
+Copyright (C) 2012 Erkin Batu Altunbaş et al.
+
+This program is free software. It comes without any warranty, to
+the extent permitted by applicable law. You can redistribute it
+and/or modify it under the terms of the Do What The Fuck You Want
+To Public License, Version 2, as published by Sam Hocevar. See
+http://sam.zoy.org/wtfpl/COPYING for more details.
+
+
+Authors of ponysay.py:
+
+ 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
+
+'''
+
+import os
+import shutil
+import sys
+import random
+from subprocess import Popen, PIPE
+
+
+'''
+The version of ponysay
+'''
+VERSION = 'dev' # this line should not be edited, it is fixed by the build system
+
+
+
+'''
+Hack to enforce UTF-8 in output (in the future, if you see anypony not using utf-8 in programs by default, report them to Princess Celestia so she can banish them to the moon)
+'''
+def print(text = '', end = '\n'):
+ sys.stdout.buffer.write((str(text) + end).encode('utf-8'))
+
+
+'''
+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['-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['-B'] is not None: self.balloonlist()
+ elif args.opts['++onelist'] is not None: self.__extraponies(); self.onelist()
+ elif args.opts['+l'] is not None: self.__extraponies(); self.list()
+ elif args.opts['+L'] is not None: self.__extraponies(); self.linklist()
+ elif args.opts['-A'] is not None: self.list(); self.__extraponies(); self.list()
+ elif args.opts['+A'] is not None: self.linklist(); self.__extraponies(); self.linklist()
+ else:
+ self.__extraponies(args)
+ self.__bestpony(args)
+ self.__ucsremap(args)
+ if args.opts['-q'] is not None: self.quote(args)
+ else: self.print_pony(args)
+
+
+ ##
+ ## Methods that run before the mane methods
+ ##
+
+ '''
+ Use extra ponies
+ '''
+ def __extraponies(self, args = None):
+ if args is None:
+ ponydirs[:] = extraponydirs
+ elif args.opts['-F'] is not None:
+ args.opts['-f'] = args.opts['-F']
+ ponydirs[:] = extraponydirs
+
+
+ '''
+ Use best.pony if nothing else is set
+ '''
+ def __bestpony(self, args):
+ if (args.opts['-f'] is None) or (args.opts['-q'] is None) or (len(args.opts['-q']) == 0):
+ for ponydir in ponydirs:
+ if os.path.isfile(ponydir + 'best.pony') or os.path.islink(ponydir + 'best.pony'):
+ pony = os.path.realpath(ponydir + 'best.pony') # Canonical path
+ args.opts['-f' if args.opts['-q'] is None else '-q'] = [pony]
+ break
+
+
+ '''
+ Apply pony name remapping to args according to UCS settings
+ '''
+ def __ucsremap(self, args):
+ env_ucs = os.environ['PONYSAY_UCS_ME'] if 'PONYSAY_UCS_ME' in os.environ else ''
+ ucs_conf = 0
+ if env_ucs in ('yes', 'y', '1'): ucs_conf = 1
+ elif env_ucs in ('harder', 'h', '2'): ucs_conf = 2
+
+ if ucs_conf == 0:
+ return
+
+ maplines = []
+ for ucsmap in ucsmaps:
+ if os.path.isfile(ucsmap):
+ mapfile = None
+ try:
+ mapfile = open(ucsmap, 'rb')
+ maplines += [line.replace('\n', '') for line in mapfile.read().decode('utf8', 'replace').split('\n')]
+ finally:
+ if mapfile is not None:
+ mapfile.close()
+
+ map = {}
+ stripset = ' \t' # must be string, wtf! and way doesn't python's doc say so
+ for line in maplines:
+ if (len(line) > 0) and not (line[0] == '#'):
+ s = line.index('→')
+ ucs = line[:s] .strip(stripset)
+ ascii = line[s + 1:].strip(stripset)
+ map[ucs] = ascii
+
+ for flag in ('-f', '-q'):
+ if args.opts[flag] is not None:
+ for i in range(0, len(args.opts[flag])):
+ if args.opts[flag][i] in map:
+ args.opts[flag][i] = map[args.opts[flag][i]]
+
+
+ ##
+ ## Auxiliary methods
+ ##
+
+ '''
+ Apply USC:ise pony names according to UCS settings
+ '''
+ def __ucsise(self, ponies, links = None):
+ env_ucs = os.environ['PONYSAY_UCS_ME'] if 'PONYSAY_UCS_ME' in os.environ else ''
+ ucs_conf = 0
+ if env_ucs in ('yes', 'y', '1'): ucs_conf = 1
+ elif env_ucs in ('harder', 'h', '2'): ucs_conf = 2
+
+ if ucs_conf == 0:
+ return
+
+ maplines = []
+ for ucsmap in ucsmaps:
+ if os.path.isfile(ucsmap):
+ mapfile = None
+ try:
+ mapfile = open(ucsmap, 'rb')
+ maplines += [line.replace('\n', '') for line in mapfile.read().decode('utf8', 'replace').split('\n')]
+ finally:
+ if mapfile is not None:
+ mapfile.close()
+
+ map = {}
+ stripset = ' \t' # must be string, wtf! and way doesn't python's doc say so
+ for line in maplines:
+ if (len(line) > 0) and not (line[0] == '#'):
+ s = line.index('→')
+ ucs = line[:s] .strip(stripset)
+ ascii = line[s + 1:].strip(stripset)
+ map[ascii] = ucs
+
+ if ucs_conf == 1:
+ for pony in ponies:
+ if pony in map:
+ ponies.append(map[pony])
+ if links is not None:
+ links[map[pony]] = pony
+ else:
+ for j in range(0, len(ponies)):
+ if ponies[j] in map:
+ ponies[j] = map[ponies[j]]
+
+
+ '''
+ Returns one file with full path, names is filter for names, also accepts filepaths
+ '''
+ def __getponypath(self, names = None):
+ ponies = {}
+
+ for ponydir in ponydirs:
+ for ponyfile in os.listdir(ponydir):
+ pony = ponyfile[:-5]
+ if pony not in ponies:
+ ponies[pony] = ponydir + ponyfile
+
+ if not names == None:
+ for name in names:
+ if os.path.exists(name):
+ ponies[name] = name
+
+ 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 heard of anypony 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(quotedir)]
+ 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
+ ##
+
+ '''
+ Columnise a list and prints it
+ '''
+ def __columnise(self, ponies):
+ termwidth = self.__gettermsize()[1] + 2
+ ponies.sort(key = lambda pony : pony[0])
+ widths = [UCS.dispLen(pony[0]) for pony in ponies]
+ width = max(widths) + 2 # longest pony file name + space between columns
+
+ cols = termwidth // width
+ rows = (len(ponies) + cols - 1) // cols
+ columns = []
+ for c in range(0, cols): columns.append([])
+
+ (y, x) = (0, 0)
+ for j in range(0, len(ponies)):
+ cell = ponies[j][1] + ' ' * (width - widths[j]);
+ columns[x].append(cell)
+ y += 1
+ if y == rows:
+ x += 1
+ y = 0
+
+ diff = rows * cols - len(ponies)
+ if diff > 2:
+ c = cols - 1
+ diff -= 1
+ while diff > 0:
+ columns[c] = columns[c - 1][-diff:] + columns[c]
+ c -= 1
+ columns[c] = columns[c][:-diff]
+ diff -= 1
+ pass
+
+ lines = []
+ for r in range(0, rows):
+ lines.append([])
+ for c in range(0, cols):
+ if r < len(columns[c]):
+ line = lines[r].append(columns[c][r])
+
+ print('\n'.join([''.join(line)[:-2] for line in lines]));
+ print()
+
+
+ '''
+ Lists the available ponies
+ '''
+ def list(self):
+ quoters = self.__quoters()
+
+ for ponydir in ponydirs: # Loop ponydirs
+ _ponies = os.listdir(ponydir)
+ ponies = []
+ for pony in _ponies:
+ if (len(pony) > 5) and (pony[-5:] == '.pony'):
+ ponies.append(pony[:-5])
+ self.__ucsise(ponies)
+
+ if len(ponies) == 0:
+ continue
+ print('\033[1mponies located in ' + ponydir + '\033[21m')
+
+ self.__columnise([(pony, '\033[1m' + pony + '\033[21m' if pony in quoters else pony) for pony in ponies])
+
+
+ '''
+ Lists the available ponies with alternatives inside brackets
+ '''
+ def linklist(self):
+ termsize = self.__gettermsize()
+ quoters = self.__quoters()
+
+ for ponydir in ponydirs: # Loop ponydirs
+ _ponies = os.listdir(ponydir)
+ ponies = []
+ for pony in _ponies:
+ if (len(pony) > 5) and (pony[-5:] == '.pony'):
+ ponies.append(pony[:-5])
+
+ if len(ponies) == 0:
+ continue
+ print('\033[1mponies located in ' + ponydir + '\033[21m')
+
+ pseudolinkmap = {}
+ self.__ucsise(ponies, pseudolinkmap)
+ pairs = []
+ for pony in ponies:
+ if pony in pseudolinkmap:
+ pairs.append((pony, pseudolinkmap[pony] + '.pony'));
+ else:
+ pairs.append((pony, os.path.realpath(ponydir + pony + '.pony') if os.path.islink(ponydir + pony + '.pony') else None))
+
+ ponymap = {}
+ for pair in pairs:
+ if (pair[1] is None) or (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 = {}
+ for pony in ponymap:
+ w = UCS.dispLen(pony)
+ item = '\033[1m' + pony + '\033[21m' if (pony in quoters) else pony
+ syms = ponymap[pony]
+ syms.sort()
+ if len(syms) > 0:
+ w += 2 + len(syms)
+ item += ' ('
+ first = True
+ for sym in syms:
+ w += UCS.dispLen(sym)
+ if first: first = False
+ else: item += ' '
+ item += '\033[1m' + sym + '\033[21m' if (sym in quoters) else sym
+ item += ')'
+ ponies[(item.replace('\033[1m', '').replace('\033[21m', ''), item)] = w
+ if width < w:
+ width = w
+
+ self.__columnise(list(ponies))
+
+
+ '''
+ Lists with all ponies that have quotes and are displayable
+ '''
+ def quoters(self):
+ last = ''
+ ponies = []
+ for pony in self.__quoters():
+ ponies.append(pony)
+ self.__ucsise(ponies)
+ 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 = []
+ for pony in _ponies:
+ if (len(pony) > 5) and (pony[-5:] == '.pony'):
+ ponies.append(pony[:-5])
+ self.__ucsise(ponies)
+ ponies.sort()
+ for pony in ponies:
+ if not pony == last:
+ last = pony
+ print(pony)
+
+
+
+ ##
+ ## Balloon methods
+ ##
+
+ '''
+ Prints a list of all balloons
+ '''
+ def balloonlist(self):
+ termsize = self.__gettermsize()
+ balloonset = set()
+
+ for balloondir in balloondirs:
+ for balloon in os.listdir(balloondir):
+ if isthink and (len(balloon) > 6) and (balloon[-6:] == '.think'):
+ balloon = balloon[:-6]
+ elif (not isthink) and (len(balloon) > 4) and (balloon[-4:] == '.say'):
+ balloon = balloon[:-4]
+ else:
+ continue
+ if balloon not in balloonset:
+ balloonset.add(balloon)
+
+ self.__columnise([(balloon, balloon) for balloon in list(balloonset)])
+
+
+ '''
+ Returns one file with full path, names is filter for style names, also accepts filepaths
+ '''
+ def __getballoonpath(self, names):
+ if names is None:
+ return None
+ balloons = {}
+
+ for balloondir in balloondirs:
+ for balloon in os.listdir(balloondir):
+ balloonfile = balloon
+ if isthink and (len(balloon) > 6) and (balloon[-6:] == '.think'):
+ balloon = balloon[:-6]
+ elif (not isthink) and (len(balloon) > 4) and (balloon[-4:] == '.say'):
+ balloon = balloon[:-4]
+ else:
+ continue
+ if balloon not in balloons:
+ balloons[balloon] = balloondir + balloonfile
+
+ for name in names:
+ if os.path.exists(name):
+ balloons[name] = name
+
+ if names == None:
+ names = list(balloons.keys())
+
+ balloon = names[random.randrange(0, len(names))]
+ if balloon not in balloons:
+ sys.stderr.write('That balloon style %s does not exist\n' % (balloon));
+ exit(1)
+ else:
+ return balloons[balloon]
+
+
+ '''
+ Creates the balloon style object
+ '''
+ def __getballoon(self, balloonfile):
+ if balloonfile is None:
+ if isthink:
+ return Balloon('o', 'o', '( ', ' )', [' _'], ['_'], ['_'], ['_'], ['_ '], ' )', ' )', ' )', ['- '], ['-'], ['-'], ['-'], [' -'], '( ', '( ', '( ')
+ return Balloon('\\', '/', '< ', ' >', [' _'], ['_'], ['_'], ['_'], ['_ '], ' \\', ' |', ' /', ['- '], ['-'], ['-'], ['-'], [' -'], '\\ ', '| ', '/ ')
+
+ map = {}
+ for elem in ('\\', '/', 'ww', 'ee', 'nw', 'nnw', 'n', 'nne', 'ne', 'nee', 'e', 'see', 'se', 'sse', 's', 'ssw', 'sw', 'sww', 'w', 'nww'):
+ map[elem] = []
+
+ balloonstream = None
+ try:
+ balloonstream = open(balloonfile, 'rb')
+ data = [line.replace('\n', '') for line in balloonstream.read().decode('utf8', 'replace').split('\n')]
+ finally:
+ if balloonstream is not None:
+ balloonstream.close()
+
+ last = None
+ for line in data:
+ if len(line) > 0:
+ if line[0] == ':':
+ map[last].append(line[1:])
+ else:
+ last = line[:line.index(':')]
+ value = line[len(last) + 1:]
+ map[last].append(value)
+
+ return Balloon(map['\\'][0], map['/'][0], map['ww'][0], map['ee'][0], map['nw'], map['nnw'], map['n'],
+ map['nne'], map['ne'], map['nee'][0], map['e'][0], map['see'][0], map['se'], map['sse'],
+ map['s'], map['ssw'], map['sw'], map['sww'][0], map['w'][0], map['nww'][0])
+
+
+
+ ##
+ ## Displaying methods
+ ##
+
+ '''
+ Prints the name of the program and the version of the program
+ '''
+ def version(self):
+ print('%s %s' % ('ponysay', VERSION))
+
+
+ '''
+ 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
+
+ if args.opts['-c'] is not None: ## This algorithm should give some result as cowsay's (according to tests)
+ buf = ''
+ last = ' '
+ CHARS = '\t \n'
+ for c in msg:
+ if (c in CHARS) and (last in CHARS):
+ if last == '\n':
+ buf += last
+ last = c
+ else:
+ buf += c
+ last = c
+ msg = buf.strip(CHARS)
+ buf = ''
+ for c in msg:
+ if (c != '\n') or (last != '\n'):
+ buf += c
+ last = c
+ msg = buf.replace('\n', '\n\n')
+
+ pony = self.__getponypath(args.opts['-f'])
+
+ 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])
+
+ pony = self.__kms(pony)
+
+ 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
+
+ balloon = self.__getballoon(self.__getballoonpath(args.opts['-b']))
+
+ backend = Backend(message = msg, ponyfile = pony, wrapcolumn = messagewrap if messagewrap is not None else 40, width = widthtruncation, balloon = balloon)
+ backend.parse()
+ output = backend.output
+ 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 linuxvt or (env_height is ('yes', 'y', '1')):
+ if env_bottom is ('yes', 'y', '1'):
+ for line in output.split('\n')[: -lines]:
+ print(line)
+ else:
+ for line in output.split('\n')[: 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 = {}
+ for pony in args.opts['-q']:
+ if (len(pony) > 5) and (pony[-5:] == '.pony'):
+ ponyname = pony[:-5]
+ if '/' in ponyname:
+ ponyname = ponyname[ponyname.rindex('/') + 1:]
+ ponyset[ponyname] = pony
+ else:
+ ponyset[pony] = pony
+ alts = []
+ for pair in pairs:
+ if pair[0] in ponyset:
+ alts.append((ponyset[pair[0]], pair[1]))
+ pairs = alts
+
+ if not len(pairs) == 0:
+ pair = pairs[random.randrange(0, len(pairs))]
+ qfile = None
+ try:
+ qfile = open(pair[1], 'rb')
+ args.message = qfile.read().decode('utf8', 'replace').strip()
+ finally:
+ if qfile is not None:
+ qfile.close()
+ args.opts['-f'] = [pair[0]]
+ elif len(args.opts['-q']) == 0:
+ sys.stderr.write('Princess Celestia! All the ponies are mute!\n')
+ exit(1)
+ else:
+ args.opts['-f'] = [args.opts['-q'][random.randrange(0, len(args.opts['-q']))]]
+ args.message = 'Zecora! Help me, I am mute!'
+
+ self.print_pony(args)
+
+
+ '''
+ Indentifies whether KMS support is utilised
+ '''
+ @staticmethod
+ def isUsingKMS():
+ if not linuxvt:
+ return False
+
+ 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]
+
+ return env_kms != ''
+
+
+ '''
+ 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
+
+ KMS_VERSION = '1'
+
+ 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', '')
+
+ cachedir = '/var/cache/ponysay'
+ if not os.path.isdir(cachedir):
+ cachedir = HOME + '/.cache/ponysay'
+ if not os.path.isdir(cachedir):
+ os.makedirs(cachedir)
+
+ newversion = False
+ if not os.path.isfile(cachedir + '/.version'):
+ newversion = True
+ else:
+ cachev = None
+ try:
+ cachev = open(cachedir + '/.version', 'rb')
+ if cachev.read().decode('utf8', 'replace').replace('\n', '') == KMS_VERSION:
+ newversion = True
+ finally:
+ if cachev is not None:
+ cachev.close()
+ if newversion:
+ for cached in os.listdir(cachedir):
+ cached = cachedir + '/' + cached
+ if os.path.isdir(cached) and not os.path.islink(cached):
+ shutil.rmtree(cached, False)
+ else:
+ os.remove(cached)
+ cachev = None
+ try:
+ cachev = open(cachedir + '/.version', 'w+')
+ cachev.write(KMS_VERSION)
+ finally:
+ if cachev is not None:
+ cachev.close()
+
+ kmsponies = cachedir + '/kmsponies/' + palettefile
+ kmspony = (kmsponies + pony).replace('//', '/')
+
+ if not os.path.isfile(kmspony):
+ protokmsponies = cachedir + '/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 -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] in ('-', '+')):
+ if (len(arg) > 2) and (arg[:2] in ('--', '++')):
+ 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('\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')
+ (lines, lens) = ([], [])
+ 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
+ (line, l) = ('', 0)
+ first = opt_alts[0]
+ last = opt_alts[-1]
+ alts = ('', last) if first is last else (first, last)
+ for opt_alt in alts:
+ if opt_alt is alts[-1]:
+ line += '%colour%' + opt_alt
+ l += len(opt_alt)
+ if opt_type == ARGUMENTED: line += ' \033[4m%s\033[24m' % (opt_arg); l += len(opt_arg) + 1
+ elif opt_type == VARIADIC: line += ' [\033[4m%s\033[24m...]' % (opt_arg); l += len(opt_arg) + 6
+ else:
+ line += ' \033[2m%s\033[22m ' % (opt_alt)
+ l += len(opt_alt) + 6
+ lines.append(line)
+ lens.append(l)
+
+ col = max(lens)
+ col += 8 - ((col - 4) & 7)
+ index = 0
+ for opt in self.__arguments:
+ opt_help = opt[3]
+ if opt_help is None:
+ continue
+ first = True
+ colour = '36' if (index & 1) == 0 else '34'
+ print(lines[index].replace('%colour%', '\033[%s;1m' % (colour)), end=' ' * (col - lens[index]))
+ for line in opt_help.split('\n'):
+ if first:
+ first = False
+ print('%s' % (line), end='\033[21;39m\n')
+ else:
+ print('%s\033[%sm%s\033[39m' % (' ' * col, colour, line))
+ index += 1
+
+ print()
+
+
+
+'''
+Balloon format class
+'''
+class Balloon():
+ '''
+ Constructor
+ '''
+ def __init__(self, link, linkmirror, ww, ee, nw, nnw, n, nne, ne, nee, e, see, se, sse, s, ssw, sw, sww, w, nww):
+ (self.link, self.linkmirror) = (link, linkmirror)
+ (self.ww, self.ee) = (ww, ee)
+ (self.nw, self.ne, self.se, self.sw) = (nw, ne, se, sw)
+ (self.nnw, self.n, self.nne) = (nnw, n, nne)
+ (self.nee, self.e, self.see) = (nee, e, see)
+ (self.sse, self.s, self.ssw) = (sse, s, ssw)
+ (self.sww, self.w, self.nww) = (sww, w, nww)
+
+ _ne = max(ne, key = UCS.dispLen)
+ _nw = max(nw, key = UCS.dispLen)
+ _se = max(se, key = UCS.dispLen)
+ _sw = max(sw, key = UCS.dispLen)
+
+ minE = UCS.dispLen(max([_ne, nee, e, see, _se, ee], key = UCS.dispLen))
+ minW = UCS.dispLen(max([_nw, nww, e, sww, _sw, ww], key = UCS.dispLen))
+ minN = len(max([ne, nne, n, nnw, nw], key = len))
+ minS = len(max([se, sse, s, ssw, sw], key = len))
+
+ self.minwidth = minE + minE
+ self.minheight = minN + minS
+
+
+ '''
+ Generates a balloon with a message
+ '''
+ def get(self, minw, minh, lines, lencalc):
+ h = self.minheight + len(lines)
+ w = self.minwidth + lencalc(max(lines, key = lencalc))
+ if w < minw: w = minw
+ if h < minh: h = minh
+
+ if len(lines) > 1:
+ (ws, es) = ({0 : self.nww, len(lines) - 1 : self.sww}, {0 : self.nee, len(lines) - 1 : self.see})
+ for j in range(1, len(lines) - 1):
+ ws[j] = self.w
+ es[j] = self.e
+ else:
+ (ws, es) = ({0 : self.ww}, {0 : self.ee})
+
+ rc = []
+
+ for j in range(0, len(self.n)):
+ outer = UCS.dispLen(self.nw[j]) + UCS.dispLen(self.ne[j])
+ inner = UCS.dispLen(self.nnw[j]) + UCS.dispLen(self.nne[j])
+ if outer + inner >= w:
+ rc.append(self.nw[j] + self.nnw[j] + self.n[j] * (w - outer - inner) + self.nne[j] + self.ne[j])
+ else:
+ rc.append(self.nw[j] + self.n[j] * (w - outer) + self.ne[j])
+
+ for j in range(0, len(lines)):
+ rc.append(ws[j] + lines[j] + ' ' * (w - lencalc(lines[j]) - UCS.dispLen(self.w) - UCS.dispLen(self.e)) + es[j])
+
+ for j in range(0, len(self.s)):
+ outer = UCS.dispLen(self.sw[j]) + UCS.dispLen(self.se[j])
+ inner = UCS.dispLen(self.ssw[j]) + UCS.dispLen(self.sse[j])
+ if outer + inner >= w:
+ rc.append(self.sw[j] + self.ssw[j] + self.s[j] * (w - outer - inner) + self.sse[j] + self.se[j])
+ else:
+ rc.append(self.sw[j] + self.s[j] * (w - outer) + self.se[j])
+
+ return '\n'.join(rc)
+
+
+
+'''
+Replacement for cowsay
+'''
+class Backend():
+ '''
+ Constructor
+ Takes message [string], ponyfile [filename string], wrapcolumn [None or an int], width [None or an int] and balloon [Balloon object]
+ '''
+ def __init__(self, message, ponyfile, wrapcolumn, width, balloon):
+ self.message = message
+ self.ponyfile = ponyfile
+ self.wrapcolumn = wrapcolumn
+ self.width = width
+ self.balloon = balloon
+
+ self.link = {'\\' : self.balloon.link, '/' : self.balloon.linkmirror}
+
+ self.output = ''
+ self.pony = None
+
+
+ '''
+ Process all data
+ '''
+ def parse(self):
+ self.__expandMessage()
+ self.__loadFile()
+ self.__processPony()
+ self.__truncate()
+
+
+ '''
+ Converts all tabs in the message to spaces by expanding
+ '''
+ 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
+ if not UCS.isCombining(c):
+ x += 1
+ buf += '\n'
+ self.message = buf[:-1]
+
+
+ '''
+ Loads the pony file
+ '''
+ def __loadFile(self):
+ ponystream = None
+ try:
+ ponystream = open(self.ponyfile, 'rb')
+ self.pony = ponystream.read().decode('utf8', 'replace')
+ finally:
+ if ponystream is not None:
+ ponystream.close()
+
+
+ '''
+ Truncate output to the width of the screen
+ '''
+ 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
+ if not UCS.isCombining(c):
+ x += 1
+ self.output += '\n'
+ self.output = self.output[:-1]
+
+
+ '''
+ Process the pony file and generate output to self.output
+ '''
+ def __processPony(self):
+ self.output = ''
+
+ AUTO_PUSH = '\033[01010~'
+ AUTO_POP = '\033[10101~'
+
+ variables = {'' : '$'}
+ for key in self.link:
+ variables[key] = AUTO_PUSH + self.link[key] + AUTO_POP
+
+ indent = 0
+ dollar = None
+ balloonLines = None
+ colourstack = ColourStack(AUTO_PUSH, AUTO_POP)
+
+ (i, n, lineindex, skip, nonskip) = (0, len(self.pony), 0, 0, 0)
+ 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'):
+ data = variables[dollar].replace('$', '$$')
+ if data == '$$': # if not handled specially we will get an infinity loop
+ if (skip == 0) or (nonskip > 0):
+ if nonskip > 0:
+ nonskip -= 1
+ self.output += '$'
+ indent += 1
+ else:
+ skip -= 1
+ else:
+ n += len(data)
+ self.pony = self.pony[:i] + data + self.pony[i:]
+ 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)
+ balloon = balloon.split('\n')
+ for b in balloon[0]:
+ self.output += b + colourstack.feed(b)
+ if lineindex == 0:
+ balloonpre = '\n' + (' ' * indent)
+ for line in balloon[1:]:
+ self.output += balloonpre;
+ for b in line:
+ self.output += b + colourstack.feed(b);
+ indent = 0
+ elif len(balloon) > 1:
+ balloonLines = balloon
+ balloonLine = 0
+ balloonIndent = indent
+ indent += self.__len(balloonLines[0])
+ balloonLines[0] = None
+ 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)
+ for b in colour:
+ self.output += b + colourstack.feed(b);
+ i += len(colour) - 1
+ elif c == '\n':
+ self.output += c
+ indent = 0
+ (skip, nonskip) = (0, 0)
+ lineindex += 1
+ if balloonLines is not None:
+ balloonLine += 1
+ if balloonLine == len(balloonLines):
+ balloonLines = None
+ else:
+ if (balloonLines is not None) and (balloonLines[balloonLine] is not None) and (balloonIndent == indent):
+ data = balloonLines[balloonLine]
+ datalen = self.__len(data)
+ skip += datalen
+ nonskip += datalen
+ data = data.replace('$', '$$')
+ n += len(data)
+ self.pony = self.pony[:i] + data + self.pony[i:]
+ balloonLines[balloonLine] = None
+ else:
+ if (skip == 0) or (nonskip > 0):
+ if nonskip > 0:
+ nonskip -= 1
+ self.output += c + colourstack.feed(c);
+ if not UCS.isCombining(c):
+ indent += 1
+ else:
+ skip -= 1
+
+ if balloonLines is not None:
+ for line in balloonLines[balloonLine:]:
+ data = ' ' * (balloonIndent - indent) + line + '\n'
+ for b in data:
+ self.output += b + colourstack.feed(b);
+ indent = 0
+
+ self.output = self.output.replace(AUTO_PUSH, '').replace(AUTO_POP, '')
+
+
+ '''
+ Gets colour code att the currect offset in a buffer
+ '''
+ 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
+
+
+ '''
+ Calculates the number of visible characters in a text
+ '''
+ def __len(self, input):
+ (rc, i, n) = (0, 0, len(input))
+ while i < n:
+ c = input[i]
+ if c == '\033':
+ i += len(self.__getcolour(input, i))
+ else:
+ i += 1
+ if not UCS.isCombining(c):
+ rc += 1
+ return rc
+
+
+ '''
+ Generates a balloon with the message
+ '''
+ def __getballoon(self, width, height, left):
+ wrap = None
+ if self.wrapcolumn is not None:
+ wrap = self.wrapcolumn - left
+ if wrap < 8:
+ wrap = 8
+
+ msg = self.message
+ if wrap is not None:
+ msg = self.__wrapMessage(msg, wrap)
+
+ if '\033' in msg:
+ AUTO_PUSH = '\033[01010~'
+ AUTO_POP = '\033[10101~'
+ cstack = ColourStack(AUTO_PUSH, AUTO_POP)
+ buf = ''
+ for c in msg:
+ if c == '\n':
+ for cc in ('%s\n%s' % (AUTO_PUSH, AUTO_POP)):
+ buf += cc
+ buf += cstack.feed(cc)
+ else:
+ buf += c
+ buf += cstack.feed(c)
+ msg = buf
+
+ return self.balloon.get(width, height, msg.split('\n'), self.__len)
+
+
+ '''
+ Wraps the message
+ '''
+ 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': # TODO this should use self.__getcolour()
+ 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
+ if not UCS.isCombining(d):
+ 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 '\n'.join(line.rstrip() for line in buf[:-1].split('\n'))
+
+
+'''
+ANSI colour stack
+'''
+class ColourStack():
+ '''
+ Constructor
+ '''
+ def __init__(self, autopush, autopop):
+ self.autopush = autopush
+ self.autopop = autopop
+ self.lenpush = len(autopush)
+ self.lenpop = len(autopop)
+ self.bufproto = ' ' * (self.lenpush if self.lenpush > self.lenpop else self.lenpop)
+ self.stack = []
+ self.push()
+ self.seq = None
+
+
+ def push(self):
+ self.stack = [[self.bufproto, None, None, [False] * 9]] + self.stack
+ if len(self.stack) == 1:
+ return None
+ return '\033[0m'
+
+
+ def pop(self):
+ old = self.stack[0]
+ self.stack = self.stack[1:]
+ rc = '\033[0;'
+ new = self.stack[0]
+ if new[1] is not None: rc += new[1] + ';'
+ if new[2] is not None: rc += new[2] + ';'
+ for i in range(0, 9):
+ if new[3][i]:
+ rc += str(i + 1) + ';'
+ return rc[:-1] + 'm'
+
+
+ def feed(self, char):
+ if self.seq is not None:
+ self.seq += char
+ if (char == '~') or (('a' <= char) and (char <= 'z')) or (('A' <= char) and (char <= 'Z')):
+ if (self.seq[0] == '[') and (self.seq[-1] == 'm'):
+ self.seq = self.seq[1:-1].split(';')
+ (i, n) = (0, len(self.seq))
+ while i < n:
+ part = self.seq[i]
+ p = 0 if part == '' else int(part)
+ i += 1
+ if p == 0: self.stack[0][1:] = [None, None, [False] * 9]
+ elif (1 <= p) and (p <= 9): self.stack[0][3][p - 1] = True
+ elif (21 <= p) and (p <= 29): self.stack[0][3][p - 21] = False
+ elif p == 39: self.stack[0][1] = None
+ elif p == 49: self.stack[0][2] = None
+ elif (30 <= p) and (p <= 37): self.stack[0][1] = part
+ elif (90 <= p) and (p <= 97): self.stack[0][1] = part
+ elif (40 <= p) and (p <= 47): self.stack[0][2] = part
+ elif (100 <= p) and (p <= 107): self.stack[0][2] = part
+ elif p == 38:
+ self.stack[0][1] = '%s;%s;%s' % (part, self.seq[i], self.seq[i + 1])
+ i += 2
+ elif p == 48:
+ self.stack[0][2] = '%s;%s;%s' % (part, self.seq[i], self.seq[i + 1])
+ i += 2
+ self.seq = None
+ elif char == '\033':
+ self.seq = ''
+ buf = self.stack[0][0]
+ buf = buf[1:] + char
+ rc = ''
+ if buf[-self.lenpush:] == self.autopush: rc = self.push()
+ elif buf[-self.lenpop:] == self.autopop: rc = self.pop()
+ self.stack[0][0] = buf
+ return rc
+
+
+
+'''
+UCS utility class
+'''
+class UCS():
+ '''
+ Checks whether a character is a combining character
+ '''
+ @staticmethod
+ def isCombining(char):
+ o = ord(char)
+ if (0x0300 <= o) and (o <= 0x036F): return True
+ if (0x20D0 <= o) and (o <= 0x20FF): return True
+ if (0x1DC0 <= o) and (o <= 0x1DFF): return True
+ if (0xFE20 <= o) and (o <= 0xFE2F): return True
+ return False
+
+
+ '''
+ Gets the number of combining characters in a string
+ '''
+ @staticmethod
+ def countCombining(string):
+ rc = 0
+ for char in string:
+ if UCS.isCombining(char):
+ rc += 1
+ return rc
+
+
+ '''
+ Gets length of a string not counting combining characters
+ '''
+ @staticmethod
+ def dispLen(string):
+ return len(string) - UCS.countCombining(string)
+
+
+
+'''
+The user's home directory
+'''
+HOME = os.environ['HOME'] if 'HOME' in os.environ else os.path.expanduser('~')
+
+
+'''
+Whether the program is execute in Linux VT (TTY)
+'''
+linuxvt = ('TERM' in os.environ) and (os.environ['TERM'] == 'linux')
+
+
+'''
+Whether the script is executed as ponythink
+'''
+isthink = (len(__file__) >= 5) and (__file__[-5:] == 'think')
+isthink = ((len(__file__) >= 8) and (__file__[-8:] == 'think.py')) or isthink
+
+
+'''
+Whether stdin is piped
+'''
+pipelinein = not sys.stdin.isatty()
+
+'''
+Whether stdout is piped
+'''
+pipelineout = not sys.stdout.isatty()
+
+'''
+Whether stderr is piped
+'''
+pipelineerr = not sys.stderr.isatty()
+
+
+'''
+Whether KMS is used
+'''
+usekms = Ponysay.isUsingKMS()
+
+
+'''
+The directories where pony files are stored, ttyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS
+'''
+appendset = set()
+ponydirs = []
+if linuxvt and not usekms: _ponydirs = [HOME + '/.local/share/ponysay/ttyponies/', '/usr/share/ponysay/ttyponies/']
+else: _ponydirs = [HOME + '/.local/share/ponysay/ponies/', '/usr/share/ponysay/ponies/']
+for ponydir in _ponydirs:
+ if os.path.isdir(ponydir) and (ponydir not in appendset):
+ ponydirs.append(ponydir)
+ appendset.add(ponydir)
+
+
+'''
+The directories where pony files are stored, extrattyponies/ are used if the terminal is Linux VT (also known as TTY) and not with KMS
+'''
+appendsetset = set()
+extraponydirs = []
+if linuxvt and not usekms: _extraponydirs = [HOME + '/.local/share/ponysay/extrattyponies/', '/usr/share/ponysay/extrattyponies/']
+else: _extraponydirs = [HOME + '/.local/share/ponysay/extraponies/', '/usr/share/ponysay/extraponies/']
+for extraponydir in _extraponydirs:
+ if os.path.isdir(extraponydir) and (extraponydir not in appendset):
+ extraponydirs.append(extraponydir)
+ appendset.add(extraponydir)
+
+
+'''
+The directories where quotes files are stored
+'''
+appendset = set()
+quotedirs = []
+_quotedirs = [HOME + '/.local/share/ponysay/quotes/', '/usr/share/ponysay/quotes/']
+for quotedir in _quotedirs:
+ if os.path.isdir(quotedir) and (quotedir not in appendset):
+ quotedirs.append(quotedir)
+ appendset.add(quotedir)
+
+
+'''
+The directories where balloon style files are stored
+'''
+appendset = set()
+balloondirs = []
+_balloondirs = [HOME + '/.local/share/ponysay/balloons/', '/usr/share/ponysay/balloons/']
+for balloondir in _balloondirs:
+ if os.path.isdir(balloondir) and (balloondir not in appendset):
+ balloondirs.append(balloondir)
+ appendset.add(balloondir)
+
+
+'''
+ucsmap files
+'''
+appendset = set()
+ucsmaps = []
+_ucsmaps = [HOME + '/.local/share/ponysay/ucsmap', '/usr/share/ponysay/ucsmap']
+for ucsmap in _ucsmaps:
+ if os.path.isdir(ucsmap) and (ucsmap not in appendset):
+ ucsmaps.append(ucsmap)
+ appendset.add(ucsmap)
+
+
+
+usage_saythink = '\033[34;1m(ponysay | ponythink)\033[21;39m'
+usage_common = '[-c] [-W\033[4mCOLUMN\033[24m] [-b\033[4mSTYLE\033[24m]'
+usage_listhelp = '(-l | -L | -B | +l | +L | -v | -h)'
+usage_file = '[-f\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]'
+usage_xfile = '[-F\033[4mPONY\033[24m]* [[--] \033[4mmessage\033[24m]'
+usage_quote = '-q [\033[4mPONY\033[24m*]'
+
+usage = '%s %s\n%s %s %s\n%s %s %s\n%s %s %s' % (usage_saythink, usage_listhelp,
+ usage_saythink, usage_common, usage_file,
+ usage_saythink, usage_common, usage_xfile,
+ usage_saythink, usage_common, 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 reimplemention 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(['++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 names.')
+opts.add_argumentless(['-L', '--symlist', '--altlist'], help = 'List pony names with alternatives.')
+opts.add_argumentless(['+l', '++list'], help = 'List non-MLP:FiM pony names.')
+opts.add_argumentless(['+L', '++symlist', '++altlist'], help = 'List non-MLP:FiM pony names with alternatives.')
+opts.add_argumentless(['-A', '--all'], help = 'List MLP:FIM and non-MLP:FiM pony names.')
+opts.add_argumentless(['+A', '++all'], help = 'List MLP:FIM and non-MLP:FiM pony names with alternatives.')
+opts.add_argumentless(['-B', '--bubblelist', '--balloonlist'], help = 'List balloon styles.')
+opts.add_argumentless(['-c', '--compact'], help = 'Compress messages.')
+opts.add_argumented( ['-W', '--wrap'], arg = 'COLUMN', help = 'Specify column where the message should be wrapped.')
+opts.add_argumented( ['-b', '--bubble', '--balloon'], arg = 'STYLE', help = 'Select a balloon style.')
+opts.add_argumented( ['-f', '--file', '--pony'], arg = 'PONY', help = 'Select a pony.\nEither a file name or a pony name.')
+opts.add_argumented( ['-F', '++file', '++pony'], arg = 'PONY', help = 'Select a non-MLP:FiM pony.')
+opts.add_variadic( ['-q', '--quote'], arg = 'PONY', help = 'Select a ponies which will quote themself.')
+
+opts.parse()
+
+
+
+'''
+Start the program from ponysay.__init__ if this is the executed file
+'''
+if __name__ == '__main__':
+ Ponysay(opts)