Feat: works on my machine

This commit is contained in:
2025-10-16 05:07:56 +08:00
commit ce3af7c682
37 changed files with 4537 additions and 0 deletions

11
handlers/auth/handlers.go Normal file
View File

@@ -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}
}

View File

@@ -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)
}

View File

@@ -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)
}

View File

@@ -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)
}