mirror of
https://github.com/lalbornoz/roar.git
synced 2024-11-14 03:26:53 +00:00
211 lines
11 KiB
Python
211 lines
11 KiB
Python
#!/usr/bin/env python3
|
|
#
|
|
# CanvasExportStore.py
|
|
# Copyright (c) 2018, 2019 Lucio Andrés Illanes Albornoz <lucio@lucioillanes.de>
|
|
#
|
|
|
|
from CanvasColours import ColourMapBold, ColourMapNormal, MiRCARTToAnsiColours
|
|
import io, os, tempfile
|
|
|
|
try:
|
|
from PIL import Image, ImageDraw, ImageFont
|
|
havePIL = True
|
|
except ImportError:
|
|
havePIL = False
|
|
|
|
try:
|
|
import base64, json, requests, urllib.request
|
|
haveUrllib = True
|
|
except ImportError:
|
|
haveUrllib = False
|
|
|
|
class CanvasExportStore():
|
|
# {{{ _CellState(): Cell state
|
|
class _CellState():
|
|
CS_NONE = 0x00
|
|
CS_BOLD = 0x01
|
|
CS_ITALIC = 0x02
|
|
CS_UNDERLINE = 0x04
|
|
# }}}
|
|
ImgurUploadUrl = "https://api.imgur.com/3/upload.json"
|
|
PastebinPostUrl = "https://pastebin.com/api/api_post.php"
|
|
|
|
# {{{ _drawUnderline(self, curPos, fillColour, fontSize, imgDraw)
|
|
def _drawUnderLine(self, curPos, fillColour, fontSize, imgDraw):
|
|
imgDraw.line( \
|
|
xy=(curPos[0], curPos[1] + (fontSize[1] - 2), \
|
|
curPos[0] + fontSize[0], curPos[1] + (fontSize[1] - 2)), \
|
|
fill=fillColour)
|
|
# }}}
|
|
|
|
# {{{ exportAnsiFile(self, canvasMap, canvasSize, outFile)
|
|
def exportAnsiFile(self, canvasMap, canvasSize, outFile):
|
|
outBuffer = ""
|
|
for inCurRow in range(len(canvasMap)):
|
|
lastAttribs, lastColours = self._CellState.CS_NONE, None
|
|
for inCurCol in range(len(canvasMap[inCurRow])):
|
|
inCurCell = canvasMap[inCurRow][inCurCol]
|
|
if lastAttribs != inCurCell[2]:
|
|
if inCurCell[2] & self._CellState.CS_BOLD:
|
|
outBuffer += "\u001b[1m"
|
|
if inCurCell[2] & self._CellState.CS_UNDERLINE:
|
|
outBuffer += "\u001b[4m"
|
|
lastAttribs = inCurCell[2]
|
|
if lastColours == None or lastColours != inCurCell[:2]:
|
|
if (inCurCell[0] == -1) \
|
|
and (inCurCell[1] == -1):
|
|
outBuffer += "\u001b[39;49m{}".format(" ")
|
|
elif inCurCell[1] == -1:
|
|
ansiFg = MiRCARTToAnsiColours[inCurCell[0]]
|
|
outBuffer += "\u001b[49;{:02d}m{}".format(ansiFg, inCurCell[3])
|
|
else:
|
|
ansiBg = MiRCARTToAnsiColours[inCurCell[1]] + 10
|
|
ansiFg = MiRCARTToAnsiColours[inCurCell[0] if inCurCell[0] != -1 else inCurCell[1]]
|
|
outBuffer += "\u001b[{:02d};{:02d}m{}".format(ansiBg, ansiFg, inCurCell[3])
|
|
lastColours = inCurCell[:2]
|
|
else:
|
|
outBuffer += inCurCell[3]
|
|
outBuffer += "\u001b[0m\n"
|
|
if len(outBuffer):
|
|
outFile.write(outBuffer)
|
|
return (True, None)
|
|
else:
|
|
return (False, "empty buffer generated")
|
|
# }}}
|
|
# {{{ exportBitmapToImgur(self, apiKey, canvasBitmap, imgName, imgTitle, imgType)
|
|
def exportBitmapToImgur(self, apiKey, canvasBitmap, imgName, imgTitle, imgType):
|
|
tmpPathName = tempfile.mkstemp()
|
|
os.close(tmpPathName[0])
|
|
canvasBitmap.ConvertToImage().SaveFile(tmpPathName[1], imgType)
|
|
with open(tmpPathName[1], "rb") as requestImage:
|
|
requestImageData = requestImage.read()
|
|
requestData = { \
|
|
"image": base64.b64encode(requestImageData), \
|
|
"key": apiKey, \
|
|
"name": imgName, \
|
|
"title": imgTitle, \
|
|
"type": "base64"}
|
|
requestHeaders = {"Authorization": "Client-ID " + apiKey}
|
|
responseHttp = requests.post(self.ImgurUploadUrl, data=requestData, headers=requestHeaders)
|
|
responseDict = json.loads(responseHttp.text)
|
|
if responseHttp.status_code == 200:
|
|
imgurResult = (True, responseHttp.status_code, responseDict.get("data").get("link"))
|
|
else:
|
|
imgurResult = (False, responseHttp.status_code, responseDict.get("data"))
|
|
os.remove(tmpPathName[1])
|
|
return imgurResult
|
|
# }}}
|
|
# {{{ exportBitmapToPngFile(self, canvasBitmap, outPathName, outType)
|
|
def exportBitmapToPngFile(self, canvasBitmap, outPathName, outType):
|
|
return canvasBitmap.ConvertToImage().SaveFile(outPathName, outType)
|
|
# }}}
|
|
# {{{ exportPastebin(self, apiDevKey, canvasMap, canvasSize, pasteName="", pastePrivate=0)
|
|
def exportPastebin(self, apiDevKey, canvasMap, canvasSize, pasteName="", pastePrivate=0):
|
|
if haveUrllib:
|
|
outFile = io.StringIO()
|
|
self.exportTextFile(canvasMap, canvasSize, outFile)
|
|
requestData = { \
|
|
"api_dev_key": apiDevKey, \
|
|
"api_option": "paste", \
|
|
"api_paste_code": outFile.getvalue().encode(), \
|
|
"api_paste_name": pasteName, \
|
|
"api_paste_private": pastePrivate}
|
|
responseHttp = requests.post(self.PastebinPostUrl, data=requestData)
|
|
if responseHttp.status_code == 200:
|
|
return (True if responseHttp.text.startswith("http") else False, responseHttp.text)
|
|
else:
|
|
return (False, str(responseHttp.status_code))
|
|
else:
|
|
return (False, "missing requests and/or urllib3 module(s)")
|
|
# }}}
|
|
# {{{ exportPngFile(self, canvasMap, fontFilePath, fontSize, outPathName)
|
|
def exportPngFile(self, canvasMap, fontFilePath, fontSize, outPathName):
|
|
if havePIL:
|
|
inSize = (len(canvasMap[0]), len(canvasMap))
|
|
outCurPos = [0, 0]
|
|
outFontFilePath, outFontSize = fontFilePath, fontSize
|
|
outImgFont = ImageFont.truetype(outFontFilePath, outFontSize)
|
|
outImgFontSize = [*outImgFont.getsize(" ")]; outImgFontSize[1] += 3;
|
|
outSize = [a * b for a, b in zip(inSize, outImgFontSize)]
|
|
outImg = Image.new("RGBA", outSize, (*ColourMapNormal[1], 255))
|
|
outImgDraw = ImageDraw.Draw(outImg); outImgDraw.fontmode = "1";
|
|
for inCurRow in range(len(canvasMap)):
|
|
for inCurCol in range(len(canvasMap[inCurRow])):
|
|
inCurCell = canvasMap[inCurRow][inCurCol]; outColours = [0, 0];
|
|
if inCurCell[2] & self._CellState.CS_BOLD:
|
|
if inCurCell[3] != " ":
|
|
if inCurCell[3] == "█":
|
|
outColours[1] = ColourMapNormal[inCurCell[0]]
|
|
else:
|
|
outColours = (ColourMapBold[inCurCell[0]], ColourMapNormal[inCurCell[1]])
|
|
else:
|
|
outColours[1] = ColourMapNormal[inCurCell[1]]
|
|
else:
|
|
if inCurCell[3] != " ":
|
|
if inCurCell[3] == "█":
|
|
outColours[1] = ColourMapNormal[inCurCell[0]]
|
|
else:
|
|
outColours = (ColourMapNormal[inCurCell[0]], ColourMapNormal[inCurCell[1]])
|
|
else:
|
|
outColours[1] = ColourMapNormal[inCurCell[1]]
|
|
outImgDraw.rectangle((*outCurPos, outCurPos[0] + outImgFontSize[0], outCurPos[1] + outImgFontSize[1]), fill=(*outColours[1], 255))
|
|
if not inCurCell[3] in " █" \
|
|
and outColours[0] != outColours[1]:
|
|
outImgDraw.text(outCurPos, inCurCell[3], (*outColours[0], 255), outImgFont)
|
|
if inCurCell[2] & self._CellState.CS_UNDERLINE:
|
|
outColours[0] = ColourMapNormal[inCurCell[0]]
|
|
self._drawUnderLine(outCurPos, (*outColours[0], 255), outImgFontSize, outImgDraw)
|
|
outCurPos[0] += outImgFontSize[0];
|
|
outCurPos[0] = 0; outCurPos[1] += outImgFontSize[1];
|
|
outImg.save(outPathName);
|
|
return (True, None)
|
|
else:
|
|
return (False, "missing PIL modules")
|
|
# }}}
|
|
# {{{ exportTextBuffer(self, canvasMap, canvasSize)
|
|
def exportTextBuffer(self, canvasMap, canvasSize):
|
|
outBuffer = ""
|
|
for canvasRow in range(canvasSize[1]):
|
|
canvasLastColours = [15, -1]
|
|
for canvasCol in range(canvasSize[0]):
|
|
canvasColColours = canvasMap[canvasRow][canvasCol][0:2]
|
|
canvasColText = canvasMap[canvasRow][canvasCol][3]
|
|
if canvasColColours[0] == -1:
|
|
canvasColColours[0] = canvasColColours[1]
|
|
if (canvasColColours[0] != canvasLastColours[0]) \
|
|
and (canvasColColours[1] != canvasLastColours[1]):
|
|
if (canvasColColours[0] == -1) \
|
|
and (canvasColColours[1] == -1):
|
|
outBuffer += "\u000f"
|
|
elif canvasColColours[1] == -1:
|
|
outBuffer += "\u0003\u0003{}".format(canvasColColours[0])
|
|
else:
|
|
outBuffer += "\u0003{},{}".format(canvasColColours[0], canvasColColours[1])
|
|
canvasLastColours = canvasColColours
|
|
elif canvasColColours[1] != canvasLastColours[1]:
|
|
if canvasColColours[1] == -1:
|
|
outBuffer += "\u0003\u0003{}".format(canvasLastColours[0])
|
|
else:
|
|
outBuffer += "\u0003{},{}".format(canvasLastColours[0], canvasColColours[1])
|
|
canvasLastColours[1] = canvasColColours[1]
|
|
elif canvasColColours[0] != canvasLastColours[0]:
|
|
if canvasColColours[0] == -1:
|
|
outBuffer += "\u000f"
|
|
else:
|
|
outBuffer += "\u0003{}".format(canvasColColours[0])
|
|
canvasLastColours[0] = canvasColColours[0]
|
|
outBuffer += canvasColText
|
|
outBuffer += "\n"
|
|
if len(outBuffer):
|
|
return (True, outBuffer)
|
|
else:
|
|
return (False, "empty buffer generated")
|
|
# }}}
|
|
# {{{ exportTextFile(self, canvasMap, canvasSize, outFile)
|
|
def exportTextFile(self, canvasMap, canvasSize, outFile):
|
|
rc, outBuffer = self.exportTextBuffer(canvasMap, canvasSize)
|
|
return outFile.write(outBuffer) if rc else (rc, outBuffer)
|
|
# }}}
|
|
|
|
# vim:expandtab foldmethod=marker sw=4 ts=4 tw=120
|