#!/usr/bin/env python3 import uuid import os import sys import subprocess import zipfile import urllib.request import gzip from io import BytesIO import time import json import pathlib import shutil from distutils.dir_util import copy_tree from flask import Flask, render_template, request, flash from threading import Thread, Lock queue_lock = Lock() ''' Directory structure: - all paths relative to where the app starts - waf builds go into build - json queued build files go into buildqueue - resulting builds go into done - ardupilot is in ardupilot directory - templates in CustomBuild/templates ''' def get_template(filename): return render_template(filename) def remove_directory_recursive(dirname): '''remove a directory recursively''' if not os.path.exists(dirname): return f = pathlib.Path(dirname) if f.is_file(): f.unlink() else: for child in f.iterdir(): shutil.rmtree(child) f.rmdir() def create_directory(dir_path): '''create a directory, don't fail if it exists''' pathlib.Path(dir_path).mkdir(parents=True, exist_ok=True) def run_build(taskfile, builddir, done_dir): # run a build with parameters from task task = json.loads(open(taskfile).read()) remove_directory_recursive(os.path.join(sourcedir, done_dir)) remove_directory_recursive(os.path.join(sourcedir, builddir)) create_directory(os.path.join(sourcedir, done_dir)) create_directory(os.path.join(sourcedir, builddir)) with open(os.path.join(sourcedir, done_dir, 'build.log'), "wb") as log: subprocess.run(['git', 'submodule', 'update', '--recursive', '--force', '--init'], stdout=log, stderr=log) subprocess.run(['./waf', 'configure', '--board', task['board'], '--out', builddir, '--extra-hwdef', task['extra_hwdef']], cwd = task['sourcedir'], stdout=log, stderr=log) subprocess.run(['./waf', 'clean'], cwd = task['sourcedir'], stdout=log, stderr=log) subprocess.run(['./waf', task['vehicle']], cwd = task['sourcedir'], stdout=log, stderr=log) # background thread to check for queued build requests def check_queue(): while(1): queue_lock.acquire() listing = os.listdir('buildqueue') queue_lock.release() if listing: for token in listing: builddir = 'build' done_dir = os.path.join('done', token) buildqueue_dir = os.path.join('buildqueue', token) # check if build exists if os.path.isdir(os.path.join(sourcedir, done_dir)): print("Build already exists") else: # run build and rename build directory f = open(os.path.join(buildqueue_dir, 'q.json')) task = json.load(f) run_build(os.path.join(buildqueue_dir, 'q.json'), builddir, done_dir) copy_tree(os.path.join(sourcedir, builddir, task['board']), os.path.join(sourcedir, done_dir)) # remove working files os.remove(os.path.join(buildqueue_dir, 'extra_hwdef.dat')) os.remove(os.path.join(buildqueue_dir, 'q.json')) os.rmdir(os.path.join(buildqueue_dir)) print("Build successful") # define source and app directories sourcedir = os.path.abspath(os.path.join('..', 'ardupilot')) appdir = os.path.abspath(os.curdir) if not os.path.isdir('buildqueue'): os.mkdir('buildqueue') thread = Thread(target=check_queue, args=()) thread.daemon = True thread.start() # Directory of this file this_path = os.path.dirname(os.path.realpath(__file__)) # Where the user requested tile are stored output_path = os.path.join(this_path, '..', 'userRequestFirmware') # Where the data database is tile_path = os.path.join(this_path, '..', 'data', 'tiles') # The output folder for all gzipped build requests app = Flask(__name__, static_url_path='/builds', static_folder=output_path, template_folder='templates') #def compressFiles(fileList, uuidkey): # create a zip file comprised of dat.gz tiles #zipthis = os.path.join(output_path, uuidkey + '.zip') # create output dirs if needed #try: # os.makedirs(output_path) #except OSError: # pass #try: # os.makedirs(tile_path) #except OSError: # pass #try: # with zipfile.ZipFile(zipthis, 'w') as terrain_zip: # for fn in fileList: # if not os.path.exists(fn): # #download if required # print("Downloading " + os.path.basename(fn)) # g = urllib.request.urlopen('https://terrain.ardupilot.org/data/tiles/' + # os.path.basename(fn)) # print("Downloaded " + os.path.basename(fn)) # with open(fn, 'b+w') as f: # f.write(g.read()) # need to decompress file and pass to zip # with gzip.open(fn, 'r') as f_in: # myio = BytesIO(f_in.read()) # print("Decomp " + os.path.basename(fn)) # and add file to zip # terrain_zip.writestr(os.path.basename(fn)[:-3], myio.read(), # compress_type=zipfile.ZIP_DEFLATED) #except Exception as ex: # print("Unexpected error: {0}".format(ex)) # return False #return True @app.route('/') def index(): return get_template('index.html') @app.route('/generate', methods=['GET', 'POST']) def generate(): #if request.method == 'POST': features = [] task = {} # fetch features from user input for i in range(1,8): value = request.form["option" + str(i)] features.append(value) undefine = "undef " + value.split()[1] features.insert(0,undefine) extra_hwdef = '\n'.join(features) #print("features: ", features) queue_lock.acquire() # create extra_hwdef.dat file and obtain md5sum file = open(os.path.join('buildqueue', 'extra_hwdef.dat'),"w") file.write(extra_hwdef) file.close() md5sum = subprocess.check_output(['md5sum', 'buildqueue/extra_hwdef.dat'], encoding = 'utf-8') md5sum = md5sum[:len(md5sum)-29] os.remove(os.path.join('buildqueue', 'extra_hwdef.dat')) # obtain git-hash of source git_hash = subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd = sourcedir, encoding = 'utf-8') git_hash = git_hash[:len(git_hash)-1] # create directories using concatenated token of git-hash and md5sum of hwdef token = git_hash + "-" + md5sum buildqueue_dir = os.path.join(appdir, 'buildqueue', token) if not os.path.isdir(buildqueue_dir): os.mkdir(buildqueue_dir) file = open(os.path.join(buildqueue_dir, 'extra_hwdef.dat'),"w") file.write(extra_hwdef) file.close() # fill dictionary of variables and create json file task['hwdef_md5sum'] = md5sum task['git_hash'] = git_hash task['token'] = token task['sourcedir'] = sourcedir task['extra_hwdef'] = os.path.join(buildqueue_dir, 'extra_hwdef.dat') task['board'] = request.form["board"] task['vehicle'] = request.form["vehicle"] jfile = open(os.path.join(buildqueue_dir, 'q.json'), "w") jfile.write(json.dumps(task)) jfile.close() queue_lock.release() #print(task) apache_build_dir = "http://localhost:8080/" + token apache_build_log = "http://localhost:8080/" + token + "/build.log" return render_template('generate.html', apache_build_dir=apache_build_dir, apache_build_log=apache_build_log, token=token,) # remove duplicates #filelist = list(dict.fromkeys(filelist)) #print(filelist) #compress #success = compressFiles(filelist, uuidkey) # as a cleanup, remove any generated terrain older than 24H #for f in os.listdir(output_path): # if os.stat(os.path.join(output_path, f)).st_mtime < time.time() - 24 * 60 * 60: # print("Removing old file: " + str(os.path.join(output_path, f))) # os.remove(os.path.join(output_path, f)) #if success: # print("Generated " + "/terrain/" + uuidkey + ".zip") # return get_template('generate.html', urlkey="/terrain/" + uuidkey + ".zip", # uuidkey=uuidkey, outsideLat=outsideLat) #else: # print("Failed " + "/terrain/" + uuidkey + ".zip") # return get_template('generate.html', error="Cannot generate terrain", # uuidkey=uuidkey) #else: # print("Bad get") # return get_template('generate.html', error="Need to use POST, not GET") @app.route('/home', methods=['POST']) def home(): return get_template('index.html') if __name__ == "__main__": app.run()