walk-ast.clj

(ns walk-ast
  (:require [com.rpl.specter :refer :all]
            [clojure.tools.analyzer.jvm :refer [analyze]]
            [fipp.edn :refer [pprint] :rename {pprint fipp}]))

(def map-key-walker
  (recursive-path
   [akey] p
   (cond-path map? [ALL (if-path [FIRST #(= % akey)]
                                 LAST
                                 [LAST p])]
              vector? [ALL (if-path map?
                                    [ALL (if-path [FIRST #(= % akey)]
                                                  LAST 
                                                  [LAST p])])])))

(defn select-locals [ast] (select (map-key-walker :locals) ast))

(def MAP-NODES
  (recursive-path
   [] p
   (cond-path map? (continue-then-stay MAP-VALS p)
              coll? [ALL p])))

(defn keep-locals [ast] (transform MAP-NODES #(select-keys % [:locals]) ast))

(def data (analyze '(fn [x y] (fn [a b] (+ x a)))))
(def unstructured (select-locals data))
(def structured (keep-locals data)) ;; => {}