52 lines
1.8 KiB
Clojure
52 lines
1.8 KiB
Clojure
(ns swank.commands.xref
|
|
(:use clojure.walk swank.util)
|
|
(:import (clojure.lang RT)
|
|
(java.io LineNumberReader InputStreamReader PushbackReader)))
|
|
|
|
;; Yoinked and modified from clojure.contrib.repl-utils.
|
|
;; Now takes a var instead of a sym in the current ns
|
|
(defn- get-source-from-var
|
|
"Returns a string of the source code for the given symbol, if it can
|
|
find it. This requires that the symbol resolve to a Var defined in
|
|
a namespace for which the .clj is in the classpath. Returns nil if
|
|
it can't find the source.
|
|
Example: (get-source-from-var 'filter)"
|
|
[v] (when-let [filepath (:file (meta v))]
|
|
(when-let [strm (.getResourceAsStream (RT/baseLoader) filepath)]
|
|
(with-open [rdr (LineNumberReader. (InputStreamReader. strm))]
|
|
(dotimes [_ (dec (:line (meta v)))] (.readLine rdr))
|
|
(let [text (StringBuilder.)
|
|
pbr (proxy [PushbackReader] [rdr]
|
|
(read [] (let [i (proxy-super read)]
|
|
(.append text (char i))
|
|
i)))]
|
|
(read (PushbackReader. pbr))
|
|
(str text))))))
|
|
|
|
(defn- recursive-contains? [coll obj]
|
|
"True if coll contains obj. Obj can't be a seq"
|
|
(not (empty? (filter #(= obj %) (flatten coll)))))
|
|
|
|
(defn- does-var-call-fn [var fn]
|
|
"Checks if a var calls a function named 'fn"
|
|
(if-let [source (get-source-from-var var)]
|
|
(let [node (read-string source)]
|
|
(if (recursive-contains? node fn)
|
|
var
|
|
false))))
|
|
|
|
(defn- does-ns-refer-to-var? [ns var]
|
|
(ns-resolve ns var))
|
|
|
|
(defn all-vars-who-call [sym]
|
|
(filter
|
|
ifn?
|
|
(filter
|
|
#(identity %)
|
|
(map #(does-var-call-fn % sym)
|
|
(flatten
|
|
(map vals
|
|
(map ns-interns
|
|
(filter #(does-ns-refer-to-var? % sym)
|
|
(all-ns)))))))))
|