clean_ns.clj

(ns refactor-nrepl.ns.clean-ns
  "Contains functionality for cleaning ns forms.

  * Eliminate :use, or :use-macro clauses.
  * Sort required libraries, imports and vectors of referred symbols.
  * Rewrite to favor prefix form, e.g. [clojure [string test]] instead
    of two separate libspecs.
  * Raise errors if any inconsistencies are found (e.g. a libspec with more than
    one alias).
  * Remove any duplication in the :require, :require-macros, :use-macros
    and :import form.
  * Remove any unused required namespaces or imported classes.
  * Remove any unused referred symbols.
  * Prune, or remove if uneeded, the :rename clause
  * Returns nil when nothing is changed, so the client knows not to do anything."
  (:require [refactor-nrepl
             [config :as config]
             [core :as core]]
            [refactor-nrepl.ns
             [ns-parser :as ns-parser]
             [prune-dependencies :refer [prune-dependencies]]
             [rebuild :refer [rebuild-ns-form]]]
            [clojure.string :as str]))

(defn- assert-no-exclude-clause
  [use-form]
  (dorun (map
          #(when (= % :exclude)
             (throw (IllegalArgumentException.
                     "Can't run clean-ns on :use clause containing :exclude!")))
          (tree-seq sequential? identity use-form)))
  use-form)

(defn- validate [ns-form]
  (assert-no-exclude-clause (core/get-ns-component ns-form :use))
  ns-form)

(defn clean-ns [{:keys [path]}]
  {:pre [(seq path) (string? path) (core/source-file? path)]}
  ;; Prefix notation not supported in cljs.
  ;; We also turn it off for cljc for reasons of symmetry
  (config/with-config {:prefix-rewriting (if (or (core/cljs-file? path)
                                                 (core/cljc-file? path))
                                           false
                                           (:prefix-rewriting config/*config*))}
    (let [ns-form (validate (core/read-ns-form-with-meta path))
          deps-preprocessor (if (get config/*config* :prune-ns-form)
                              #(prune-dependencies % path)
                              identity)
          new-ns-form (-> (ns-parser/parse-ns path)
                          deps-preprocessor
                          (rebuild-ns-form ns-form))]
      (when-not (= ns-form new-ns-form)
        new-ns-form))))