plumb.rkt

;; The MIT License (MIT)
;; 
;; Copyright (c) 2015 Matthew C. Jadud
;; 
;; Permission is hereby granted, free of charge, to any person obtaining a copy
;; of this software and associated documentation files (the "Software"), to deal
;; in the Software without restriction, including without limitation the rights
;; to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
;; copies of the Software, and to permit persons to whom the Software is
;; furnished to do so, subject to the following conditions:
;; 
;; The above copyright notice and this permission notice shall be included in
;; all copies or substantial portions of the Software.
;; 
;; THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
;; IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
;; FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
;; AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
;; LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
;; OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
;; THE SOFTWARE.

#lang racket
(require yaml
         "util.rkt"
         "config.rkt"
         "debug.rkt"
         "model-plumb.rkt"
         (prefix-in ccmds: "client-cmds.rkt")
         (prefix-in cmds: "cmds.rkt"))


(define cmd-line-params (make-hash))


(define (driver)
  
  (define resp (ccmds:compile))
  (define outp (open-output-file (conf-get "firmware-name") #:exists 'replace))
  (fprintf outp (hash-ref resp "hex"))
  (close-output-port outp)
  (flush-ports)
  
  ;; Last step:
  ;; Config does not seem to be essential...
  ;; But will be on Windows, Mac.
  ;; avrdude -C /usr/share/arduino/hardware/tools/avrdude.conf \
  ;;         -V -F -P /dev/ttyUSB0 -p m328p -b 57600 -c arduino -U flash:w:firmware.hex:i
  
  (cond
    [(not (conf-get "use-mqtt?"))
     (debug 'DRIVER "Using AVRDUDE.")
     (ccmds:avrdude resp)]
    [(conf-get "use-mqtt?")
     (debug 'DRIVER "Using MQTT.")
     (ccmds:mqtt-notify resp)]))


;; Defaults

(define (make-board-string)
  (let ([all '()])
    ;;(define boards (file->yaml "boards.yaml"))
    (define boards (conf-get "boards"))
    (for ([b boards])
      (for ([n (hash-ref b "names")])
        (set! all (snoc all (format "~a (family: ~a)" n (hash-ref b "family"))))
        ))
    all))

(define (get-board b)
  ;;(define boards (file->yaml "boards.yaml"))
  (define boards (conf-get "boards"))
  (define result false)
  (for ([brd boards])
    (debug 'GB "~a : ~a" b (hash-ref brd "family"))
    (when (equal? b (hash-ref brd "family"))
      (set! result (hash-ref brd "params"))))
  result)


(config-file false)
(define (init)  
  (define my-path (extract-filedir (find-system-path 'run-file)))
  (unless (config-file)
    (config-file (build-path my-path
                             "client-config"
                             "client.yaml")))
  (load-additional-client-config)  
  (load-config)
  
  (conf-add "boards" 
            (string->yaml
             (read-url (format "http://~a:~a/ide/boards.yaml" 
                               (conf-get 'server)
                               (conf-get 'port)))))  
  )


;; Load our configuration (client.yaml by default)
(init)

(define plumb
  (command-line
   #:program "plumb"
   #:once-each 
   [("-d" "--debug") "Turn on verbose debugging."
                     (set-textual-debug)
                     (enable-debug! 'ALL)]
   
   [("--config") c
                 "Choose the client YAML config."
                 (config-file c)
                 (init)]
   
   [("--board") board
                ((format "~n\tChoose the board family for your Arduino.")
                 (apply string-append
                        (map (λ (b) (format "\t* ~a~n" b)) (make-board-string))))
                
                
                (hash-set! cmd-line-params "board" board)
                ]
   
   [("--serial") serial
                 "Arduino serial port."
                 (debug 'SERIAL "Setting serial port to: ~a~n" serial)
                 (hash-set! cmd-line-params "serial" serial)]
   
   [("--mqtt-host") mqtt-host
                    "Set the MQTT host for notifications."
                    (hash-set! cmd-line-params "use-mqtt?" true)
                    (hash-set! cmd-line-params "mqtt-host" mqtt-host)]
   [("--mqtt-port") mqtt-port
                    "Set the MQTT port."
                    (hash-set! cmd-line-params "use-mqtt?" true)
                    (hash-set! cmd-line-params "mqtt-port" (string->number mqtt-port))]
   [("--mqtt-channel") mqtt-channel
                       "Set the MQTT channel."
                       (hash-set! cmd-line-params "use-mqtt?" true)
                       (hash-set! cmd-line-params "mqtt-channel" mqtt-channel)]
   
   [("--mqtt-url") mqtt-url
                   "The root URL for where files can be picked up."
                   (hash-set! cmd-line-params "use-mqtt?" true)
                   (hash-set! cmd-line-params "mqtt-url" mqtt-url)]
   
   ;; One file on the command line
   #:args (file)
   
   ;; Provide a default board.
   (unless (hash-ref cmd-line-params "board" false)
     (define board-params (get-board "arduino"))
     (debug 'BOARDPARAMS (~s board-params))
     (hash-set! cmd-line-params "board" board-params))
   
   ;; Merge the command-line params with the standing config.
   (for ([(k v) cmd-line-params])
     (conf-add k v))
   
   ;; Noisy. Announce our presence.
   (debug 'PLUMB "Server is at http://~a:~a/" 
          (conf-get 'server)
          (conf-get 'port))
   
   ;; Store the source file's path in the config.
   ;; I don't know if these are ever used.
   (conf-add 'source-path (extract-filedir (path->complete-path file)))
   (conf-add 'source-file file)
   
   ;; Drive the compilation.
   (driver)
   ))