From ee64cdc2712a8e7f77c6af02385afd43a2a1e61b Mon Sep 17 00:00:00 2001 From: Jacob McDonnell Date: Sun, 26 Apr 2026 16:52:03 -0400 Subject: refactor: Split Go code into multiple files --- main.go | 128 ++-------------------------------------------------------------- 1 file changed, 3 insertions(+), 125 deletions(-) (limited to 'main.go') diff --git a/main.go b/main.go index b8a49198..69897f4c 100644 --- a/main.go +++ b/main.go @@ -1,136 +1,14 @@ package main import ( + "database/sql" "fmt" "html/template" "log" "net/http" "os" - "slices" - "regexp" - "database/sql" - _ "modernc.org/sqlite" ) -type Entry struct { - os string - name string - section int - path string -} - -type Database struct { - DB *sql.DB -} - -func (d *Database) GetOS(w http.ResponseWriter, r *http.Request) { - rows, err := d.DB.Query("SELECT DISTINCT os FROM manpages;") - if err != nil { - log.Println(err) - ErrorPage(w, 500) - return - } - defer rows.Close() - - for rows.Next() { - var osName string - - if err := rows.Scan(&osName); err != nil { - log.Println(err) - ErrorPage(w, 500) - return - } - - fmt.Fprintf(w, "", osName, osName) - } -} - -func (d *Database) Search(w http.ResponseWriter, r *http.Request) { - query := r.URL.Query() - - log.Println("Request for:", r.URL.String()) - - name := query["name"][0] - section := query["section"][0] - osName := query["os"][0] - - log.Println(name) - - rows, err := d.DB.Query("SELECT * FROM manpages WHERE name == ? AND section == ? AND os == ?;", name, section, osName) - if err != nil { - log.Println(err) - ErrorPage(w, 500) - return - } - defer rows.Close() - - var entries []Entry - for rows.Next() { - var entry Entry - - if err := rows.Scan(&entry.os, &entry.name, &entry.section, &entry.path); err != nil { - log.Println(err) - ErrorPage(w, 500) - return - } - - entries = append(entries, entry) - } - - if len(entries) > 0 { - fmt.Fprintf(w, "
") - for _, e := range entries { - data, err := os.ReadFile(e.path) - if err != nil { - log.Println(err) - ErrorPage(w, 500) - return - } - fmt.Fprintf(w, "%s", data) - } - fmt.Fprintf(w, "
") - } else { - fmt.Fprintf(w, "

No results found for %s %s(%s)

", osName, name, section) - } -} - -type AllowedFiles struct { - allowedFiles []string -} - -func (a *AllowedFiles) GetFiles(w http.ResponseWriter, r *http.Request) { - log.Println("Request for:", r.URL.String()) - url := fmt.Sprintf("static%s", r.URL.String()) - htmlRegex, err := regexp.Compile("^static/.*\\.html$") - if err != nil { - ErrorPage(w, 500) - return - } - - if url == "static/" { - url = "static/index.html" - } - - if !slices.Contains(a.allowedFiles, url) { - ErrorPage(w, 404) - return - } - - if htmlRegex.MatchString(url) { - t := template.Must(template.ParseFiles("templates/template.html")) - - html, err := os.ReadFile(url) - if err != nil { - ErrorPage(w, 500) - return - } - - t.Execute(w, template.HTML(html)) - } else { - http.ServeFile(w, r, url) - } -} - func ErrorPage(w http.ResponseWriter, code int) { log.Println(code, http.StatusText(code)) t := template.Must(template.ParseFiles("templates/error.html")) @@ -174,8 +52,8 @@ func main() { log.Fatal(err) } - dbWrapper := &Database{ DB: db } - allowed := &AllowedFiles { + dbWrapper := &Database{DB: db} + allowed := &AllowedFiles{ allowedFiles: []string{ "static/htmx.min.js", "static/index.html", -- cgit v1.2.3