(ns leiningen.jar

"Package up all the project's files into a jar file."

(:require [leiningen.pom :as pom]

[leiningen.clean :as clean]

[leiningen.core.classpath :as classpath]

[leiningen.core.project :as project]

[leiningen.core.eval :as eval]

[leiningen.core.main :as main]

[leiningen.core.utils :as utils]

[bultitude.core :as b]

[clojure.set :as set]

[clojure.string :as string]

[clojure.java.io :as io])

(:import (java.util.jar Manifest JarEntry JarOutputStream)

(java.util.regex Pattern)

(java.util.jar JarFile)

(java.io BufferedOutputStream FileOutputStream

ByteArrayInputStream)))

(defn- unix-path [path]

(.replace path "\\" "/"))

(def ^:private default-manifest

[["Created-By" (str "Leiningen " (main/leiningen-version))]

["Built-By" (System/getProperty "user.name")]

["Build-Jdk" (System/getProperty "java.version")]])

(declare ^:private manifest-entry)

(defn- manifest-entries [project manifest-seq]

(map (partial manifest-entry project) manifest-seq))

(defn- manifest-entry [project [k v]]

(cond (symbol? v) (manifest-entry project [k (resolve v)])

(fn? v) (manifest-entry project [k (v project)])

(coll? v) (->> v ;; Sub-manifest = manifest section

(manifest-entries project)

(cons (str "\nName: " (name k) "\n"))

(string/join))

:else (->> (str (name k) ": " v)

(partition-all 70) ;; Manifest spec says lines <= 72 chars

(map (partial apply str))

(string/join "\n ") ;; Manifest spec says join with "\n "

(format "%s\n"))))

(defn- manifest-map-to-reordered-seq [mf]

(sort

(comparator

(fn [e1 e2]

(not (coll? (second e1)))))

(seq mf)))

(defn ^:internal make-manifest [project]

(let [initial-mf

(concat (if (get (:manifest project) "Main-Class")

default-manifest

(conj default-manifest

["Main-Class"

(munge (str (:main project 'clojure.main)))]))

(manifest-map-to-reordered-seq (:manifest project)))]

(->> initial-mf

(manifest-entries project)

(cons "Manifest-Version: 1.0\n") ;; Manifest-Version line must be first

(string/join "")

.getBytes

ByteArrayInputStream.

Manifest.)))

(defn ^:internal manifest-map [manifest]

(let [attrs (.getMainAttributes manifest)]

(zipmap (map str (keys attrs)) (vals attrs))))

(defn- added-file?

"Returns true if the file is already added to the jar, false otherwise. Prints

[file relative-path added-paths]

;; Path may be blank if it is the root path

(if (or (string/blank? relative-path) (added-paths relative-path))

(do

(when-not (.isDirectory file)

(main/info "Warning: skipped duplicate file:" relative-path))

true)))

(defn- skip-file?

"Skips the file if it doesn't exist. If the file is not the

[file relative-path root-file patterns]

(or (not (.exists file))

(and

(not= file root-file)

(or

(re-find #"^\.?#" (.getName file))

(re-find #"~$" (.getName file))

(some #(re-find % relative-path) patterns)))))

(defmulti ^:private copy-to-jar (fn [project jar-os acc spec] (:type spec)))

(defn- relativize-path

"Relativizes a path: Removes the root-path of a path if not already removed."

[path root-path]

(if (.startsWith path root-path)

(.substring path (.length root-path))

path))

(defn- full-path ;; Q: is this a good name for this action?

"Appends the path string with a '/' if the file is a directory."

[file path]

(if (.isDirectory file)

(str path "/")

path))

(defn- dir-string

"Returns the file's directory as a string, or the string representation of the

[file]

(if-not (.isDirectory file)

(str (.getParent file) "/")

(str file "/")))

(defn- put-jar-entry!

"Adds a jar entry to the Jar output stream."

[jar-os file path]

(.putNextEntry jar-os (doto (JarEntry. path)

(.setTime (.lastModified file))))

(when-not (.isDirectory file)

(io/copy file jar-os)))

(defmethod copy-to-jar :path [project jar-os acc spec]

(let [root-file (io/file (:path spec))

root-dir-path (unix-path (dir-string root-file))

paths (for [child (file-seq root-file)

:let [path (relativize-path

(full-path child (unix-path (str child)))

root-dir-path)]]

(when-not (or (skip-file? child path root-file

(:jar-exclusions project))

(added-file? child path acc))

(put-jar-entry! jar-os child path)

path))]

(into acc paths)))

(defmethod copy-to-jar :paths [project jar-os acc spec]

(reduce (partial copy-to-jar project jar-os) acc

(for [path (:paths spec)]

{:type :path :path path})))

(defmethod copy-to-jar :bytes [project jar-os acc spec]

(let [path (unix-path (:path spec))]

(when-not (some #(re-find % path) (:jar-exclusions project))

(.putNextEntry jar-os (JarEntry. path))

(let [bytes (if (string? (:bytes spec))

(.getBytes (:bytes spec))

(:bytes spec))]

(io/copy (ByteArrayInputStream. bytes) jar-os)))

(conj acc path)))

(defmethod copy-to-jar :fn [project jar-os acc spec]

(let [f (eval (:fn spec))

dynamic-spec (f project)]

(copy-to-jar project jar-os acc dynamic-spec)))

(defn write-jar [project out-file filespecs]

(with-open [jar-os (-> out-file

(FileOutputStream.)

(BufferedOutputStream.)

(JarOutputStream. (make-manifest project)))]

(let [jar-paths (reduce (partial copy-to-jar project jar-os)

#{}

filespecs)]

(if (:main project)

(let [main-path (str (-> (string/replace (:main project) "." "/")

(string/replace "-" "_"))

".class")]

(if (not (some #{main-path} jar-paths))

(main/info "Warning: The Main-Class specified does not exist"

"within the jar. It may not be executable as expected."

"A gen-class directive may be missing in the namespace"

"which contains the main method."))))

jar-paths)))

(defn- filespecs [project]

(let [root-files (.list (io/file (:root project)))

readmes (filter (partial re-find #"^(?i)readme") root-files)

licenses (filter (partial re-find #"^(?i)license") root-files)

scope (partial format "META-INF/leiningen/%s/%s/%s"

(:group project) (:name project))]

(concat [{:type :bytes

:path (format "META-INF/maven/%s/%s/pom.xml"

(:group project) (:name project))

:bytes (.getBytes (pom/make-pom project))}

{:type :bytes :path (scope "project.clj")

:bytes (.getBytes (slurp (str (:root project) "/project.clj")))}

{:type :bytes :path "project.clj"

:bytes (.getBytes (slurp (str (:root project) "/project.clj")))}]

(for [doc (map (partial io/file (:root project))

(concat readmes licenses))

:when (.isFile doc)]

{:type :bytes :path (scope (.getName doc))

:bytes (.getBytes (slurp doc))})

[{:type :path :path (:compile-path project)}

{:type :paths :paths (:resource-paths project)}]

(if-not (:omit-source project)

[{:type :paths

:paths (set (concat (:source-paths project)

(:java-source-paths project)))}])

(:filespecs project))))

(defn get-classified-jar-filename [project classifier]

(let [target (doto (io/file (:target-path project)) .mkdirs)

suffix (if classifier (str "-" (name classifier) ".jar") ".jar")

name-kw (if (= classifier :standalone) :uberjar-name :jar-name)

jar-name (or (project name-kw) (str (:name project) "-%s" suffix))

jar-name (format jar-name (:version project))]

(str (io/file target jar-name))))

(def whitelist-keys

"Project keys which don't affect the production of the jar should be

[:offline? :local-repo :certificates :warn-on-reflection :mirrors])

(defn- retain-whitelisted-keys

"Retains the whitelisted keys from the original map in the new one."

[new original]

(merge new (select-keys original whitelist-keys)))

(defn- compile-main? [{:keys [main source-paths] :as project}]

(and main (not (:skip-aot (meta main)))

(some #(.exists (io/file % (b/path-for main))) source-paths)))

(def ^:private implicit-aot-warning

(delay

(main/info "Warning: specified :main without including it in :aot."

"\nImplicit AOT of :main will be removed in Leiningen 3.0.0."

"\nIf you only need AOT for your uberjar, consider adding"

":aot :all into your\n:uberjar profile instead.")))

(defn warn-implicit-aot [orig-project]

(let [project (project/merge-profiles orig-project [:uberjar])]

(when (and (:main project) (not (:skip-aot (meta (:main project))))

(not= :all (:aot project))

(not= [:all] (:aot project))

(not (some #{(:main project)} (:aot project))))

(force implicit-aot-warning))))

(defn- add-main [project given-main]

(warn-implicit-aot project)

(let [project (if given-main

(assoc project :main (symbol given-main))

project)]

(if (and (compile-main? project)

(not= :all (:aot project))

(not= [:all] (:aot project)))

(update-in project [:aot] conj (:main project))

project)))

(defn- process-project

"Like update-in, but for preparing projects for (uber)jaring. f is a function

[project main f & args]

(-> (apply f project args)

(retain-whitelisted-keys project)

(add-main main)))

(defn- preprocess-project [project & [main]]

(process-project project main project/unmerge-profiles

(project/non-leaky-profiles project)))

(defn- get-jar-filename*

[project uberjar?]

(get-classified-jar-filename project (when uberjar? :standalone)))

(defn get-jar-filename [project & [uberjar?]]

(get-jar-filename* (preprocess-project project) uberjar?))

(defn build-jar

"Build a jar for the given project and jar-file."

[project jar-file]

(eval/prep project)

(write-jar project jar-file (filespecs project))

(main/info "Created" (str jar-file))

jar-file)

(defn main-jar

[project provided-profiles main]

(let [project (process-project project main project/merge-profiles

provided-profiles)]

{[:extension "jar"] (build-jar project (get-jar-filename* project nil))}))

(defn classifier-jar

"Package up all the project's classified files into a jar file.

[{:keys [target-path] :as project} provided-profiles classifier spec]

(when (:dependencies spec)

(main/warn

"WARNING: Classifier specifies :dependencies which will be ignored."))

(let [profiles (concat provided-profiles [::target ::classifier])

target-profile {:target-path

(.getPath (io/file target-path (name classifier)))}

project (-> project

(vary-meta assoc-in [:profiles ::classifier] spec)

(vary-meta assoc-in [:profiles ::target] target-profile)

(process-project nil project/merge-profiles profiles))]

[[:classifier (name classifier) :extension "jar"]

(build-jar project (get-classified-jar-filename project classifier))]))

(defn classifier-jars

"Package up all the project's classified files into jar files.

[{:keys [classifiers] :as project} provided-profiles]

(into {}

(map #(apply classifier-jar project provided-profiles %) classifiers)))

(defn jar

"Package up all the project's files into a jar file.

([project main]

(utils/with-write-permissions (:root project)

(when (:auto-clean project true)

(clean/clean project))

(let [scoped-profiles (set (project/pom-scope-profiles project :provided))

default-profiles (set (project/expand-profile project :default))

provided-profiles (remove

(set/difference default-profiles scoped-profiles)

(-> project meta :included-profiles))

project (preprocess-project project main)]

(merge (main-jar project provided-profiles main)

(classifier-jars project provided-profiles)))))

([project] (jar project nil)))