89 lines
1.9 KiB
Go
89 lines
1.9 KiB
Go
package web
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"html/template"
|
|
"log"
|
|
"math/big"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/vilmibm/trunkless/db"
|
|
)
|
|
|
|
// TODO detect max id on startup
|
|
|
|
type source struct {
|
|
ID string
|
|
Name string
|
|
}
|
|
|
|
type phrase struct {
|
|
ID int64
|
|
Text string
|
|
Source source
|
|
}
|
|
|
|
func Serve() error {
|
|
r := gin.Default()
|
|
r.SetFuncMap(template.FuncMap{
|
|
"upper": strings.ToUpper,
|
|
})
|
|
r.LoadHTMLFiles("templates/index.tmpl")
|
|
r.StaticFile("/cutive.ttf", "./web/assets/cutive.ttf")
|
|
r.StaticFile("/favicon.ico", "./web/assets/favicon.ico")
|
|
r.StaticFile("/bg_light.gif", "./web/assets/bg_light.gif")
|
|
r.StaticFile("/bg_dark.gif", "./web/assets/bg_dark.gif")
|
|
r.StaticFile("/main.js", "./web/assets/main.js")
|
|
r.StaticFile("/html2canvas.min.js", "./web/assets/html2canvas.min.js")
|
|
|
|
// TODO use new db functions for id ranges
|
|
randMax := big.NewInt(db.MaxID)
|
|
|
|
r.HEAD("/", func(c *gin.Context) {
|
|
c.String(http.StatusOK, "")
|
|
})
|
|
|
|
r.GET("/", func(c *gin.Context) {
|
|
c.HTML(http.StatusOK, "index.tmpl", struct {
|
|
// TODO handle multiple corpora
|
|
MaxID int
|
|
}{db.MaxID})
|
|
})
|
|
|
|
// TODO retool this for pg
|
|
|
|
r.GET("/line", func(c *gin.Context) {
|
|
conn, err := db.Connect()
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
c.String(http.StatusInternalServerError, "oh no.")
|
|
return
|
|
}
|
|
defer conn.Close(context.Background())
|
|
|
|
id, err := rand.Int(rand.Reader, randMax)
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
c.String(http.StatusInternalServerError, "oh no.")
|
|
return
|
|
}
|
|
|
|
row := conn.QueryRow(context.Background(), "select p.phrase, s.id, s.name from phrases p join sources s on p.sourceid = s.id where p.id = $1", id.Int64())
|
|
var p phrase
|
|
var s source
|
|
err = row.Scan(&p.Text, &s.ID, &s.Name)
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
c.String(http.StatusInternalServerError, "oh no.")
|
|
}
|
|
p.Source = s
|
|
p.ID = id.Int64()
|
|
c.JSON(http.StatusOK, p)
|
|
})
|
|
|
|
return r.Run() // 8080
|
|
}
|