diff options
Diffstat (limited to 'cmd/parser')
| -rw-r--r-- | cmd/parser/main.go | 109 |
1 files changed, 98 insertions, 11 deletions
diff --git a/cmd/parser/main.go b/cmd/parser/main.go index 3933935..c9c80c8 100644 --- a/cmd/parser/main.go +++ b/cmd/parser/main.go @@ -2,48 +2,86 @@ package main import ( "bytes" + "context" + "database/sql" "fmt" "log" "net/http" "os" "strconv" + "time" "github.com/joho/godotenv" "github.com/yuin/goldmark" "github.com/yuin/goldmark/extension" "github.com/yuin/goldmark/parser" "github.com/yuin/goldmark/renderer/html" + "golang.org/x/crypto/bcrypt" + + _ "github.com/jackc/pgx/v5/stdlib" ) +var db *sql.DB var mdParser goldmark.Markdown +func writeErr(w http.ResponseWriter, errcode int, msg string) { + w.WriteHeader(errcode) + w.Write([]byte(msg)) + return +} + func mdServe(w http.ResponseWriter, r *http.Request) { if r.Method != "POST" { - w.WriteHeader(405) - w.Write([]byte("Method Not Allowed")) + writeErr(w, 405, "Method Not Allowed") return } - err := r.ParseForm() + err := r.ParseMultipartForm(4 << 20) if err != nil { - w.WriteHeader(500) - w.Write([]byte("Failed to retrieve form data")) + writeErr(w, 500, "Failed to retrieve form data") return } var buf bytes.Buffer - - log.Print(r.Form) + md := r.PostForm.Get("raw") - log.Print(md) + err = mdParser.Convert([]byte(md), &buf) if err != nil { - w.WriteHeader(500) - w.Write([]byte("Failed to compile markdown into html")) + writeErr(w, 500, fmt.Sprintf("Failed to compile markdown into html: %v", err)) return } - w.Write([]byte(md)) + ctx, cancel := context.WithTimeout(r.Context(), 5 * time.Second) + defer cancel() + + tx, err := db.BeginTx(ctx, nil) + if err != nil { + writeErr(w, 500, fmt.Sprintf("Failed to initialize transaction: %v", err)) + return + } + defer tx.Rollback() + + stmt, err := tx.PrepareContext(ctx, "INSERT INTO posts (name, content) VALUES ($1, $2);") + if err != nil { + writeErr(w, 500, fmt.Sprintf("Failed to prepare DB statement: %v", err)) + return + } + defer stmt.Close() + + _, err = stmt.ExecContext(ctx, os.Getenv("webmaster"), buf.Bytes()) + if err != nil { + writeErr(w, 500, fmt.Sprintf("Failed to execute statement: %v", err)) + return + } + + err = tx.Commit() + if err != nil { + writeErr(w, 500, fmt.Sprintf("Failed to commit DB transaction: %v", err)) + } + + w.WriteHeader(200) + w.Write([]byte("Successfully parsed and stored markdown")) } func main() { @@ -66,6 +104,55 @@ func main() { ), ) + host := "localhost" + port := 5432 + dbName := os.Getenv("db_name") + user := os.Getenv("db_user") + pass := os.Getenv("db_pass") + connStr := fmt.Sprintf("host=%s port=%d user=%s password=%s dbname=%s sslmode=disable", host, port, user, pass, dbName) + + db, err = sql.Open("pgx", connStr) + if err != nil { + fmt.Fprintf(os.Stderr, "Unable to connect to DB: %v\n", err) + os.Exit(1) + } + defer db.Close() + + _, table_check := db.Query("SELECT * FROM posts;") + if table_check != nil { + _, err = db.Exec("CREATE TABLE posts (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, time DATE DEFAULT CURRENT_DATE, content TEXT NOT NULL);") + if err != nil { + fmt.Fprintf(os.Stderr, "Unable to create posts table: %v\n", err) + os.Exit(1) + } + } + + webmaster := os.Getenv("webmaster") + passOne := os.Getenv("blog_pass1") + passTwo := os.Getenv("blog_pass2") + + _, table_check = db.Query("SELECT * FROM users;") + if table_check != nil { + _, err = db.Exec("CREATE TABLE logins (id SERIAL PRIMARY KEY, name VARCHAR(50) NOT NULL, time DATE DEFAULT CURRENT_DATE, pass_one TEXT NOT NULL, pass_two TEXT NOT NULL);") + if err != nil { + fmt.Fprintf(os.Stderr, "Unable to create logins table: %v\n", err) + os.Exit(1) + } + + hashOne, errHashOne := bcrypt.GenerateFromPassword([]byte(passOne), 12) + hashTwo, errHashTwo := bcrypt.GenerateFromPassword([]byte(passTwo), 12) + if errHashOne != nil || errHashTwo != nil { + fmt.Fprintf(os.Stderr, "Failed to hash password") + os.Exit(1) + } + + _, err = db.Exec(fmt.Sprintf("INSERT INTO logins (name, pass_one, pass_two) VALUES ('%s', '%s', '%s');", webmaster, hashOne, hashTwo)) + if err != nil { + fmt.Fprintf(os.Stderr, "Unable to add webmaster to logins table: %v\n", err) + os.Exit(1) + } + } + router := http.NewServeMux() router.HandleFunc("/", mdServe) |