diff options
Diffstat (limited to 'main.go')
| -rw-r--r-- | main.go | 128 |
1 files changed, 3 insertions, 125 deletions
@@ -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, "<option value=\"%s\">%s</option>", 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, "<hr>") - 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, "<hr>") - } else { - fmt.Fprintf(w, "<center><h1>No results found for %s %s(%s)</h1></center>", 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", |