1 files changed, 75 insertions, 0 deletions
diff --git a/jarmonbuild/commands.py b/jarmonbuild/commands.py
new file mode 100644
index 0000000..e78a78e
--- /dev/null
+++ b/jarmonbuild/commands.py
@@ -0,0 +1,75 @@
+import hashlib
+import os
+import sys
+
+from subprocess import check_call
+from tempfile import gettempdir
+from urllib2 import urlopen
+from zipfile import ZipFile
+
+
+YUIDOC_URL = 'http://yuilibrary.com/downloads/yuidoc/yuidoc_1.0.0b1.zip'
+YUIDOC_MD5 = 'cd5545d2dec8f7afe3d18e793538162c'
+
+class BuildApidocsCommand(object):
+    def __init__(self, stdout=sys.stdout, stderr=sys.stderr):
+        self.stdout = stdout
+        self.stderr = stderr
+
+    def log(self, message, newline=os.linesep):
+        self.stderr.write(''.join((message, newline)))
+
+    def main(self, argv=sys.argv):
+        # setup working dir
+        tmpdir = os.path.join(gettempdir(), 'jarmonbuild')
+        if not os.path.isdir(tmpdir):
+            os.mkdir(tmpdir)
+
+        # download and cache yuidoc
+        yuizip_path = os.path.join(tmpdir, os.path.basename(YUIDOC_URL))
+        if os.path.exists(yuizip_path):
+            def producer():
+                self.log('Using cached YUI doc')
+                yield open(yuizip_path).read()
+        else:
+            def producer():
+                with open(yuizip_path, 'w') as yuizip:
+                    self.log('Downloading YUI Doc', newline='')
+                    download = urlopen(YUIDOC_URL)
+                    while True:
+                        bytes = download.read(1024*10)
+                        if not bytes:
+                            self.log('')
+                            break
+                        else:
+                            yuizip.write(bytes)
+                            self.log('.', newline='')
+                            yield bytes
+
+        checksum = hashlib.md5()
+        for bytes in producer():
+            checksum.update(bytes)
+
+        if checksum.hexdigest() != YUIDOC_MD5:
+            sys.log('checksum mismatch')
+
+        # extract yuidoc folder from the downloaded zip file
+        zip = ZipFile(yuizip_path)
+        zip.extractall(
+            tmpdir, (m for m in zip.namelist() if m.startswith('yuidoc')))
+
+        workingbranch_dir = os.path.join(os.path.dirname(__file__), '..')
+        # Use the yuidoc script that we just extracted to generate new docs
+        check_call((
+            sys.executable,
+            os.path.join(tmpdir, 'yuidoc', 'bin', 'yuidoc.py'),
+            workingbranch_dir,
+            '-p', os.path.join(workingbranch_dir, 'docs', 'apidocs'),
+            '-o', os.path.join(workingbranch_dir, 'docs', 'apidocs'),
+            '-t', os.path.join(
+                        workingbranch_dir, 'jarmonbuild', 'yuidoc_template'),
+            '-v', '10.8',
+            '-Y', '2',
+            '--project=Jarmon',
+            '--projecturl=http://www.launchpad.net/jarmon'
+        ))