AVRSack/AVRsack/ASBuilder.swift

//
//  ASBuilder.swift
//  AVRsack
//
//  Created by Matthias Neeracher on 11/24/14.
//  Copyright © 2014 Aere Perennius. All rights reserved.
//

import Foundation

class ASBuilder {
    var dir         = NSURL()
    var task        : NSTask?
    var continuation: (()->())?
    var termination : AnyObject?
    
    init() {
        termination = NSNotificationCenter.defaultCenter().addObserverForName(NSTaskDidTerminateNotification,
            object: nil, queue: nil, usingBlock:
        { (notification: NSNotification!) in
            if notification.object as? NSTask == self.task && self.task!.terminationStatus == 0 {
                if let cont = self.continuation {
                    self.continuation = nil
                    cont()
                }
            }
        })
    }
    func finalize() {
        NSNotificationCenter.defaultCenter().removeObserver(termination!)
    }
    
    func setProjectURL(url: NSURL) {
        dir       = url.URLByDeletingLastPathComponent!.standardizedURL!
    }

    func stop() {
        task?.terminate()
        task?.waitUntilExit()
    }
    
    func cleanProject() {
        NSFileManager.defaultManager().removeItemAtURL(dir.URLByAppendingPathComponent("build"), error: nil)
    }
    
    func buildProject(board: String, files: ASFileTree) {
        task = NSTask()
        task!.currentDirectoryPath  = dir.path!
        task!.launchPath            = NSBundle.mainBundle().pathForResource("BuildProject", ofType: "")!
        
        let fileManager = NSFileManager.defaultManager()
        let libPath     = (ASLibraries.instance().directories as NSArray).componentsJoinedByString(":")
        var args        = [NSString]()
        let boardProp   = ASHardware.instance().boards[board]!
        var corePath    = ""
        var variantPath : NSString?
        for hw in ASHardware.instance().directories {
            corePath = hw+"/cores/"+boardProp["build.core"]!
            if fileManager.fileExistsAtPath(corePath) {
                if let variantName = boardProp["build.variant"] {
                    variantPath = hw+"/variants/"+variantName
                    if fileManager.fileExistsAtPath(variantPath!) {
                        args.append("variant="+variantName)
                   } else {
                        variantPath = nil
                    }
                }
                break
            } else {
                corePath = ""
            }
        }
        if corePath == "" {
            NSLog("Unable to find core %s\n", boardProp["build.core"]!)
            return
        }
        args.append("project="+dir.lastPathComponent)
        args.append("board="+board)
        args.append("mcu="+boardProp["build.mcu"]!)
        args.append("f_cpu="+boardProp["build.f_cpu"]!)
        args.append("max_size"+boardProp["upload.maximum_size"]!)
        args.append("core="+boardProp["build.core"]!)
        args.append("libs="+libPath)
        args.append("core_path="+corePath)
        if variantPath != nil {
            args.append("variant_path="+variantPath!)
        }
        args.append("--")
        args                   += files.paths
        task!.arguments         =   args;
        task!.launch()
    }
    
    func uploadProject(board: String, programmer: String, port: String) {
        task = NSTask()
        task!.currentDirectoryPath  = dir.path!
        task!.launchPath            = "/usr/local/CrossPack-AVR/bin/avrdude"
        
        let fileManager = NSFileManager.defaultManager()
        let logURL              = dir.URLByAppendingPathComponent("build/upload.log")
        fileManager.createFileAtPath(logURL.path!, contents: NSData(), attributes: nil)
        let logOut              = NSFileHandle(forWritingAtPath: logURL.path!)!
        task!.standardOutput    = logOut
        task!.standardError     = logOut
        
        let libPath     = (ASLibraries.instance().directories as NSArray).componentsJoinedByString(":")
        let boardProp   = ASHardware.instance().boards[board]!
        let progProp    = ASHardware.instance().programmers[programmer]
        let proto       = boardProp["upload.protocol"] ?? progProp?["protocol"]
        let speed       = boardProp["upload.speed"]    ?? progProp?["speed"]
        var args        = ["-v", "-v", "-v", "-v", "-D",
            "-C", "/usr/local/CrossPack-AVR/etc/avrdude.conf",
            "-p", boardProp["build.mcu"]!, "-c", proto!, "-P", port,
            "-U", "flash:w:build/"+board+"/"+dir.lastPathComponent+".hex:i"]
        if speed != nil {
            args.append("-b")
            args.append(speed!)
        }
        let cmdLine = task!.launchPath+" "+(args as NSArray).componentsJoinedByString(" ")+"\n"
        logOut.writeData(cmdLine.dataUsingEncoding(NSUTF8StringEncoding, allowLossyConversion: true)!)
        task!.arguments         =   args;
        task!.launch()
    }
}