Adding initial code for the main.go as well as initial code for uploading members

server/db/db.go

@@ -31,6 +31,7 @@ CREATE TABLE IF NOT EXISTS voters (
 CREATE TABLE IF NOT EXISTS members (
     email TEXT NOT NULL,
     member_name TEXT,
+    school_year UNSIGNED INT NOT NULL,
     PRIMARY KEY (email)
 );
 `

server/main.go

@@ -0,0 +1,126 @@
+package main
+
+import (
+	"encoding/json"
+	"io/ioutil"
+	"log"
+	"net/http"
+
+	"github.com/gorilla/mux"
+	"go-sjles-pta-vote/server/services"
+)
+
+func voteHandler(w http.ResponseWriter, r *http.Request) {
+	var vote services.Vote
+	if err := json.NewDecoder(r.Body).Decode(&vote); err != nil {
+		http.Error(w, "Invalid request payload", http.StatusBadRequest)
+		return
+	}
+
+	err := services.SetVote(&vote)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	w.WriteHeader(http.StatusOK)
+}
+
+func voteIDHandler(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	id := vars["id"]
+
+	vote := services.Vote{
+		PollId: id,
+		Email:  "example@example.com", // Replace with actual email retrieval logic
+		Vote:   true,                    // Replace with actual vote retrieval logic
+	}
+
+	err := services.SetVote(&vote)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	w.WriteHeader(http.StatusOK)
+}
+
+func statsHandler(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	id := vars["id"]
+
+	poll, err := services.GetPollByQuestion(id)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	json.NewEncoder(w).Encode(poll)
+}
+
+func statsIDHandler(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	id := vars["id"]
+
+	poll, err := services.GetPollByQuestion(id)
+	if err != nil {
+		http.Error(w, err.Error(), http.StatusInternalServerError)
+		return
+	}
+
+	json.NewEncoder(w).Encode(poll)
+}
+
+func adminHandler(w http.ResponseWriter, r *http.Request) {
+	// Add admin functionality here
+	w.WriteHeader(http.StatusOK)
+}
+
+func adminIDHandler(w http.ResponseWriter, r *http.Request) {
+	vars := mux.Vars(r)
+	id := vars["id"]
+
+	// Add admin functionality here
+	w.WriteHeader(http.StatusOK)
+}
+
+func membersHandler(w http.ResponseWriter, r *http.Request) {
+	if r.Method == "POST" {
+		file, handler, err := r.FormFile("members.csv")
+		if err != nil {
+			http.Error(w, "Failed to upload file", http.StatusBadRequest)
+			return
+		}
+		defer file.Close()
+
+		fileBytes, err := ioutil.ReadAll(file)
+		if err != nil {
+			http.Error(w, "Failed to read file", http.StatusInternalServerError)
+			return
+		}
+
+		members, err := services.ParseMembersFromBytes(2023, fileBytes) // Assuming year 2023 for demonstration purposes
+		if err != nil {
+			http.Error(w, err.Error(), http.StatusInternalServerError)
+			return
+		}
+
+		json.NewEncoder(w).Encode(members)
+	} else {
+		w.WriteHeader(http.StatusMethodNotAllowed)
+	}
+}
+
+func main() {
+	r := mux.NewRouter()
+
+	r.HandleFunc("/vote", voteHandler).Methods("POST")
+	r.HandleFunc("/vote/{id}", voteIDHandler).Methods("POST")
+	r.HandleFunc("/stats", statsHandler).Methods("POST")
+	r.HandleFunc("/stats/{id}", statsIDHandler).Methods("POST")
+	r.HandleFunc("/admin", adminHandler).Methods("GET")
+	r.HandleFunc("/admin/{id}", adminIDHandler).Methods("GET")
+	r.HandleFunc("/admin/members", membersHandler).Methods("POST")
+
+	log.Fatal(http.ListenAndServe(":8080", r))
+}

server/models/members.go

@@ -3,4 +3,5 @@ package models
 type Members struct {
 	Name string `json:"name"`
 	Email string `json:"email"`
+	SchoolYear int `json:"school_year"`
 }

server/services/putmembers.go

@@ -0,0 +1,45 @@
+package services
+
+import (
+	"encoding/csv"
+	"fmt"
+	"strings"
+
+	"github.com/pkg/errors"
+)
+
+type Member struct {
+	Name  string
+	Email string
+}
+
+func ParseMembersFromBytes(year int, fileBytes []byte) ([]Member, error) {
+	reader := csv.NewReader(strings.NewReader(string(fileBytes)))
+	records, err := reader.ReadAll()
+	if err != nil {
+		return nil, errors.Wrap(err, "failed to read CSV from bytes")
+	}
+
+	var members []Member
+	for i, record := range records {
+		if i == 0 {
+			continue // Skip the first line (column headers)
+		}
+		if len(record) < 4 {
+			continue
+		}
+		firstName := record[1]
+		lastName := record[2]
+		email := record[3]
+		members = append(members, Member{
+			Name:  fmt.Sprintf("%s %s", firstName, lastName),
+			Email: email,
+		})
+	}
+
+	if len(members) == 0 {
+		members = []Member{}
+	}
+
+	return members, nil
+}

server/services/putmembers_test.go

@@ -0,0 +1,45 @@
+package services
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+)
+
+func TestParseMembersFromBytes(t *testing.T) {
+	testCases := []struct {
+		name     string
+		input    string
+		expected []Member
+	}{
+		{
+			name: "Valid CSV with multiple members",
+			input: `date,First,Last,Email
+2023-01-01,John,Doe,john.doe@example.com
+2023-01-02,Jane,Smith,jane.smith@example.com`,
+			expected: []Member{
+				{Name: "John Doe", Email: "john.doe@example.com"},
+				{Name: "Jane Smith", Email: "jane.smith@example.com"},
+			},
+		},
+		{
+			name: "CSV with missing fields",
+			input: `date,First,Last
+2023-01-01,John,Doe`,
+			expected: []Member{},
+		},
+		{
+			name: "Empty CSV",
+			input: ``,
+			expected: []Member{},
+		},
+	}
+
+	for _, tc := range testCases {
+		t.Run(tc.name, func(t *testing.T) {
+			members, err := ParseMembersFromBytes(2023, []byte(tc.input))
+			assert.NoError(t, err)
+			assert.Equal(t, tc.expected, members)
+		})
+	}
+}

server/services/services_test.go

@@ -109,7 +109,7 @@ func PreLoadDB() error {
 
 	// Insert members
 	for i := range new_members {
-		_, err := db_conn.Exec(`INSERT INTO members (email, member_name) VALUES (?, ?)`, new_members[i].email, new_members[i].member_name)
+		_, err := db_conn.Exec(`INSERT INTO members (email, member_name, school_year) VALUES (?, ?, ?)`, new_members[i].email, new_members[i].member_name, 2023)
 		if err != nil {
 			return err
 		}