summary refs log blame commit diff stats
path: root/handlers.go
blob: 9f1dd8cd1a6e2f1a2033901d17afb776bfc953df (plain) (tree)
thor Ben Morrison <ben@gbmor.dev> 2019-05-20 04:22:17 -0400 committer Ben Morrison <ben@gbmor.dev> 2019-05-20 04:22:17 -0400 renamed files for clarity' href='/gbmor/getwtxt/commit/handlers.go?h=v0.2.2&id=1168570b071d8bb483e4028f3a8c45a222c93ecb'>1168570 ^
1
2
3
4
5
6
7




                       
                  
                 






                                                           


                                                                                       






                                                                               
                                            









                                                             
                          




                                                               
                          




















                                                                 


                                                    

                                
                                                   

                                   
                                                         

                

                                                        

                                  


                                 
 

                                                      
                                               
 
                              
                       








                                                                     
                         




                                                                 




                                                 
 

                                  

                                                      
                                               
 
                              
                       












                                                             
                                    
                                                        




                                  
                                                         



                                 
                                    
                                                         






                                  
                       


                                  

                                                      
                                               
 









                                           
                                           

                                                         
                                                                                   
                                                                                     



                                               


                                          






                                 
package main

import (
	"crypto/sha256"
	"fmt"
	"net/http"
	"strings"

	"github.com/gorilla/mux"
)

// handles "/"
func indexHandler(w http.ResponseWriter, r *http.Request) {

	pingAssets()

	etag := fmt.Sprintf("%x", sha256.Sum256([]byte(staticCache.indexMod.String())))

	// Take the hex-encoded sha256 sum of the index template's mod time
	// to send as an ETag. If an error occured when grabbing the file info,
	// the ETag will be empty.
	w.Header().Set("ETag", "\""+etag+"\"")
	w.Header().Set("Content-Type", htmlutf8)

	_, err := w.Write(staticCache.index)
	if err != nil {
		log500(w, r, err)
		return
	}

	log200(r)
}

// handles "/api"
func apiBaseHandler(w http.ResponseWriter, r *http.Request) {
	indexHandler(w, r)
}

// handles "/api/plain"
// maybe add json/xml support later
func apiFormatHandler(w http.ResponseWriter, r *http.Request) {
	indexHandler(w, r)
}

// handles "/api/plain/(users|mentions|tweets)"
func apiEndpointHandler(w http.ResponseWriter, r *http.Request) {

	err := r.ParseForm()
	if err != nil {
		log500(w, r, err)
		return
	}

	if r.FormValue("q") != "" || r.FormValue("url") != "" {
		err := apiEndpointQuery(w, r)
		if err != nil {
			log500(w, r, err)
			return
		}
		log200(r)
		return
	}

	// if there's no query, return everything in
	// registry for a given endpoint
	var out []string
	switch r.URL.Path {
	case "/api/plain/users":
		out, err = twtxtCache.QueryUser("")

	case "/api/plain/mentions":
		out, err = twtxtCache.QueryInStatus("@<")

	default:
		out, err = twtxtCache.QueryAllStatuses()
	}

	data := parseQueryOut(out)
	if err != nil {
		data = []byte("")
	}

	etag := fmt.Sprintf("%x", sha256.Sum256(data))
	w.Header().Set("ETag", etag)
	w.Header().Set("Content-Type", txtutf8)

	_, err = w.Write(data)
	if err != nil {
		log500(w, r, err)
		return
	}

	log200(r)
}

// handles POST for "/api/plain/users"
func apiEndpointPOSTHandler(w http.ResponseWriter, r *http.Request) {
	apiPostUser(w, r)
}

// handles "/api/plain/tags"
func apiTagsBaseHandler(w http.ResponseWriter, r *http.Request) {

	out, err := twtxtCache.QueryInStatus("#")
	if err != nil {
		log500(w, r, err)
		return
	}

	data := parseQueryOut(out)

	etag := fmt.Sprintf("%x", sha256.Sum256(data))
	w.Header().Set("ETag", etag)
	w.Header().Set("Content-Type", txtutf8)

	_, err = w.Write(data)
	if err != nil {
		log500(w, r, err)
		return
	}

	log200(r)
}

// handles "/api/plain/tags/[a-zA-Z0-9]+"
func apiTagsHandler(w http.ResponseWriter, r *http.Request) {

	vars := mux.Vars(r)
	tags := vars["tags"]

	tags = strings.ToLower(tags)
	out, err := twtxtCache.QueryInStatus("#" + tags)
	if err != nil {
		log500(w, r, err)
		return
	}
	tags = strings.Title(tags)
	out2, err := twtxtCache.QueryInStatus("#" + tags)
	if err != nil {
		log500(w, r, err)
		return
	}
	tags = strings.ToUpper(tags)
	out3, err := twtxtCache.QueryInStatus("#" + tags)
	if err != nil {
		log500(w, r, err)
		return
	}

	out = append(out, out2...)
	out = append(out, out3...)
	out = uniq(out)

	data := parseQueryOut(out)

	etag := fmt.Sprintf("%x", sha256.Sum256(data))
	w.Header().Set("ETag", etag)
	w.Header().Set("Content-Type", txtutf8)

	_, err = w.Write(data)
	if err != nil {
		log500(w, r, err)
		return
	}

	log200(r)
}

// Serving the stylesheet virtually because
// files aren't served directly in getwtxt.
func cssHandler(w http.ResponseWriter, r *http.Request) {

	// Sending the sha256 sum of the modtime in hexadecimal for the ETag header
	etag := fmt.Sprintf("%x", sha256.Sum256([]byte(staticCache.cssMod.String())))

	w.Header().Set("ETag", "\""+etag+"\"")
	w.Header().Set("Content-Type", cssutf8)

	pingAssets()

	n, err := w.Write(staticCache.css)
	if err != nil || n == 0 {
		log500(w, r, err)
		return
	}

	log200(r)
}