Merge pull request #1 from dxe/jake/add-tally-endpoint

add tally endpoint

service/db.go

@@ -1,9 +1,10 @@
 package main
 
 import (
+	"log"
+
 	"github.com/dxe/helptheducks.com/service/config"
 	"github.com/jmoiron/sqlx"
-	"log"
 )
 
 // TODO: make a separate script to init db b/c this is kinda sketch for prod.

service/handlers.go

@@ -4,10 +4,12 @@ import (
 	"database/sql"
 	"encoding/json"
 	"fmt"
-	"github.com/dxe/helptheducks.com/service/config"
-	"github.com/dxe/helptheducks.com/service/model"
 	"net/http"
 	"net/mail"
+
+	"github.com/dxe/helptheducks.com/service/config"
+	"github.com/dxe/helptheducks.com/service/model"
+	"github.com/go-chi/chi/v5"
 )
 
 type CreateMessageInput struct {
@@ -92,3 +94,30 @@ func createMessageHandler(w http.ResponseWriter, r *http.Request) {
 
 	w.Write([]byte("ok"))
 }
+
+type GetTallyResp struct {
+	Total int `json:"total"`
+}
+
+func getTallyHandler(w http.ResponseWriter, r *http.Request) {
+	campaign := chi.URLParam(r, "campaign")
+
+	tally, err := model.GetTally(db, campaign)
+	if err != nil {
+		http.Error(w, fmt.Sprintf("error getting tally: %v", err), http.StatusInternalServerError)
+		return
+	}
+
+	resp := GetTallyResp{
+		Total: tally,
+	}
+	json, err := json.Marshal(resp)
+	if err != nil {
+		http.Error(w, "failed to marshal json", http.StatusInternalServerError)
+		return
+	}
+
+	w.Header().Set("Content-Type", "application/json")
+	w.WriteHeader(http.StatusOK)
+	w.Write(json)
+}

service/main.go

@@ -2,6 +2,11 @@ package main
 
 import (
 	"fmt"
+	"net/http"
+	"strings"
+	"time"
+	"unicode"
+
 	"github.com/aws/aws-sdk-go/service/ses"
 	"github.com/dxe/helptheducks.com/service/config"
 	"github.com/dxe/helptheducks.com/service/mailer"
@@ -11,10 +16,6 @@ import (
 	"github.com/go-chi/cors"
 	_ "github.com/go-sql-driver/mysql"
 	"github.com/jmoiron/sqlx"
-	"net/http"
-	"strings"
-	"time"
-	"unicode"
 
 	"golang.org/x/text/runes"
 	"golang.org/x/text/transform"
@@ -53,6 +54,8 @@ func main() {
 		r.Post("/create", createMessageHandler)
 	})
 
+	r.Get("/tally", getTallyHandler)
+
 	go worker()
 
 	fmt.Printf("Listening on port %v\n", config.Port)

service/model/tally.go

@@ -0,0 +1,18 @@
+package model
+
+import "github.com/jmoiron/sqlx"
+
+func GetTally(db *sqlx.DB, campaign string) (int, error) {
+	var tally int
+	err := db.Get(
+		&tally,
+		`SELECT count(*)
+		FROM messages
+		WHERE campaign = ?`,
+		campaign,
+	)
+	if (err != nil) {
+		return 0, err
+	}
+	return tally, nil 
+}

src/components/petition.tsx

@@ -121,7 +121,7 @@ export const Petition = () => {
             ...(data.zip && { zip: data.zip }),
             ...(data.city && { city: data.city }),
             message: data.message,
-            campaign: "duck",
+            campaign: import.meta.env.PROD ? "duck" : "test",
             token,
           },
           headers: {

src/routeTree.gen.ts

@@ -1,17 +1,9 @@
-// This file is auto-generated by TanStack Router
-
 import { FileRoute, lazyRouteComponent } from '@tanstack/react-router'
 
-// Import Routes
-
 import { Route as rootRoute } from './routes/__root'
 
-// Create Virtual Routes
-
 const IndexComponentImport = new FileRoute('/').createRoute()
 
-// Create/Update Routes
-
 const IndexComponentRoute = IndexComponentImport.update({
   path: '/',
   getParentRoute: () => rootRoute,
@@ -21,9 +13,6 @@ const IndexComponentRoute = IndexComponentImport.update({
     'component',
   ),
 })
-
-// Populate the FileRoutesByPath interface
-
 declare module '@tanstack/react-router' {
   interface FileRoutesByPath {
     '/': {
@@ -32,7 +21,4 @@ declare module '@tanstack/react-router' {
     }
   }
 }
-
-// Create and export the route tree
-
 export const routeTree = rootRoute.addChildren([IndexComponentRoute])