entity-pull: a datomic.api/pull that returns values the way datomic.api/entity would, with sets and ident keywords

entity-pull.clj

(require '[datomic.api :as d])

(defn entity-pull
  "Like pull, but returns values consistent with d/entity, i.e.,
  entities with :db/ident are represented as keywords and sets are
  used instead of vectors."
  [db pat eid]
  (->> (d/pull db pat eid)
       (clojure.walk/prewalk
         (fn [x]
           (cond
             (and (not (map-entry? x)) (vector? x)) (set x)
             (and (map? x) (:db/ident x)) (:db/ident x)
             (and (map? x) (:db/id x)) (or
                                         (:db/ident (d/entity db (:db/id x)))
                                         x)
             :else x)))))