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 d4243d62 2021-10-15 op If the returned map contains an `:error` key the request failed for
93 d4243d62 2021-10-15 op the specified reason. Otherwise, the map contains the following keys:
95 d4243d62 2021-10-15 op - `:uri`: the URI of the request. May be different from the
96 d4243d62 2021-10-15 op requested one if `:follow-redirects?` was specified.
98 d4243d62 2021-10-15 op - `:request`: the object backing the request.
100 d4243d62 2021-10-15 op - `:code` and `:meta` are the parsed header response.
102 d4243d62 2021-10-15 op - `:body` an instance of a `BufferedReader`. Note: closing the body
103 d4243d62 2021-10-15 op is not enugh, always call `clase` on the returned map.
105 d4243d62 2021-10-15 op - `:redirected?` true if a redirect was followed."
107 0c76ea8b 2021-10-15 op (let [{:keys [host port request follow-redirects?] :as orig} (parse-params params)]
108 0c76ea8b 2021-10-15 op (loop [n follow-redirects?
109 0c76ea8b 2021-10-15 op request request
110 0c76ea8b 2021-10-15 op redirected? false]
111 0c76ea8b 2021-10-15 op (let [res (fetch' host port request)
112 0c76ea8b 2021-10-15 op redirect? (and (not (:error res))
113 0c76ea8b 2021-10-15 op (is-redirect? res))]
115 0c76ea8b 2021-10-15 op (:error res) res
116 0c76ea8b 2021-10-15 op (= follow-redirects? 0) res
118 0c76ea8b 2021-10-15 op redirect?) (do (.close ^Request (:request res))
119 0c76ea8b 2021-10-15 op (throw (ex-info "too many redirects"
120 0c76ea8b 2021-10-15 op {:original orig
121 0c76ea8b 2021-10-15 op :redirects follow-redirects?
122 0c76ea8b 2021-10-15 op :code (:code res)
123 0c76ea8b 2021-10-15 op :meta (:meta res)})))
124 0c76ea8b 2021-10-15 op redirect? (do (.close ^Request (:request res))
125 0c76ea8b 2021-10-15 op (recur (dec n)
126 0c76ea8b 2021-10-15 op (resolve-uri request (:meta res))
128 9142afc2 2021-10-15 op :else (assoc res :redirected? redirected?))))))
130 0c76ea8b 2021-10-15 op (defn body-as-string!
131 0c76ea8b 2021-10-15 op "Read all the response into a strings and returns it. The request
132 0c76ea8b 2021-10-15 op will be closed."
133 0c76ea8b 2021-10-15 op [{r :request}]
134 0c76ea8b 2021-10-15 op (let [sw (java.io.StringWriter.)]
135 0c76ea8b 2021-10-15 op (with-open [r ^Request r]
136 0c76ea8b 2021-10-15 op (io/copy (.body r) sw)
137 0c76ea8b 2021-10-15 op (.toString sw))))
140 0c76ea8b 2021-10-15 op "Close a request."
141 0c76ea8b 2021-10-15 op [{r :request}]
142 0c76ea8b 2021-10-15 op (.close ^Request r))
144 0c76ea8b 2021-10-15 op (defmacro with-request
145 0c76ea8b 2021-10-15 op "Make a request, eval `body` when it succeed and automatically close
146 0c76ea8b 2021-10-15 op the request, or throw an exception if the request fails."
147 05d91d67 2021-10-15 op [[var spec] & body]
148 05d91d67 2021-10-15 op `(let [~var (fetch ~spec)]
149 0c76ea8b 2021-10-15 op (when-let [e# (:error ~var)]
151 eca2fb1a 2021-10-15 op (with-open [req# ^Request (:request ~var)]
