Add package to build submission zephyrs

[invirt/packages/invirt-dev.git] / invirt-submit-build

diff --git a/invirt-submit-build b/invirt-submit-build
index 7115bf6..f202c3d 100755 (executable)
--- a/invirt-submit-build
+++ b/invirt-submit-build
@@ -18,16 +18,24 @@ keeping.
 
 
 import datetime
 
 
 import datetime

+import optparse

 import os
 import sys
 import tempfile
 import uuid
 
 import os
 import sys
 import tempfile
 import uuid
 

+from invirt.config import structs as config
+import invirt.common as c

 import invirt.builder as b
 
 
 def main():
 import invirt.builder as b
 
 
 def main():

-    pocket, package, commit = sys.argv[1:4]
+    parser = optparse.OptionParser('Usage: %prog pocket package commit')
+    opts, args = parser.parse_args()
+    if len(args) != 3:
+        parser.print_help()
+        return 1
+    pocket, package, commit = args

     principal = os.environ['REMOTE_USER']
     request_time = datetime.datetime.utcnow()
     q_path = os.path.join(b._QUEUE_DIR,
     principal = os.environ['REMOTE_USER']
     request_time = datetime.datetime.utcnow()
     q_path = os.path.join(b._QUEUE_DIR,

@@ -35,9 +43,18 @@ def main():
                                      uuid.uuid4()))
 
     try:
                                      uuid.uuid4()))
 
     try:

-        validateBuild(pocket, package, commit)
+        # TODO: clean up this interface.
+        b.ensureValidPackage(package)
+        if config.build.get('mirror'):
+            c.captureOutput(['git', 'fetch'], cwd=b.getRepo(package))
+        commit = b.canonicalize_commit(package, commit)
+        b.validateBuild(pocket, package, commit)

     except b.InvalidBuild, e:
     except b.InvalidBuild, e:

-        print >>sys.stderr, "E: %s" % e
+        msg = "E: %s" % e
+        print >>sys.stderr, msg
+        # Prevent an attack by submitting excessively long arguments
+        args = [arg[:min(len(arg), 80)] for arg in (pocket, package, commit)]
+        b.runHook('failed-submit', args + [principal], stdin_str=msg)

         sys.exit(1)
 
     # To keep from triggering the Invirtibuilder before we've actually
         sys.exit(1)
 
     # To keep from triggering the Invirtibuilder before we've actually

@@ -46,7 +63,11 @@ def main():
     q_fd, q_name = tempfile.mkstemp()
     q = os.fdopen(q_fd, 'r+')
     print >>q, "%s %s %s %s" % (pocket, package, commit, principal)
     q_fd, q_name = tempfile.mkstemp()
     q = os.fdopen(q_fd, 'r+')
     print >>q, "%s %s %s %s" % (pocket, package, commit, principal)

+    q.close()

     os.rename(q_name, q_path)
     os.rename(q_name, q_path)

+    short_commit = b.canonicalize_commit(package, commit, shorten=True)
+    b.runHook('post-submit', [pocket, package, short_commit, principal])
+    print '%s, your job to build %s for %s:%s has been submitted!' % (principal, short_commit, package, pocket)

 
 
 if __name__ == '__main__':
 
 
 if __name__ == '__main__':