Feat: works on my machine

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

60
middlewares/auth.go Normal file
View File

@@ -0,0 +1,60 @@
package middlewares
import (
"context"
"database/sql"
"errors"
"net/http"
"gitea.konchin.com/ytshih/inp2025/models"
"gitea.konchin.com/ytshih/inp2025/types"
"github.com/uptrace/bunrouter"
)
func (self *Handlers) Auth(
next bunrouter.HandlerFunc,
) bunrouter.HandlerFunc {
return func(w http.ResponseWriter, req bunrouter.Request) error {
ctx := req.Context()
username, password, ok := req.BasicAuth()
if !ok {
return HTTPError{
StatusCode: http.StatusNotFound,
Message: "username not exist",
}
}
dbUser := models.User{Username: username}
err := self.db.NewSelect().
Model(&dbUser).
WherePK().
Scan(ctx)
if errors.Is(err, sql.ErrNoRows) {
return HTTPError{
StatusCode: http.StatusUnauthorized,
Message: "username not exist",
OriginError: err,
}
}
if err != nil {
return HTTPError{
StatusCode: http.StatusInternalServerError,
Message: "failed to select user from db",
OriginError: err,
}
}
if password != dbUser.Password {
return HTTPError{
StatusCode: http.StatusUnauthorized,
Message: "password incorrect",
OriginError: err,
}
}
return next(w, req.WithContext(context.WithValue(
ctx, types.UserKey, models.User{
Username: username,
Password: password,
})))
}
}

View File

@@ -0,0 +1,63 @@
package middlewares
import (
"fmt"
"net/http"
"os"
"github.com/uptrace/bunrouter"
"go.uber.org/zap"
)
type HTTPError struct {
StatusCode int `json:"code"`
Message string `json:"message"`
OriginError error `json:"-"`
}
func (e HTTPError) Error() string {
return e.Message
}
func NewHTTPError(err error) HTTPError {
return HTTPError{
StatusCode: http.StatusInternalServerError,
Message: "Internal server error with unknown reason",
OriginError: err,
}
}
func (self *Handlers) ErrorHandler(
next bunrouter.HandlerFunc,
) bunrouter.HandlerFunc {
return func(w http.ResponseWriter, req bunrouter.Request) error {
err := next(w, req)
var httpErr HTTPError
switch err := err.(type) {
case nil:
return nil
case HTTPError:
httpErr = err
default:
fmt.Fprintf(os.Stderr, "unhandled error, %v\n", err)
zap.L().Error("unhandled error",
zap.Error(err))
httpErr = NewHTTPError(err)
}
if httpErr.OriginError == nil {
zap.L().Warn(httpErr.Message)
} else {
zap.L().Warn(httpErr.Message,
zap.Error(httpErr.OriginError))
}
w.WriteHeader(httpErr.StatusCode)
_ = bunrouter.JSON(w, httpErr)
return err
}
}

11
middlewares/handlers.go Normal file
View File

@@ -0,0 +1,11 @@
package middlewares
import "github.com/uptrace/bun"
type Handlers struct {
db *bun.DB
}
func NewHandlers(db *bun.DB) *Handlers {
return &Handlers{db: db}
}