http.go (view raw)
1package main
2
3import (
4 "bytes"
5 "database/sql"
6 "fmt"
7 gmi "git.sr.ht/~adnano/go-gemini"
8 "github.com/gorilla/handlers"
9 "github.com/gorilla/sessions"
10 _ "github.com/mattn/go-sqlite3"
11 "golang.org/x/crypto/bcrypt"
12 "html/template"
13 "io"
14 "io/ioutil"
15 "log"
16 "mime"
17 "net/http"
18 "os"
19 "path"
20 "path/filepath"
21 "strings"
22 "time"
23)
24
25var t *template.Template
26var DB *sql.DB
27var SessionStore *sessions.CookieStore
28
29const InternalServerErrorMsg = "500: Internal Server Error"
30
31func renderError(w http.ResponseWriter, errorMsg string, statusCode int) {
32 data := struct {
33 PageTitle string
34 ErrorMsg string
35 }{"Error!", errorMsg}
36 err := t.ExecuteTemplate(w, "error.html", data)
37 if err != nil { // shouldn't happen probably
38 http.Error(w, errorMsg, statusCode)
39 }
40}
41
42func rootHandler(w http.ResponseWriter, r *http.Request) {
43 // serve everything inside static directory
44 if r.URL.Path != "/" {
45 fileName := path.Join(c.TemplatesDirectory, "static", filepath.Clean(r.URL.Path))
46 http.ServeFile(w, r, fileName)
47 return
48 }
49 authd, _, isAdmin := getAuthUser(r)
50 indexFiles, err := getIndexFiles()
51 if err != nil {
52 log.Println(err)
53 renderError(w, InternalServerErrorMsg, 500)
54 return
55 }
56 allUsers, err := getActiveUserNames()
57 if err != nil {
58 log.Println(err)
59 renderError(w, InternalServerErrorMsg, 500)
60 return
61 }
62 data := struct {
63 Host string
64 PageTitle string
65 Files []*File
66 Users []string
67 LoggedIn bool
68 IsAdmin bool
69 }{c.Host, c.SiteTitle, indexFiles, allUsers, authd, isAdmin}
70 err = t.ExecuteTemplate(w, "index.html", data)
71 if err != nil {
72 log.Println(err)
73 renderError(w, InternalServerErrorMsg, 500)
74 return
75 }
76}
77
78func editFileHandler(w http.ResponseWriter, r *http.Request) {
79 session, _ := SessionStore.Get(r, "cookie-session")
80 authUser, ok := session.Values["auth_user"].(string)
81 if !ok {
82 renderError(w, "403: Forbidden", 403)
83 return
84 }
85 fileName := filepath.Clean(r.URL.Path[len("/edit/"):])
86 isText := strings.HasPrefix(mime.TypeByExtension(path.Ext(fileName)), "text")
87 if !isText {
88 renderError(w, "Not a text file, cannot be edited here", 400) // correct status code?
89 return
90 }
91 filePath := path.Join(c.FilesDirectory, authUser, fileName)
92
93 if r.Method == "GET" {
94 err := checkIfValidFile(filePath, nil)
95 if err != nil {
96 log.Println(err)
97 renderError(w, err.Error(), 400)
98 return
99 }
100 // create directories if dne
101 f, err := os.OpenFile(filePath, os.O_RDONLY, 0644)
102 var fileBytes []byte
103 if os.IsNotExist(err) {
104 fileBytes = []byte{}
105 err = nil
106 } else {
107 defer f.Close()
108 fileBytes, err = ioutil.ReadAll(f)
109 }
110 if err != nil {
111 log.Println(err)
112 renderError(w, InternalServerErrorMsg, 500)
113 return
114 }
115 data := struct {
116 FileName string
117 FileText string
118 PageTitle string
119 AuthUser string
120 Host string
121 }{fileName, string(fileBytes), c.SiteTitle, authUser, c.Host}
122 err = t.ExecuteTemplate(w, "edit_file.html", data)
123 if err != nil {
124 log.Println(err)
125 renderError(w, InternalServerErrorMsg, 500)
126 return
127 }
128 } else if r.Method == "POST" {
129 // get post body
130 r.ParseForm()
131 fileBytes := []byte(r.Form.Get("file_text"))
132 err := checkIfValidFile(filePath, fileBytes)
133 if err != nil {
134 log.Println(err)
135 renderError(w, err.Error(), 400)
136 return
137 }
138 // create directories if dne
139 os.MkdirAll(path.Dir(filePath), os.ModePerm)
140 err = ioutil.WriteFile(filePath, fileBytes, 0644)
141 if err != nil {
142 log.Println(err)
143 renderError(w, InternalServerErrorMsg, 500)
144 return
145 }
146 newName := filepath.Clean(r.Form.Get("rename"))
147 err = checkIfValidFile(newName, fileBytes)
148 if err != nil {
149 log.Println(err)
150 renderError(w, err.Error(), 400)
151 return
152 }
153 if newName != fileName {
154 newPath := path.Join(c.FilesDirectory, authUser, newName)
155 os.MkdirAll(path.Dir(newPath), os.ModePerm)
156 os.Rename(filePath, newPath)
157 }
158 http.Redirect(w, r, path.Join("/edit", fileName), 303)
159 }
160}
161
162func uploadFilesHandler(w http.ResponseWriter, r *http.Request) {
163 if r.Method == "POST" {
164 session, _ := SessionStore.Get(r, "cookie-session")
165 authUser, ok := session.Values["auth_user"].(string)
166 if !ok {
167 renderError(w, "403: Forbidden", 403)
168 return
169 }
170 r.ParseMultipartForm(10 << 6) // why does this not work
171 file, fileHeader, err := r.FormFile("file")
172 fileName := filepath.Clean(fileHeader.Filename)
173 defer file.Close()
174 if err != nil {
175 log.Println(err)
176 renderError(w, err.Error(), 400)
177 return
178 }
179 dest, _ := ioutil.ReadAll(file)
180 err = checkIfValidFile(fileName, dest)
181 if err != nil {
182 log.Println(err)
183 renderError(w, err.Error(), 400)
184 return
185 }
186 destPath := path.Join(c.FilesDirectory, authUser, fileName)
187
188 f, err := os.OpenFile(destPath, os.O_WRONLY|os.O_CREATE, 0644)
189 if err != nil {
190 log.Println(err)
191 renderError(w, InternalServerErrorMsg, 500)
192 return
193 }
194 defer f.Close()
195 io.Copy(f, bytes.NewReader(dest))
196 }
197 http.Redirect(w, r, "/my_site", 303)
198}
199
200// bool whether auth'd, string is auth user
201func getAuthUser(r *http.Request) (bool, string, bool) {
202 session, _ := SessionStore.Get(r, "cookie-session")
203 user, ok := session.Values["auth_user"].(string)
204 isAdmin, _ := session.Values["admin"].(bool)
205 return ok, user, isAdmin
206}
207func deleteFileHandler(w http.ResponseWriter, r *http.Request) {
208 authd, authUser, _ := getAuthUser(r)
209 if !authd {
210 renderError(w, "403: Forbidden", 403)
211 return
212 }
213 fileName := filepath.Clean(r.URL.Path[len("/delete/"):])
214 filePath := path.Join(c.FilesDirectory, authUser, fileName)
215 if r.Method == "POST" {
216 os.Remove(filePath) // suppress error
217 }
218 http.Redirect(w, r, "/my_site", 303)
219}
220
221func mySiteHandler(w http.ResponseWriter, r *http.Request) {
222 authd, authUser, isAdmin := getAuthUser(r)
223 if !authd {
224 renderError(w, "403: Forbidden", 403)
225 return
226 }
227 // check auth
228 userFolder := path.Join(c.FilesDirectory, authUser)
229 files, _ := getMyFilesRecursive(userFolder, authUser)
230 data := struct {
231 Host string
232 PageTitle string
233 AuthUser string
234 Files []*File
235 LoggedIn bool
236 IsAdmin bool
237 }{c.Host, c.SiteTitle, authUser, files, authd, isAdmin}
238 _ = t.ExecuteTemplate(w, "my_site.html", data)
239}
240
241func archiveHandler(w http.ResponseWriter, r *http.Request) {
242 authd, authUser, _ := getAuthUser(r)
243 if !authd {
244 renderError(w, "403: Forbidden", 403)
245 return
246 }
247 if r.Method == "GET" {
248 userFolder := filepath.Join(c.FilesDirectory, filepath.Clean(authUser))
249 err := zipit(userFolder, w)
250 if err != nil {
251 log.Println(err)
252 renderError(w, InternalServerErrorMsg, 500)
253 return
254 }
255
256 }
257}
258func loginHandler(w http.ResponseWriter, r *http.Request) {
259 if r.Method == "GET" {
260 // show page
261 data := struct {
262 Error string
263 PageTitle string
264 }{"", "Login"}
265 err := t.ExecuteTemplate(w, "login.html", data)
266 if err != nil {
267 log.Println(err)
268 renderError(w, InternalServerErrorMsg, 500)
269 return
270 }
271 } else if r.Method == "POST" {
272 r.ParseForm()
273 name := r.Form.Get("username")
274 password := r.Form.Get("password")
275 row := DB.QueryRow("SELECT username, password_hash, active, admin FROM user where username = $1 OR email = $1", name)
276 var db_password []byte
277 var username string
278 var active bool
279 var isAdmin bool
280 _ = row.Scan(&username, &db_password, &active, &isAdmin)
281 if db_password != nil && !active {
282 data := struct {
283 Error string
284 PageTitle string
285 }{"Your account is not active yet. Pending admin approval", c.SiteTitle}
286 t.ExecuteTemplate(w, "login.html", data)
287 return
288 }
289 if bcrypt.CompareHashAndPassword(db_password, []byte(password)) == nil {
290 log.Println("logged in")
291 session, _ := SessionStore.Get(r, "cookie-session")
292 session.Values["auth_user"] = username
293 session.Values["admin"] = isAdmin
294 session.Save(r, w)
295 http.Redirect(w, r, "/my_site", 303)
296 } else {
297 data := struct {
298 Error string
299 PageTitle string
300 }{"Invalid login or password", c.SiteTitle}
301 err := t.ExecuteTemplate(w, "login.html", data)
302 if err != nil {
303 log.Println(err)
304 renderError(w, InternalServerErrorMsg, 500)
305 return
306 }
307 }
308 }
309}
310
311func logoutHandler(w http.ResponseWriter, r *http.Request) {
312 session, _ := SessionStore.Get(r, "cookie-session")
313 session.Options.MaxAge = -1
314 session.Save(r, w)
315 http.Redirect(w, r, "/", 303)
316}
317
318const ok = "-0123456789abcdefghijklmnopqrstuvwxyz"
319
320func isOkUsername(s string) bool {
321 if len(s) < 1 {
322 return false
323 }
324 if len(s) > 31 {
325 return false
326 }
327 for _, char := range s {
328 if !strings.Contains(ok, strings.ToLower(string(char))) {
329 return false
330 }
331 }
332 return true
333}
334func registerHandler(w http.ResponseWriter, r *http.Request) {
335 if r.Method == "GET" {
336 data := struct {
337 Host string
338 Errors []string
339 PageTitle string
340 }{c.Host, nil, "Register"}
341 err := t.ExecuteTemplate(w, "register.html", data)
342 if err != nil {
343 log.Println(err)
344 renderError(w, InternalServerErrorMsg, 500)
345 return
346 }
347 } else if r.Method == "POST" {
348 r.ParseForm()
349 email := r.Form.Get("email")
350 password := r.Form.Get("password")
351 errors := []string{}
352 if !strings.Contains(email, "@") {
353 errors = append(errors, "Invalid Email")
354 }
355 if r.Form.Get("password") != r.Form.Get("password2") {
356 errors = append(errors, "Passwords don't match")
357 }
358 if len(password) < 6 {
359 errors = append(errors, "Password is too short")
360 }
361 username := strings.ToLower(r.Form.Get("username"))
362 if !isOkUsername(username) {
363 errors = append(errors, "Username is invalid: can only contain letters, numbers and hypens. Maximum 32 characters.")
364 }
365 hashedPassword, err := bcrypt.GenerateFromPassword([]byte(password), 8) // TODO handle error
366 reference := r.Form.Get("reference")
367 if len(errors) == 0 {
368 _, err = DB.Exec("insert into user (username, email, password_hash, reference) values ($1, $2, $3, $4)", username, email, string(hashedPassword), reference)
369 if err != nil {
370 log.Println(err)
371 errors = append(errors, "Username or email is already used")
372 }
373 }
374 if len(errors) > 0 {
375 data := struct {
376 Host string
377 Errors []string
378 PageTitle string
379 }{c.Host, errors, "Register"}
380 t.ExecuteTemplate(w, "register.html", data)
381 } else {
382 data := struct {
383 Host string
384 Message string
385 PageTitle string
386 }{c.Host, "Registration complete! The server admin will approve your request before you can log in.", "Registration Complete"}
387 t.ExecuteTemplate(w, "message.html", data)
388 }
389 }
390}
391
392func adminHandler(w http.ResponseWriter, r *http.Request) {
393 _, _, isAdmin := getAuthUser(r)
394 if !isAdmin {
395 renderError(w, "403: Forbidden", 403)
396 return
397 }
398 allUsers, err := getUsers()
399 if err != nil {
400 log.Println(err)
401 renderError(w, InternalServerErrorMsg, 500)
402 return
403 }
404 data := struct {
405 Users []User
406 LoggedIn bool
407 IsAdmin bool
408 PageTitle string
409 Host string
410 }{allUsers, true, true, "Admin", c.Host}
411 err = t.ExecuteTemplate(w, "admin.html", data)
412 if err != nil {
413 log.Println(err)
414 renderError(w, InternalServerErrorMsg, 500)
415 return
416 }
417}
418
419func getFavicon(user string) string {
420 faviconPath := path.Join(c.FilesDirectory, filepath.Clean(user), "favicon.txt")
421 content, err := ioutil.ReadFile(faviconPath)
422 if err != nil {
423 return ""
424 }
425 strcontent := []rune(string(content))
426 if len(strcontent) > 0 {
427 return string(strcontent[0])
428 }
429 return ""
430}
431
432// Server a user's file
433func userFile(w http.ResponseWriter, r *http.Request) {
434 userName := filepath.Clean(strings.Split(r.Host, ".")[0]) // clean probably unnecessary
435 query := r.URL.Query()
436 p := filepath.Clean(r.URL.Path)
437 // chcek if is directory for index.gmi file
438 var isDir bool
439 fileName := path.Join(c.FilesDirectory, userName, p)
440 stat, _ := os.Stat(fileName)
441 if stat != nil {
442 isDir = stat.IsDir()
443 }
444 if p == "/" || isDir {
445 fileName = path.Join(fileName, "index.gmi")
446 } else if strings.HasPrefix(p, "/.hidden") {
447 renderError(w, "404: file not found", 404)
448 return
449 }
450 extension := path.Ext(fileName)
451 if r.URL.Path == "/style.css" {
452 http.ServeFile(w, r, path.Join(c.TemplatesDirectory, "static/style.css"))
453 }
454 _, err := os.Stat(fileName)
455 if err != nil {
456 renderError(w, "404: file not found", 404)
457 return
458 }
459 // dumb content negotiation
460 _, raw := query["raw"]
461 acceptsGemini := strings.Contains(r.Header.Get("Accept"), "text/gemini")
462 if !raw && !acceptsGemini && (extension == ".gmi" || extension == ".gemini") {
463
464 file, _ := os.Open(fileName)
465
466 htmlString := textToHTML(gmi.ParseText(file))
467 favicon := getFavicon(userName)
468 data := struct {
469 SiteBody template.HTML
470 Favicon string
471 PageTitle string
472 }{template.HTML(htmlString), favicon, userName + p}
473 t.ExecuteTemplate(w, "user_page.html", data)
474 } else {
475 http.ServeFile(w, r, fileName) // TODO make errors pretty
476 }
477}
478
479func adminUserHandler(w http.ResponseWriter, r *http.Request) {
480 _, _, isAdmin := getAuthUser(r)
481 if r.Method == "POST" {
482 if !isAdmin {
483 renderError(w, "403: Forbidden", 403)
484 return
485 }
486 components := strings.Split(r.URL.Path, "/")
487 if len(components) < 5 {
488 renderError(w, "Invalid action", 400)
489 return
490 }
491 userName := components[3]
492 action := components[4]
493 var err error
494 if action == "activate" {
495 err = activateUser(userName)
496 } else if action == "delete" {
497 err = deleteUser(userName)
498 }
499 if err != nil {
500 log.Println(err)
501 renderError(w, InternalServerErrorMsg, 500)
502 return
503 }
504 http.Redirect(w, r, "/admin", 303)
505 }
506}
507
508func runHTTPServer() {
509 log.Printf("Running http server with hostname %s on port %d. TLS enabled: %t", c.Host, c.HttpPort, c.HttpsEnabled)
510 var err error
511 t, err = template.ParseGlob(path.Join(c.TemplatesDirectory, "*.html"))
512 if err != nil {
513 log.Fatal(err)
514 }
515 serveMux := http.NewServeMux()
516
517 s := strings.SplitN(c.Host, ":", 2)
518 hostname := s[0]
519 port := c.HttpPort
520
521 serveMux.HandleFunc(hostname+"/", rootHandler)
522 serveMux.HandleFunc(hostname+"/my_site", mySiteHandler)
523 serveMux.HandleFunc(hostname+"/my_site/flounder-archive.zip", archiveHandler)
524 serveMux.HandleFunc(hostname+"/admin", adminHandler)
525 serveMux.HandleFunc(hostname+"/edit/", editFileHandler)
526 serveMux.HandleFunc(hostname+"/upload", uploadFilesHandler)
527 serveMux.HandleFunc(hostname+"/login", loginHandler)
528 serveMux.HandleFunc(hostname+"/logout", logoutHandler)
529 serveMux.HandleFunc(hostname+"/register", registerHandler)
530 serveMux.HandleFunc(hostname+"/delete/", deleteFileHandler)
531
532 // admin commands
533 serveMux.HandleFunc(hostname+"/admin/user/", adminUserHandler)
534
535 // TODO rate limit login https://github.com/ulule/limiter
536
537 wrapped := handlers.LoggingHandler(log.Writer(), serveMux)
538
539 // handle user files based on subdomain
540 serveMux.HandleFunc("/", userFile)
541 // login+register functions
542 srv := &http.Server{
543 ReadTimeout: 5 * time.Second,
544 WriteTimeout: 10 * time.Second,
545 IdleTimeout: 120 * time.Second,
546 Addr: fmt.Sprintf(":%d", port),
547 // TLSConfig: tlsConfig,
548 Handler: wrapped,
549 }
550 if c.HttpsEnabled {
551 log.Fatal(srv.ListenAndServeTLS(c.TLSCertFile, c.TLSKeyFile))
552 } else {
553 log.Fatal(srv.ListenAndServe())
554 }
555}