(require '[clojure.pprint :refer [pprint]]
         '[clojure.java.shell :as shell]
         '[clojure.java.io :as io]
         '[clojure.edn :as edn]
         '[clojure.string :as str]
         '[environ.core :refer [env]])
(require '[cljs.build.api :as api]
         '[cljs.repl :as repl]
         '[cljs.repl.node :as node])
(require '[rebel-readline.core]
         '[rebel-readline.clojure.main]
         '[rebel-readline.clojure.line-reader]
         '[rebel-readline.clojure.service.local]
         '[rebel-readline.cljs.service.local]
         '[rebel-readline.cljs.repl])

(import 'java.io.ByteArrayOutputStream)

(defmulti task first)

(defmethod task :default
  [args]
  (let [all-tasks  (-> task methods (dissoc :default) keys sort)
        interposed (->> all-tasks (interpose ", ") (apply str))]
    (println "Unknown or missing task. Choose one of:" interposed)
    (System/exit 1)))

;; --- Generic Build Options

(def default-build-options
  {:cache-analysis true
   :parallel-build true
   :language-in  :ecmascript6
   :language-out :ecmascript5
   :anon-fn-naming-policy :mapped
   :optimizations :none
   :infer-externs true
   :verbose true
   :source-map true
   :static-fns false
   :pretty-print true
   :elide-asserts false})

(def dist-build-options
  {:optimizations :advanced
   :pretty-print false
   :static-fns true
   ;; :fn-invoke-direct true
   :elide-asserts true})

;; --- Specific Build Options

(def main-build-options
  {:output-dir "resources/public/js/main/"
   :output-to "resources/public/js/main.js"
   :main 'uxbox.main
   :asset-path "/js/main"})

;; (def worker-build-options
;;   {:main 'uxbox.worker
;;    :target :webworker
;;    :output-to "resources/public/js/worker.js"
;;    :output-dir "resources/public/js/worker"
;;    :asset-path "/js/worker"})

(def main-dist-build-options
  (-> (merge default-build-options
             main-build-options
             dist-build-options)
      (assoc :output-dir "target/dist/js/main/"
             :source-map "target/dist/js/main.js.map"
             :output-to "target/dist/js/main.js")))

(def main-dist-dbg-build-options
  (assoc main-dist-build-options
         :optimizations :advanced
         :pseudo-names true
         :pretty-print true
         :output-dir "target/dist/dbg/js/main/"
         :source-map "target/dist/dbg/js/main.js.map"
         :output-to "target/dist/dbg/js/main.js"))

;; (def worker-dist-build-options
;;   (merge default-build-options
;;          worker-build-options
;;          dist-build-options
;;          {:output-to  "target/dist/js/worker.js"
;;           :output-dir "target/dist/js/worker"
;;           :source-map "target/dist/js/worker.js.map"}))

;; (def worker-dist-dbg-build-options
;;   (merge worker-dist-build-options
;;          {:optimizations :advanced
;;           :pseudo-names true
;;           :pretty-print true
;;           :output-to  "target/dist/dbg/js/worker.js"
;;           :output-dir "target/dist/dbg/js/worker"
;;           :source-map "target/dist/dbg/js/worker.js.map"}))

;; --- Tasks Definitions

(defmethod task "dist:main"
  [args]
  (let [cfg main-dist-build-options]
    ;; (pprint cfg)
    (api/build (api/inputs "src") cfg)))

;; (defmethod task "dist:worker"
;;   [args]
;;   (let [cfg worker-dist-build-options]
;;     ;; (pprint cfg)
;;     (api/build (api/inputs "src") cfg)))

(defmethod task "dist-dbg:main"
  [args]
  (let [cfg main-dist-dbg-build-options]
    ;; (pprint cfg)
    (api/build (api/inputs "src") cfg)))

;; (defmethod task "dist-dbg:worker"
;;   [args]
;;   (let [cfg worker-dist-dbg-build-options]
;;     ;; (pprint cfg)
;;     (api/build (api/inputs "src") cfg)))

(defmethod task "dist:all"
  [args]
  (task ["dist:main"])
  #_(task ["dist:worker"])
  (task ["dist-dbg:main"])
  #_(task ["dist-dbg:worker"]))

(defmethod task "repl:node"
  [args]
  (rebel-readline.core/with-line-reader
    (rebel-readline.clojure.line-reader/create
     (rebel-readline.cljs.service.local/create))
    (cljs.repl/repl
     (node/repl-env)
     :prompt (fn []) ;; prompt is handled by line-reader
     :read (rebel-readline.cljs.repl/create-repl-read)
     :output-dir "out"
     :cache-analysis false)))

;; --- Tests Tasks

(defmethod task "build:tests"
  [& args]
  (api/build (api/inputs "src" "test")
             (assoc default-build-options
                    :main 'uxbox.tests.main
                    :verbose true
                    :target :nodejs
                    :source-map true
                    :output-to "target/tests/main.js"
                    :output-dir "target/tests/main"
                    :optimizations :none)))

(defmethod task "watch:tests"
  [args]
  (println "Start watch loop...")
  (letfn [(run-tests []
            (let [{:keys [out err]} (shell/sh "node" "target/tests/main.js")]
              (println out err)))
          (start-watch []
            (try
              (api/watch (api/inputs "src" "test")
                         (assoc default-build-options
                                :main 'uxbox.tests.main
                                :watch-fn run-tests
                                :target :nodejs
                                :source-map true
                                :output-to "target/tests/main.js"
                                :output-dir "target/tests/main"
                                :optimizations :none))
              (catch Exception e
                (println "ERROR:" e)
                (Thread/sleep 2000)
                start-watch)))]
    (trampoline start-watch)))

;;; Build script entrypoint. This should be the last expression.

(task *command-line-args*)