implement /remove, /new, and /replace

This commit is contained in:
2025-08-02 17:04:04 -07:00
parent 773ee1ebb8
commit 9404a91f51
3 changed files with 186 additions and 0 deletions

17
go.mod Normal file
View File

@@ -0,0 +1,17 @@
module miningtcup.me/noteserver
go 1.24.5
require (
github.com/dustin/go-humanize v1.0.1 // indirect
github.com/google/uuid v1.6.0 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect
github.com/ncruces/go-strftime v0.1.9 // indirect
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b // indirect
golang.org/x/sys v0.34.0 // indirect
modernc.org/libc v1.66.3 // indirect
modernc.org/mathutil v1.7.1 // indirect
modernc.org/memory v1.11.0 // indirect
modernc.org/sqlite v1.38.2 // indirect
)

23
go.sum Normal file
View File

@@ -0,0 +1,23 @@
github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
github.com/ncruces/go-strftime v0.1.9 h1:bY0MQC28UADQmHmaF5dgpLmImcShSi2kHU9XLdhx/f4=
github.com/ncruces/go-strftime v0.1.9/go.mod h1:Fwc5htZGVVkseilnfgOVb9mKy6w1naJmn9CehxcKcls=
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94icq4NjY3clb7Lk8O1qJ8BdBEF8z0ibU0rE=
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b h1:M2rDM6z3Fhozi9O7NWsxAkg/yqS/lQJ6PmkyIV3YP+o=
golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b/go.mod h1:3//PLf8L/X+8b4vuAfHzxeRUl04Adcb341+IGKfnqS8=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.34.0 h1:H5Y5sJ2L2JRdyv7ROF1he/lPdvFsd0mJHFw2ThKHxLA=
golang.org/x/sys v0.34.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
modernc.org/libc v1.66.3 h1:cfCbjTUcdsKyyZZfEUKfoHcP3S0Wkvz3jgSzByEWVCQ=
modernc.org/libc v1.66.3/go.mod h1:XD9zO8kt59cANKvHPXpx7yS2ELPheAey0vjIuZOhOU8=
modernc.org/mathutil v1.7.1 h1:GCZVGXdaN8gTqB1Mf/usp1Y/hSqgI2vAGGP4jZMCxOU=
modernc.org/mathutil v1.7.1/go.mod h1:4p5IwJITfppl0G4sUEDtCr4DthTaT47/N3aT6MhfgJg=
modernc.org/memory v1.11.0 h1:o4QC8aMQzmcwCK3t3Ux/ZHmwFPzE6hf2Y5LbkRs+hbI=
modernc.org/memory v1.11.0/go.mod h1:/JP4VbVC+K5sU2wZi9bHoq2MAkCnrt2r98UGeSK7Mjw=
modernc.org/sqlite v1.38.2 h1:Aclu7+tgjgcQVShZqim41Bbw9Cho0y/7WzYptXqkEek=
modernc.org/sqlite v1.38.2/go.mod h1:cPTJYSlgg3Sfg046yBShXENNtPrWrDX8bsbAQBzgQ5E=

146
main.go Normal file
View File

@@ -0,0 +1,146 @@
package main
import (
"database/sql"
"encoding/json"
"fmt"
"log"
"net/http"
"strconv"
_ "modernc.org/sqlite"
"github.com/google/uuid"
)
var (
db *sql.DB
)
func main() {
fmt.Println("hello, world!")
var err error
db, err = sql.Open("sqlite", "./db/notes.db")
if err != nil {
log.Fatal(err)
}
defer db.Close()
http.HandleFunc("/get", func(w http.ResponseWriter, r *http.Request) {
r.ParseForm()
if !r.Form.Has("user") {
w.WriteHeader(400)
fmt.Fprint(w, "This endpoint requires parameters 'user'")
return
}
limit, err := strconv.Atoi(r.Form.Get("max"))
if err != nil {
limit = 100
}
rows := getNotes(r.Form.Get("user"), limit)
jsonString, err := json.Marshal(rows)
if err != nil {
log.Fatal(err)
}
w.Header().Add("Content-Type", "text/json")
fmt.Fprint(w, string(jsonString))
fmt.Println(string(jsonString))
})
http.HandleFunc("/new", func(w http.ResponseWriter, r *http.Request) {
r.ParseForm()
if !r.Form.Has("user") || !r.Form.Has("data") {
w.WriteHeader(400)
fmt.Fprint(w, "This endpoint requires parameters 'user' and 'data'")
return
}
uuid, err := uuid.NewUUID()
if err != nil {
log.Fatal(err)
}
newNote(r.Form.Get("user"), r.Form.Get("data"), uuid.String())
fmt.Fprint(w, uuid)
})
http.HandleFunc("/replace", func(w http.ResponseWriter, r *http.Request) {
r.ParseForm()
if !r.Form.Has("user") || !r.Form.Has("data") || !r.Form.Has("id") {
w.WriteHeader(400)
fmt.Fprint(w, "This endpoint requires parameters 'user' and 'data'")
return
}
removeNote(r.Form.Get("user"), r.Form.Get("id"))
newNote(r.Form.Get("user"), r.Form.Get("data"), r.Form.Get("id"))
fmt.Fprint(w, http.StatusText(200))
})
http.HandleFunc("/remove", func(w http.ResponseWriter, r *http.Request) {
r.ParseForm()
if !r.Form.Has("user") || !r.Form.Has("id") {
w.WriteHeader(400)
fmt.Fprint(w, "This endpoint requires parameters 'user' and 'id'")
return
}
removeNote(r.Form.Get("user"), r.Form.Get("id"))
fmt.Fprint(w, http.StatusText(200))
})
log.Fatal(http.ListenAndServe(":3000", nil))
}
func getNotes(user string, limit int) [][2]string {
var rowCount int
err := db.QueryRow("SELECT COUNT(*) FROM notes WHERE user = ?", user).Scan(&rowCount)
if err != nil {
log.Fatal(err)
}
if rowCount > limit {
rowCount = limit
}
rows, err := db.Query("SELECT * FROM notes WHERE user = ? LIMIT ?", user, limit)
if err != nil {
log.Fatal(err)
}
defer rows.Close()
notes := make([][2]string, rowCount)
var dummy string
for i := 0; rows.Next(); i++ {
var currentRow [2]string
if err := rows.Scan(&currentRow[0], &dummy, &currentRow[1]); err != nil {
log.Fatal(err)
}
notes[i] = currentRow
}
return notes
}
func newNote(user string, data string, uuid string) {
stmt, err := db.Prepare("INSERT INTO notes VALUES(?, ?, ?)")
if err != nil {
log.Fatal(err)
}
if _, err := stmt.Exec(uuid, user, data); err != nil {
log.Fatal(err)
}
}
func removeNote(user string, uuid string) {
stmt, err := db.Prepare("DELETE FROM notes WHERE user = ? AND id = ?")
if err != nil {
log.Fatal(err)
}
if _, err := stmt.Exec(user, uuid); err != nil {
log.Fatal(err)
}
}