updater.py

import ssl
import subprocess
import plistlib
import AppKit
import re
import traceback
import tempfile
import ssl
from urllib.request import urlopen, Request

from packaging.version import Version

import vanilla
from vanilla.dialogs import message
from defconAppKit.windows.progressWindow import ProgressWindow

from drawBot import __version__
from .misc import DrawBotError, getDefault


_versionRE = re.compile(r'__version__\s*=\s*\"([^\"]+)\"')

__fallback_version__ = "0.0"


def getCurrentVersion():
    """
    Return tuple (succesfully retreived, newest version number) from github.
    """
    errors = []
    if not getDefault("checkForUpdatesAtStartup", True):
        return __fallback_version__, errors
    path = "https://raw.github.com/typemytype/drawbot/master/drawBot/drawBotSettings.py"
    code = None
    try:
        context = ssl._create_unverified_context()
        response = urlopen(path, timeout=5, context=context)
        code = response.read()
        # convert to ascii and stri
        # in py3 this are bytes and a string object is needed
        code = str(code.decode("ascii"))
        response.close()
    except Exception as e:
        # just silently fail, its not so important
        errors.append(f"Cannot retrieve version number from DrawBot repo:\n{e}")
        pass
    if code:
        found = _versionRE.search(code)
        if found:
            return found.group(1), errors
        else:
            errors.append(f"Cannot retrieve version number from the following code:\n{code}")
    return __fallback_version__, errors


def downloadCurrentVersion():
    """
    Download the current version (dmg) and mount it
    """
    path = "https://github.com/typemytype/drawbot/releases/latest/download/DrawBot.dmg"
    try:
        context = ssl._create_unverified_context()
        request = Request(path, headers={'User-Agent': 'Drawbot'})
        # download and mount
        with tempfile.NamedTemporaryFile(mode='w+b') as dmgFile:
            response = urlopen(request, timeout=5, context=context)
            dmgFile.write(response.read())
            response.close()
            cmds = ["hdiutil", "attach", "-plist", dmgFile.name]
            popen = subprocess.Popen(cmds, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
            out, err = popen.communicate()
        if popen.returncode != 0:
            raise DrawBotError("Mounting failed")
        output = plistlib.loads(out)
        dmgPath = None
        for item in output["system-entities"]:
            if "mount-point" in item:
                dmgPath = item["mount-point"]
                break
        AppKit.NSWorkspace.sharedWorkspace().openFile_(dmgPath)
    except Exception:
        exc = traceback.format_exc()
        message("Something went wrong while downloading %s" % path, exc)


class Updater:
    """
    Small controller poping up if a update is found.
    """
    def __init__(self, parentWindow=None):
        self.needsUpdate = False
        self.__version__ = __version__
        if not getDefault("DrawBotCheckForUpdatesAtStartup", True):
            return
        self.currentVersion, self.currentVersionErrors = getCurrentVersion()
        self.needsUpdate = Version(__version__) < Version(self.currentVersion)
        if self.currentVersionErrors:
            # print them out so the debugger window catch this
            print("\n".join(self.currentVersionErrors))
        if not self.needsUpdate:
            return
        if parentWindow:
            self.w = vanilla.Sheet((450, 130), parentWindow=parentWindow)
        else:
            self.w = vanilla.Window((450, 130))

        self.w.appIcon = vanilla.ImageView((25, 15, 65, 65))
        self.w.appIcon.setImage(imageObject=AppKit.NSApp().applicationIconImage())

        title = "There is a new version of DrawBot!"
        txt = AppKit.NSAttributedString.alloc().initWithString_attributes_(title, {AppKit.NSFontAttributeName: AppKit.NSFont.boldSystemFontOfSize_(0)})
        self.w.introBold = vanilla.TextBox((100, 15, -15, 20), txt)

        self.w.intro = vanilla.TextBox((100, 45, -15, 200), "DrawBot %s is out now (you have %s).\nWould you like to download it now?" % (self.currentVersion, __version__), sizeStyle="small")

        self.w.cancelButton = vanilla.Button((-270, -30, 60, 20), "Cancel", callback=self.cancelCallback, sizeStyle="small")
        self.w.cancelButton.bind(".", ["command"])
        self.w.cancelButton.bind(chr(27), [])

        self.w.openInBrowser = vanilla.Button((-200, -30, 120, 20), "Show In Browser", callback=self.openInBrowserCallback, sizeStyle="small")
        self.w.okButton = vanilla.Button((-70, -30, 55, 20), "OK", callback=self.okCallback, sizeStyle="small")
        self.w.setDefaultButton(self.w.okButton)

        self.w.open()

    def cancelCallback(self, sender):
        self.w.close()

    def openInBrowserCallback(self, sender):
        url = "http://www.drawbot.com/content/download.html"
        url = AppKit.NSURL.URLWithString_(url)
        AppKit.NSWorkspace.sharedWorkspace().openURL_(url)
        self.w.close()

    def okCallback(self, sender):
        self.w.close()
        progress = ProgressWindow("Downloading DrawBot %s" % self.currentVersion)
        downloadCurrentVersion()
        progress.close()