65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package handle
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"net/http"
|
|
"net/netip"
|
|
|
|
"github.com/julienschmidt/httprouter"
|
|
"github.com/samber/lo"
|
|
"github.com/xmdhs/authlib-skin/model"
|
|
"github.com/xmdhs/authlib-skin/service"
|
|
"github.com/xmdhs/authlib-skin/utils"
|
|
)
|
|
|
|
func (h *Handel) Reg() httprouter.Handle {
|
|
return func(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
|
|
ctx := r.Context()
|
|
|
|
u, err := utils.DeCodeBody[model.User](r.Body, h.validate)
|
|
if err != nil {
|
|
h.logger.InfoContext(ctx, err.Error())
|
|
handleError(ctx, w, err.Error(), model.ErrInput, 400)
|
|
return
|
|
}
|
|
rip, err := getPrefix(r, h.config.RaelIP)
|
|
if err != nil {
|
|
h.logger.WarnContext(ctx, err.Error())
|
|
handleError(ctx, w, err.Error(), model.ErrUnknown, 500)
|
|
return
|
|
}
|
|
err = h.webService.Reg(ctx, u, rip)
|
|
if err != nil {
|
|
if errors.Is(err, service.ErrExistUser) {
|
|
h.logger.DebugContext(ctx, err.Error())
|
|
handleError(ctx, w, err.Error(), model.ErrExistUser, 400)
|
|
return
|
|
}
|
|
if errors.Is(err, service.ErrRegLimit) {
|
|
h.logger.DebugContext(ctx, err.Error())
|
|
handleError(ctx, w, err.Error(), model.ErrRegLimit, 400)
|
|
return
|
|
}
|
|
h.logger.WarnContext(ctx, err.Error())
|
|
handleError(ctx, w, err.Error(), model.ErrService, 500)
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func getPrefix(r *http.Request, fromHeader bool) (string, error) {
|
|
ip, err := utils.GetIP(r, fromHeader)
|
|
if err != nil {
|
|
return "", fmt.Errorf("getPrefix: %w", err)
|
|
}
|
|
ipa, err := netip.ParseAddr(ip)
|
|
if err != nil {
|
|
return "", fmt.Errorf("getPrefix: %w", err)
|
|
}
|
|
if ipa.Is6() {
|
|
return lo.Must1(ipa.Prefix(48)).String(), nil
|
|
}
|
|
return lo.Must1(ipa.Prefix(24)).String(), nil
|
|
}
|