1 files changed, 194 insertions, 0 deletions
diff --git a/endpoint_index.go b/endpoint_index.go
new file mode 100644
index 0000000..512fe1d
--- /dev/null
+++ b/endpoint_index.go
@@ -0,0 +1,194 @@
+/*
+ * Index page
+ *
+ * Copyright (C) 2024  Runxi Yu <https://runxiyu.org>
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ *
+ * This program is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <https://www.gnu.org/licenses/>.
+ */
+
+package main
+
+import (
+	"errors"
+	"fmt"
+	"log"
+	"net/http"
+	"sync/atomic"
+
+	"github.com/jackc/pgx/v5"
+)
+
+func handleIndex(w http.ResponseWriter, req *http.Request) {
+	sessionCookie, err := req.Cookie("session")
+	if errors.Is(err, http.ErrNoCookie) {
+		authURL, err := generateAuthorizationURL()
+		if err != nil {
+			wstr(
+				w,
+				http.StatusInternalServerError,
+				"Cannot generate authorization URL",
+			)
+			return
+		}
+		err = tmpl.ExecuteTemplate(
+			w,
+			"login",
+			struct {
+				AuthURL string
+				Notes   string
+			}{
+				authURL,
+				"",
+			},
+		)
+		if err != nil {
+			log.Println(err)
+			return
+		}
+		return
+	} else if err != nil {
+		wstr(w, http.StatusBadRequest, "Error: Unable to check cookie.")
+		return
+	}
+
+	var userID, userName, userDepartment string
+	err = db.QueryRow(
+		req.Context(),
+		"SELECT id, name, department FROM users WHERE session = $1",
+		sessionCookie.Value,
+	).Scan(&userID, &userName, &userDepartment)
+	if err != nil {
+		if errors.Is(err, pgx.ErrNoRows) {
+			authURL, err := generateAuthorizationURL()
+			if err != nil {
+				wstr(
+					w,
+					http.StatusInternalServerError,
+					"Cannot generate authorization URL",
+				)
+				return
+			}
+			err = tmpl.ExecuteTemplate(
+				w,
+				"login",
+				struct {
+					AuthURL string
+					Notes   string
+				}{
+					authURL,
+					"Your session is invalid or has expired.",
+				},
+			)
+			if err != nil {
+				log.Println(err)
+				return
+			}
+			return
+		}
+		wstr(
+			w,
+			http.StatusInternalServerError,
+			fmt.Sprintf(
+				"Error: Unexpected database error: %s",
+				err,
+			),
+		)
+		return
+	}
+
+	/* TODO: The below should be completed on-update. */
+	type groupT struct {
+		Handle  courseGroupT
+		Name    string
+		Courses *map[int]*courseT
+	}
+	_groups := make(map[courseGroupT]groupT)
+	for k, v := range courseGroups {
+		_coursemap := make(map[int]*courseT)
+		_groups[k] = groupT{
+			Handle:  k,
+			Name:    v,
+			Courses: &_coursemap,
+		}
+	}
+	courses.Range(func(key, value interface{}) bool {
+		courseID, ok := key.(int)
+		if !ok {
+			panic("courses map has non-\"int\" keys")
+		}
+		course, ok := value.(*courseT)
+		if !ok {
+			panic("courses map has non-\"*courseT\" items")
+		}
+		(*_groups[course.Group].Courses)[courseID] = course
+		return true
+	})
+
+	if userDepartment == staffDepartment {
+		err := tmpl.ExecuteTemplate(
+			w,
+			"staff",
+			struct {
+				Name   string
+				State  uint32
+				Groups *map[courseGroupT]groupT
+			}{
+				userName,
+				state,
+				&_groups,
+			},
+		)
+		if err != nil {
+			log.Println(err)
+		}
+		return
+	}
+
+	if atomic.LoadUint32(&state) == 0 {
+		err := tmpl.ExecuteTemplate(
+			w,
+			"student_disabled",
+			struct {
+				Name       string
+				Department string
+			}{
+				userName,
+				userDepartment,
+			},
+		)
+		if err != nil {
+			log.Println(err)
+		}
+		return
+	}
+
+	err = tmpl.ExecuteTemplate(
+		w,
+		"student",
+		struct {
+			Name       string
+			Department string
+			Groups     *map[courseGroupT]groupT
+		}{
+			userName,
+			userDepartment,
+			&_groups,
+		},
+	)
+	if err != nil {
+		log.Println(err)
+		return
+	}
+}