147 lines
5.2 KiB
Hy
Executable file
147 lines
5.2 KiB
Hy
Executable file
#!/usr/bin/env hy
|
|
;; Hypo -- Quickly share stuff
|
|
;; Copyright (C) 2013 Tom Willemse
|
|
|
|
;; Hypo is free software: you can redistribute it and/or modify it
|
|
;; under the terms of the GNU Affero General Public License as
|
|
;; published by the Free Software Foundation, either version 3 of the
|
|
;; License, or (at your option) any later version.
|
|
|
|
;; Hypo is distributed in the hope that it will be useful, but WITHOUT
|
|
;; ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
;; or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General
|
|
;; Public License for more details.
|
|
|
|
;; You should have received a copy of the GNU Affero General Public
|
|
;; License along with Hypo. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
(import web sys os hashlib datetime
|
|
[pygments [highlight]]
|
|
[pygments.lexers [get-lexer-by-name guess-lexer-for-filename]]
|
|
[pygments.formatters [HtmlFormatter]]
|
|
[pygments.util [ClassNotFound]])
|
|
|
|
(try (import [config [*]])
|
|
(catch [ImportError]
|
|
(print "Please copy the config.example.hy to config.hy and set"
|
|
"the values to your preference.")
|
|
(sys.exit 1)))
|
|
|
|
(def render (web.template.render "templates/"))
|
|
(def urls (, (+ "/" *prefix*) "index"
|
|
(+ "/" *prefix* "raw/(.*)") "raw"
|
|
(+ "/" *prefix* "dl/(.*)") "download"
|
|
(+ "/" *prefix* "([a-f0-9]{7})$") "html"
|
|
(+ "/" *prefix* "(.*)") "upload"))
|
|
(def db
|
|
(kwapply (web.database)
|
|
{"dbn" "postgres" "user" *dbuser* "pw" *dbpw* "db" *dbname*}))
|
|
|
|
(defun hashes [name]
|
|
(let ((hasher (hashlib.sha1)))
|
|
(hasher.update name)
|
|
(hasher.update (str (datetime.datetime.now)))
|
|
(let ((digest (hasher.hexdigest)))
|
|
(, (slice digest 0 7) digest))))
|
|
|
|
(defun get-type [ext]
|
|
(cond
|
|
((in ext (, ".jpg" ".jpeg" ".png" ".gif")) "image")
|
|
(true "text")))
|
|
|
|
(defun read-file [filename]
|
|
(let (res)
|
|
(with [f (file filename "r")]
|
|
(setv res (f.read)))
|
|
res))
|
|
|
|
(defun no-such-file []
|
|
(setv web.ctx.status (str "404 Not Found"))
|
|
"No such file.\n")
|
|
|
|
(defun get-file [name]
|
|
(let ((res (kwapply (db.select "hfile" {"shash" name})
|
|
{"where" "shash = $shash"})))
|
|
(if res (car res))))
|
|
|
|
(defun get-lexer [filename content]
|
|
"Try to guess the correct lexer by FILENAME and CONTENT.
|
|
|
|
If no lexer is found fallback onto the text lexer."
|
|
(try (guess-lexer-for-filename filename content)
|
|
(catch [ClassNotFound]
|
|
(get-lexer-by-name "text"))))
|
|
|
|
(defclass raw []
|
|
[[GET (lambda [self name]
|
|
(let ((filename (+ "files/" name))
|
|
(resp (if (os.path.exists filename)
|
|
(read-file filename))))
|
|
(or resp (no-such-file))))]])
|
|
|
|
(defclass download []
|
|
[[GET (lambda [self name]
|
|
(let ((hfile (get-file name))
|
|
(filename (+ "files/" name)))
|
|
(if (and hfile (os.path.exists filename))
|
|
(progn
|
|
(web.header "Content-Disposition"
|
|
(+ "attachment; filename=\""
|
|
hfile.filename "\""))
|
|
(read-file filename))
|
|
(no-such-file))))]])
|
|
|
|
(defclass html []
|
|
[[GET (lambda [self name]
|
|
(let ((hfile (get-file name))
|
|
(filename (+ "files/" name)))
|
|
(if (and hfile (os.path.exists filename))
|
|
(cond
|
|
((= hfile.type "text")
|
|
(progn
|
|
(let ((content (read-file filename))
|
|
(lexer (get-lexer hfile.filename content))
|
|
(formatter (HtmlFormatter)))
|
|
(kwapply (render.main)
|
|
{"content" (highlight content lexer
|
|
formatter)
|
|
"style" (formatter.get-style-defs
|
|
".highlight")
|
|
"file" hfile})
|
|
)))
|
|
((= hfile.type "image")
|
|
(kwapply (render.main)
|
|
{"content" (kwapply (render.image)
|
|
{"name" name})
|
|
"style" ""
|
|
"file" hfile})))
|
|
(no-such-file))))]
|
|
|
|
[DELETE (lambda [self name]
|
|
(let ((filename (+ "files/" name)))
|
|
(kwapply (db.delete "hfile")
|
|
{"where" (+ "shash = '" name "'")})
|
|
(if (os.path.exists filename)
|
|
(os.remove filename)
|
|
(no-such-file))))]])
|
|
|
|
(defclass upload []
|
|
[[PUT (lambda [self name]
|
|
(let ((h (hashes name)))
|
|
(with [f (file (+ "files/" (get h 0)) "w")]
|
|
(.write f (web.data)))
|
|
(kwapply (db.insert "hfile")
|
|
{"shash" (get h 0)
|
|
"hash" (get h 1)
|
|
"filename" name
|
|
"type" (get-type (get (os.path.splitext name) 1))})
|
|
(setv web.ctx.status (str "201 Created"))
|
|
(+ web.ctx.home "/" *prefix* (get h 0) "\n")))]])
|
|
|
|
(defclass index []
|
|
[[GET (lambda [self] (render.index))]])
|
|
|
|
(defun hypo-start [argv]
|
|
(let ((sys.argv (cdr sys.argv))
|
|
(app (web.application urls (globals))))
|
|
(.run app)))
|