summaryrefslogtreecommitdiff
path: root/src/utils.clj
blob: ea696e5258e4b3c61546781bfe4d3707b43e1aec (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
(ns utils
  (:import java.text.SimpleDateFormat
           java.util.Date
           java.net.URLDecoder
           org.antlr.stringtemplate.StringTemplateGroup)
  (:use clojure.contrib.json.write
        clojure.contrib.sql))

(let [db-host "localhost"
      db-port 5432
      db-name "dumpfm"]
  (def *db* {:classname "org.postgresql.Driver"
             :subprotocol "postgresql"
             :subname (str "//" db-host ":" db-port "/" db-name)
             :user "postgres"
             :password "root"}))

;;  moved this to here which doesn't seem right... maybe a 'settings.clj' or something?
(def *dumps-per-page* 20)

;; Misc

(defn ms-in-future [ms]
  (+ ms (System/currentTimeMillis)))

(defn swap [f]
  (fn [& more] (apply f (reverse more))))

(def YYYYMMDD-format (new SimpleDateFormat "yyyyMMdd"))

(defn today []
  (.format YYYYMMDD-format (new Date)))

(def formatter (new SimpleDateFormat "h:mm a EEE M/d"))

(defn non-empty-string? [s]
  (cond (string? s) (> (count s) 0)
        :else s))

(defn seconds [t] (* t 1000))
(defn minutes [t] (* t 60 1000))

(defn kbytes [b] (* b 1024))
(defn mbytes [b] (* b 1024 1024))

;; JSON responses

(def yyyy-mm-dd-formatter (new SimpleDateFormat "yyyy-MM-dd"))

(defmethod print-json Date
  [d]
  (print-json (.format yyyy-mm-dd-formatter d)))

(defn resp-error [message]
  {:status 400 :headers {} :body message})

(defn resp-success [message]
  {:status 200 :headers {} :body (json-str message)})

;; Database

(defn do-select [query]
  (with-connection *db*
    (with-query-results rs query
      (doall rs))))

(defn do-count [query]
  ((first (with-connection *db*
	    (with-query-results rs query
	      (doall rs))))
   :count))

(defn do-delete [table query]
  (with-connection *db*
    (delete-rows table query)))

(defn do-insert [table cols values]
  (with-connection *db*
    (insert-values table cols values)))

;; Parsing

(defn maybe-parse-int 
  ([s] (Integer/parseInt s))
  ([s default]
     (try (Integer/parseInt s)
          (catch NumberFormatException _ default))))

(defn maybe-parse-long [s f]
  (if s (Long/parseLong s) f))

(defn url-decode [text]
  (URLDecoder/decode text "UTF-8"))

(defn #^String lower-case
  "Converts string to all lower-case."
  [#^String s]
  (.toLowerCase s))

;; 404

(defn unknown-page [& more]
  [404 "Page not Found"])

;; Templates

(def template-group (new StringTemplateGroup "dumpfm" "template"))
(.setRefreshInterval template-group 3)

;; TODO: handle exception
(defn fetch-template [template session]
  (let [st (.getInstanceOf template-group template)]
    (if (session :nick)
      (do (.setAttribute st "user_email" (session :email))
          (.setAttribute st "user_nick" (session :nick))
          (if (non-empty-string? (session :avatar)) (.setAttribute st "user_avatar" (session :avatar)))
          (.setAttribute st "isadmin" (session :is_admin)))) ;; TODO: consolidate session/user code
    st))

(defn serve-template [template session]
  (.toString (fetch-template template session)))


;; User authentication
; TODO: create user module

(defn first-or-nil [l]
  (if (empty? l) nil (first l)))

(def nick-regex #"^[A-Za-z0-9\-_∆˚†]*$")

(defn is-invalid-nick? [n]
  (cond
   (< (count n) 3) "NICK_TOO_SHORT"
   (not (re-matches nick-regex n)) "NICK_INVALID_CHARS"))

(defn check-nick [nick]
  (let [query "SELECT * FROM users WHERE LOWER(nick) = ? LIMIT 1"]
    (> (count (do-select [query (lower-case nick)])) 0)))

(defn fetch-nick [nick]
  (let [q1 "SELECT * FROM users WHERE nick = ? LIMIT 1"
        ; ORDER BY ensures consistent retrieval of ambiguious names
        q2 "SELECT * FROM users WHERE lower(nick) = ? ORDER BY nick LIMIT 1"]
    (or (first-or-nil (do-select [q1 nick]))
        (first-or-nil (do-select [q2 (lower-case nick)])))))

(defn authorize-nick-hash [nick hash]
  (let [db-user (fetch-nick nick)]
    (and db-user (= (db-user :hash) hash) db-user)))

(defn is-vip? [session]
  (session :is_admin))

(defmacro if-vip [e]
  "Evaluates expr if user is vipm otherwise returns 404 string. Can only be used
   where session is defined."
  `(if (is-vip? ~'session) ~e (unknown-page)))