early database work

Signed-off-by: ari melody <ari@arimelody.me>
This commit is contained in:
ari melody 2024-03-18 20:19:04 +00:00
parent 6efd47c6c6
commit 5eecef1d78
8 changed files with 512 additions and 96 deletions

106
main.go
View file

@ -10,6 +10,7 @@ import (
"log"
"strings"
"time"
"strconv"
"github.com/gomarkdown/markdown"
"github.com/gomarkdown/markdown/html"
@ -35,49 +36,69 @@ var templates = template.Must(template.ParseFiles(
"views/music-gateway.html",
))
func web_handler(writer http.ResponseWriter, req *http.Request) {
uri := req.URL.Path
fmt.Printf("[%s] %s %s (%s)\n",
time.Now().Format(time.UnixDate),
func log_request(req *http.Request, code int, start_time time.Time) {
now := time.Now()
difference := (now.Nanosecond() - start_time.Nanosecond()) / 1_000_000
elapsed := "<1"
if difference >= 1 {
elapsed = strconv.Itoa(difference)
}
fmt.Printf("[%s] %s %s - %d (%sms) (%s)\n",
now.Format(time.UnixDate),
req.Method,
req.URL.Path,
code,
elapsed,
req.Header["User-Agent"][0],
)
}
func web_handler(writer http.ResponseWriter, req *http.Request) {
uri := req.URL.Path
start_time := time.Now()
if req.URL.Path == "/" {
index_handler(writer, req)
code := index_handler(writer, req)
log_request(req, code, start_time)
return
}
if uri == "/music" {
music_handler(writer, req)
code := music_handler(writer, req)
log_request(req, code, start_time)
return
}
if strings.HasPrefix(uri, "/music/") {
music_gateway_handler(writer, req)
code := music_gateway_handler(writer, req)
log_request(req, code, start_time)
return
}
static_handler(writer, req)
code := static_handler(writer, req)
log_request(req, code, start_time)
}
func index_handler(writer http.ResponseWriter, req *http.Request) {
func index_handler(writer http.ResponseWriter, req *http.Request) int {
err := templates.ExecuteTemplate(writer, "index.html", nil)
if err != nil {
http.Error(writer, err.Error(), http.StatusInternalServerError)
return 500
}
return 200
}
func music_handler(writer http.ResponseWriter, req *http.Request) {
err := templates.ExecuteTemplate(writer, "music.html", music.QueryAll())
func music_handler(writer http.ResponseWriter, req *http.Request) int {
err := templates.ExecuteTemplate(writer, "music.html", music.QueryAllAlbums())
if err != nil {
http.Error(writer, err.Error(), http.StatusInternalServerError)
return 500
}
return 200
}
func music_gateway_handler(writer http.ResponseWriter, req *http.Request) {
func music_gateway_handler(writer http.ResponseWriter, req *http.Request) int {
if len(req.URL.Path) <= len("/music/") {
http.Error(writer, "400 bad request", http.StatusBadRequest)
return
return 400
}
id := req.URL.Path[len("/music/"):]
@ -86,28 +107,61 @@ func music_gateway_handler(writer http.ResponseWriter, req *http.Request) {
album, ok := music.GetAlbum(id)
if !ok {
http.Error(writer, "404 not found", http.StatusNotFound)
return
return 404
}
err := templates.ExecuteTemplate(writer, "music-gateway.html", album)
if err != nil {
http.Error(writer, err.Error(), http.StatusInternalServerError)
return 500
}
return 200
}
func static_handler(writer http.ResponseWriter, req *http.Request) {
func static_handler(writer http.ResponseWriter, req *http.Request) int {
filename := "public/" + req.URL.Path[1:]
body, err := os.ReadFile(filename)
// check the file's metadata
info, err := os.Stat(filename)
if err != nil {
http.Error(writer, "404 not found", http.StatusNotFound)
return
return 404
}
if len(req.Header["If-Modified-Since"]) > 0 && req.Header["If-Modified-Since"][0] != "" {
if_modified_since_time, err := time.Parse(http.TimeFormat, req.Header["If-Modified-Since"][0])
if err != nil {
http.Error(writer, "400 bad request", http.StatusBadRequest)
return 400
}
if req.Header["If-Modified-Since"][0] == info.ModTime().Format(http.TimeFormat) || if_modified_since_time.After(info.ModTime()) {
writer.WriteHeader(304) // not modified
return 304
}
}
// set other nice headers
writer.Header().Set("Cache-Control", "max-age=86400")
writer.Header().Set("Last-Modified", info.ModTime().Format(http.TimeFormat))
// Last-Modified: <day-name>, <day> <month> <year> <hour>:<minute>:<second> GMT
// Last-Modified: Wed, 21 Oct 2015 07:28:00 GMT
// read the file
body, err := os.ReadFile(filename)
if err != nil {
http.Error(writer, err.Error(), http.StatusInternalServerError)
return 500
}
// setting MIME types
filetype := filename[strings.LastIndex(filename, ".") + 1:]
if mime_type, ok := mime_types[filetype]; ok {
writer.Header().Set("Content-Type", mime_type)
} else {
writer.Header().Set("Content-Type", "text/plain; charset=utf-8")
}
writer.Write([]byte(body))
return 200
}
func parse_markdown(md []byte) []byte {
@ -122,7 +176,23 @@ func parse_markdown(md []byte) []byte {
return markdown.Render(doc, renderer)
}
func push_to_db_this_is_a_testing_thing_and_will_be_superfluous_later() {
db := InitDatabase()
for _, album := range music.QueryAllAlbums() {
PushAlbum(db, album)
}
for _, artist := range music.QueryAllArtists() {
PushArtist(db, artist)
}
defer db.Close()
}
func main() {
// push_to_db_this_is_a_testing_thing_and_will_be_superfluous_later()
http.HandleFunc("/", web_handler)
log.Fatal(http.ListenAndServe(":8080", nil))
}