roar/libcanvas/CanvasExportStore.py

#!/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 += "\u0003 "
                    elif canvasColColours[1] == -1:
                        outBuffer += "\u0003\u0003{}".format(canvasColColours[0])
                    elif (canvasColColours[0] == canvasLastColours[1])  \
                    and  (canvasColColours[1] == canvasLastColours[0]):
                        outBuffer += "\u0016"
                    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]:
                    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