1 1d126b15 2021-10-13 op (ns gemini.core
3 1d126b15 2021-10-13 op [clojure.java.io :as io])
6 5243b9ac 2021-10-13 op (com.omarpolo.gemini Request)))
9 0c76ea8b 2021-10-15 op (set! *warn-on-reflection* true)
15 1d126b15 2021-10-13 op (def code-description
16 1d126b15 2021-10-13 op "Human description for every response code."
18 1d126b15 2021-10-13 op 11 "sensitive input"
20 1d126b15 2021-10-13 op 30 "temporary redirect"
21 1d126b15 2021-10-13 op 31 "permanent redirect"
22 1d126b15 2021-10-13 op 40 "temporary failure"
23 1d126b15 2021-10-13 op 41 "server unavailable"
24 1d126b15 2021-10-13 op 42 "CGI error"
25 1d126b15 2021-10-13 op 43 "proxy error"
26 1d126b15 2021-10-13 op 44 "slow down"
27 1d126b15 2021-10-13 op 50 "permanent failure"
28 1d126b15 2021-10-13 op 51 "not found"
30 1d126b15 2021-10-13 op 53 "proxy request refused"
31 1d126b15 2021-10-13 op 59 "bad request"
32 1d126b15 2021-10-13 op 60 "client certificate required"
33 1d126b15 2021-10-13 op 61 "certificate not authorized"
34 1d126b15 2021-10-13 op 62 "certificate not valid"})
36 62094a4e 2021-10-15 op (defn is-input? [{c :code}] (= 1 (quot c 10)))
37 62094a4e 2021-10-15 op (defn is-success? [{c :code}] (= 2 (quot c 10)))
38 62094a4e 2021-10-15 op (defn is-redirect? [{c :code}] (= 3 (quot c 10)))
39 62094a4e 2021-10-15 op (defn is-temporary-failure? [{c :code}] (= 4 (quot c 10)))
40 62094a4e 2021-10-15 op (defn is-permanent-failure? [{c :code}] (= 5 (quot c 10)))
41 62094a4e 2021-10-15 op (defn is-client-cert-required? [{c :code}] (= 6 (quot c 10)))
45 0c76ea8b 2021-10-15 op (defn- parse-params [{:keys [proxy request follow-redirects?]}]
46 0c76ea8b 2021-10-15 op (when (and (:host proxy)
47 0c76ea8b 2021-10-15 op (not (:port proxy)))
48 0c76ea8b 2021-10-15 op (throw (ex-info "invalid proxy definition" {:got proxy
49 0c76ea8b 2021-10-15 op :reason "missing proxy host"})))
50 0c76ea8b 2021-10-15 op {:host (:host proxy)
51 0c76ea8b 2021-10-15 op :port (:port proxy)
52 0c76ea8b 2021-10-15 op :request (or request
53 0c76ea8b 2021-10-15 op (throw (ex-info ":request is nil" {})))
54 0c76ea8b 2021-10-15 op :follow-redirects? (case follow-redirects?
58 0c76ea8b 2021-10-15 op follow-redirects?)})
60 0c76ea8b 2021-10-15 op (defn- resolve-uri [request meta]
61 0c76ea8b 2021-10-15 op (let [uri (URI. request)
62 0c76ea8b 2021-10-15 op rel (URI. meta)]
63 0c76ea8b 2021-10-15 op (str (.resolve uri rel))))
65 0c76ea8b 2021-10-15 op (defn- fetch' [host port uri]
67 0c76ea8b 2021-10-15 op (let [req (cond
68 0c76ea8b 2021-10-15 op host (Request. ^String host ^int port ^String uri)
69 0c76ea8b 2021-10-15 op :else (Request. ^String uri))]
72 0c76ea8b 2021-10-15 op :code (.getCode req)
73 0c76ea8b 2021-10-15 op :meta (.getMeta req)
74 0c76ea8b 2021-10-15 op :body (.body req)})
75 0c76ea8b 2021-10-15 op (catch Throwable e
79 0c76ea8b 2021-10-15 op "Make a gemini request. `params` is a map with the following
80 0c76ea8b 2021-10-15 op keys (only `:request` is mandatory):
82 0c76ea8b 2021-10-15 op - `:proxy`: a map of `:host` and `:port`, identifies the server to
83 0c76ea8b 2021-10-15 op send the requests to. This allows to use a gemini server as a
84 0c76ea8b 2021-10-15 op proxy, it doesn't do any other kind of proxying (e.g. SOCK5.)
86 0c76ea8b 2021-10-15 op - `:request` the URI (as string) to require.
88 0c76ea8b 2021-10-15 op - `:follow-redirects?` if `false` or `nil` don't follow redirects,
89 0c76ea8b 2021-10-15 op if `true` follow up to 5 redirects, or the number of redirects to
92 0c76ea8b 2021-10-15 op Return a map with `:request`, `:code`, `:meta`, `:body` on success
93 0c76ea8b 2021-10-15 op or `:error` on failure. The request needs to be closed when done
94 0c76ea8b 2021-10-15 op usign `close`."
96 0c76ea8b 2021-10-15 op (let [{:keys [host port request follow-redirects?] :as orig} (parse-params params)]
97 0c76ea8b 2021-10-15 op (loop [n follow-redirects?
98 0c76ea8b 2021-10-15 op request request
99 0c76ea8b 2021-10-15 op redirected? false]
100 0c76ea8b 2021-10-15 op (let [res (fetch' host port request)
101 0c76ea8b 2021-10-15 op redirect? (and (not (:error res))
102 0c76ea8b 2021-10-15 op (is-redirect? res))]
104 0c76ea8b 2021-10-15 op (:error res) res
105 0c76ea8b 2021-10-15 op (= follow-redirects? 0) res
107 0c76ea8b 2021-10-15 op redirect?) (do (.close ^Request (:request res))
108 0c76ea8b 2021-10-15 op (throw (ex-info "too many redirects"
109 0c76ea8b 2021-10-15 op {:original orig
110 0c76ea8b 2021-10-15 op :redirects follow-redirects?
111 0c76ea8b 2021-10-15 op :code (:code res)
112 0c76ea8b 2021-10-15 op :meta (:meta res)})))
113 0c76ea8b 2021-10-15 op redirect? (do (.close ^Request (:request res))
114 0c76ea8b 2021-10-15 op (recur (dec n)
115 0c76ea8b 2021-10-15 op (resolve-uri request (:meta res))
117 9142afc2 2021-10-15 op :else (assoc res :redirected? redirected?))))))
119 0c76ea8b 2021-10-15 op (defn body-as-string!
120 0c76ea8b 2021-10-15 op "Read all the response into a strings and returns it. The request
121 0c76ea8b 2021-10-15 op will be closed."
122 0c76ea8b 2021-10-15 op [{r :request}]
123 0c76ea8b 2021-10-15 op (let [sw (java.io.StringWriter.)]
124 0c76ea8b 2021-10-15 op (with-open [r ^Request r]
125 0c76ea8b 2021-10-15 op (io/copy (.body r) sw)
126 0c76ea8b 2021-10-15 op (.toString sw))))
129 0c76ea8b 2021-10-15 op "Close a request."
130 0c76ea8b 2021-10-15 op [{r :request}]
131 0c76ea8b 2021-10-15 op (.close ^Request r))
133 0c76ea8b 2021-10-15 op (defmacro with-request
134 0c76ea8b 2021-10-15 op "Make a request, eval `body` when it succeed and automatically close
135 0c76ea8b 2021-10-15 op the request, or throw an exception if the request fails."
136 0c76ea8b 2021-10-15 op [[var req] & body]
137 0c76ea8b 2021-10-15 op `(let [~var ~req]
138 0c76ea8b 2021-10-15 op (when-let [e# (:error ~var)]
140 0c76ea8b 2021-10-15 op (with-open [req# (:request ~var)]
