Feat: works on my machine

2025-10-16 05:07:56 +08:00
commit 5bbab63a2c
37 changed files with 4553 additions and 0 deletions
--- a/handlers/auth/handlers.go
+++ b/handlers/auth/handlers.go
@@ -0,0 +1,11 @@
+package auth
+
+import "github.com/uptrace/bun"
+
+type Handlers struct {
+	db *bun.DB
+}
+
+func NewHandlers(db *bun.DB) *Handlers {
+	return &Handlers{db: db}
+}
--- a/handlers/auth/postLogin.go
+++ b/handlers/auth/postLogin.go
@@ -0,0 +1,59 @@
+package auth
+
+import (
+	"net/http"
+
+	"gitea.konchin.com/ytshih/inp2025/middlewares"
+	"gitea.konchin.com/ytshih/inp2025/models"
+	"gitea.konchin.com/ytshih/inp2025/types"
+	"gitea.konchin.com/ytshih/inp2025/utils"
+
+	"github.com/uptrace/bunrouter"
+)
+
+func (self *Handlers) PostLogin(
+	w http.ResponseWriter,
+	req bunrouter.Request,
+) error {
+	ctx := req.Context()
+	user, ok := ctx.Value(types.UserKey).(models.User)
+	if !ok {
+		return middlewares.HTTPError{
+			StatusCode: http.StatusUnauthorized,
+			Message:    "user not login",
+		}
+	}
+
+	res, err := self.db.NewUpdate().
+		Model((*models.User)(nil)).
+		Set("login_count = login_count + ?", 1).
+		Set("is_logged = ?", true).
+		Where("is_logged = ?", false).
+		Where("username = ?", user.Username).
+		Exec(ctx)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to update login count",
+			OriginError: err,
+		}
+	}
+	if cnt, err := res.RowsAffected(); err != nil || cnt == 0 {
+		if err != nil {
+			return middlewares.HTTPError{
+				StatusCode:  http.StatusInternalServerError,
+				Message:     "failed to get affected row count",
+				OriginError: err,
+			}
+		}
+		// debug
+		return utils.Success(w)
+		if cnt == 0 {
+			return middlewares.HTTPError{
+				StatusCode: http.StatusUnauthorized,
+				Message:    "already logged in",
+			}
+		}
+	}
+	return utils.Success(w)
+}
--- a/handlers/auth/postLogout.go
+++ b/handlers/auth/postLogout.go
@@ -0,0 +1,39 @@
+package auth
+
+import (
+	"net/http"
+
+	"gitea.konchin.com/ytshih/inp2025/middlewares"
+	"gitea.konchin.com/ytshih/inp2025/models"
+	"gitea.konchin.com/ytshih/inp2025/types"
+	"gitea.konchin.com/ytshih/inp2025/utils"
+	"github.com/uptrace/bunrouter"
+)
+
+func (self *Handlers) PostLogout(
+	w http.ResponseWriter,
+	req bunrouter.Request,
+) error {
+	ctx := req.Context()
+	user, ok := ctx.Value(types.UserKey).(models.User)
+	if !ok {
+		return middlewares.HTTPError{
+			StatusCode: http.StatusUnauthorized,
+			Message:    "user not login",
+		}
+	}
+
+	_, err := self.db.NewUpdate().
+		Set("is_logged = ?", false).
+		Where("username = ?", user.Username).
+		Exec(ctx)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to update logged in status",
+			OriginError: err,
+		}
+	}
+
+	return utils.Success(w)
+}
--- a/handlers/auth/postRegister.go
+++ b/handlers/auth/postRegister.go
@@ -0,0 +1,66 @@
+package auth
+
+import (
+	"encoding/json"
+	"io"
+	"net/http"
+
+	"gitea.konchin.com/ytshih/inp2025/middlewares"
+	"gitea.konchin.com/ytshih/inp2025/models"
+	"gitea.konchin.com/ytshih/inp2025/utils"
+	"github.com/uptrace/bunrouter"
+)
+
+func (self *Handlers) PostRegister(
+	w http.ResponseWriter,
+	req bunrouter.Request,
+) error {
+	ctx := req.Context()
+
+	b, err := io.ReadAll(req.Body)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusBadRequest,
+			Message:     "failed to read body payload",
+			OriginError: err,
+		}
+	}
+
+	var user models.User
+	if err := json.Unmarshal(b, &user); err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusBadRequest,
+			Message:     "failed to unmarshal json into user",
+			OriginError: err,
+		}
+	}
+
+	res, err := self.db.NewInsert().
+		Model(&user).
+		On("CONFLICT (username) DO NOTHING").
+		Exec(ctx)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to insert user",
+			OriginError: err,
+		}
+	}
+	if cnt, err := res.RowsAffected(); err != nil || cnt == 0 {
+		if err != nil {
+			return middlewares.HTTPError{
+				StatusCode:  http.StatusInternalServerError,
+				Message:     "failed to get affected row count",
+				OriginError: err,
+			}
+		}
+		if cnt == 0 {
+			return middlewares.HTTPError{
+				StatusCode: http.StatusBadRequest,
+				Message:    "username already exist",
+			}
+		}
+	}
+
+	return utils.Success(w)
+}
--- a/handlers/wordle/getState.go
+++ b/handlers/wordle/getState.go
@@ -0,0 +1,55 @@
+package wordle
+
+import (
+	"net/http"
+
+	"gitea.konchin.com/ytshih/inp2025/middlewares"
+	"gitea.konchin.com/ytshih/inp2025/types"
+	"gitea.konchin.com/ytshih/inp2025/utils"
+	"github.com/gorilla/websocket"
+	"github.com/uptrace/bunrouter"
+	"github.com/vmihailenco/msgpack/v5"
+)
+
+func (self *Handlers) GetState(
+	w http.ResponseWriter,
+	req bunrouter.Request,
+) error {
+	// fmt.Fprintf(os.Stderr, "GET /api/state\n")
+	c, err := self.upgrader.Upgrade(w, req.Request, nil)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusInternalServerError,
+			Message:     "failed to upgrade websocket",
+			OriginError: err,
+		}
+	}
+	defer c.Close()
+
+	username, _, ok := req.BasicAuth()
+	if !ok {
+		return middlewares.HTTPError{
+			StatusCode: http.StatusBadRequest,
+			Message:    "username not exist",
+		}
+	}
+	dataCh := make(chan types.WordleState)
+	self.opCh <- &OperationSubs{
+		Username: username,
+		SubsCh:   &dataCh,
+	}
+
+	for data := range dataCh {
+		b, err := msgpack.Marshal(data)
+		if err != nil {
+			return middlewares.HTTPError{
+				StatusCode:  http.StatusInternalServerError,
+				Message:     "failed to marshal data into msgpack",
+				OriginError: err,
+			}
+		}
+		c.WriteMessage(websocket.BinaryMessage, b)
+	}
+
+	return utils.Success(w)
+}
--- a/handlers/wordle/handlers.go
+++ b/handlers/wordle/handlers.go
@@ -0,0 +1,98 @@
+package wordle
+
+import (
+	"fmt"
+	"net/http"
+
+	"gitea.konchin.com/ytshih/inp2025/types"
+	"github.com/gorilla/websocket"
+)
+
+type Operation interface {
+	Run(self *Handlers) error
+}
+
+type OperationSubs struct {
+	Username string
+	SubsCh   *chan types.WordleState
+}
+
+func (op *OperationSubs) Run(self *Handlers) error {
+	_, ok := self.state.History[op.Username]
+	if !ok {
+		self.state.History[op.Username] = [types.GUESS_COUNT_LIMIT]types.Guess{}
+	}
+	// fmt.Fprintf(os.Stderr, "[DEBUG] %+v\n", self.state.History)
+	self.subs = append(self.subs, op.SubsCh)
+
+	return nil
+}
+
+type OperationGuess struct {
+	Username string `msgpack:"username"`
+	Guess    string `msgpack:"guess"`
+}
+
+func (op *OperationGuess) Run(self *Handlers) error {
+	self.state.CurrentGuess[op.Username] = op.Guess
+
+	if len(self.state.CurrentGuess) < len(self.state.History) {
+		return nil
+	}
+
+	for user, guess := range self.state.CurrentGuess {
+		if guess == self.ans {
+			self.state.GameEnd = true
+		}
+		guesses := self.state.History[user]
+		guesses[self.state.Round] = types.NewGuess(guess, self.ans)
+		self.state.History[user] = guesses
+	}
+	self.state.CurrentGuess = make(map[types.UsernameType]string)
+	self.state.Round++
+	if self.state.Round >= types.GUESS_COUNT_LIMIT {
+		self.state.GameEnd = true
+	}
+
+	return nil
+}
+
+type Handlers struct {
+	ans      string
+	state    types.WordleState
+	upgrader websocket.Upgrader
+	subs     []*chan types.WordleState
+
+	opCh chan Operation
+}
+
+func NewHandlers(ans string) *Handlers {
+	ret := &Handlers{
+		ans:   ans,
+		state: types.NewWordleState(),
+		upgrader: websocket.Upgrader{
+			ReadBufferSize:  1024,
+			WriteBufferSize: 1024,
+			CheckOrigin: func(r *http.Request) bool {
+				return true
+			},
+		},
+
+		opCh: make(chan Operation),
+	}
+	go ret.GameFlow()
+	return ret
+}
+
+func (self *Handlers) GameFlow() {
+	for op := range self.opCh {
+		if err := op.Run(self); err != nil {
+			panic(fmt.Errorf("game flow operation failed, %w", err))
+		}
+
+		// fmt.Fprintf(os.Stderr, "[DEBUG] %+v\n", self.state.History)
+		for _, ch := range self.subs {
+			*ch <- self.state
+		}
+	}
+}
--- a/handlers/wordle/postGuess.go
+++ b/handlers/wordle/postGuess.go
@@ -0,0 +1,54 @@
+package wordle
+
+import (
+	"io"
+	"net/http"
+
+	"gitea.konchin.com/ytshih/inp2025/middlewares"
+	"gitea.konchin.com/ytshih/inp2025/utils"
+	"github.com/uptrace/bunrouter"
+	"github.com/vmihailenco/msgpack/v5"
+)
+
+type PostGuessInput struct {
+	Guess string `msgpack:"guess"`
+}
+
+func (self *Handlers) PostGuess(
+	w http.ResponseWriter,
+	req bunrouter.Request,
+) error {
+	// fmt.Fprintf(os.Stderr, "POST /api/guess\n")
+	b, err := io.ReadAll(req.Body)
+	if err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusBadRequest,
+			Message:     "failed to read body payload",
+			OriginError: err,
+		}
+	}
+
+	var input PostGuessInput
+	if err := msgpack.Unmarshal(b, &input); err != nil {
+		return middlewares.HTTPError{
+			StatusCode:  http.StatusBadRequest,
+			Message:     "failed to unmarshal from msgpack",
+			OriginError: err,
+		}
+	}
+
+	username, _, ok := req.BasicAuth()
+	if !ok {
+		return middlewares.HTTPError{
+			StatusCode: http.StatusBadRequest,
+			Message:    "username not exist",
+		}
+	}
+
+	self.opCh <- &OperationGuess{
+		Username: username,
+		Guess:    input.Guess,
+	}
+
+	return utils.Success(w)
+}