commit b924e94c63e3da299273baf343c076aa059abfe5 Author: juzeon <812312770@qq.com> Date: Wed Nov 23 20:56:28 2022 +0800 直接完成 diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..eaefa03 --- /dev/null +++ b/.gitignore @@ -0,0 +1,2 @@ +.idea +integrity-checkin diff --git a/dao/checkin.go b/dao/checkin.go new file mode 100644 index 0000000..dc25531 --- /dev/null +++ b/dao/checkin.go @@ -0,0 +1,104 @@ +package dao + +import ( + "errors" + "gorm.io/gorm" + "integrity-checkin/data/dto" + "integrity-checkin/data/po" + "time" +) + +type CheckinDAO struct { + Tx *gorm.DB +} + +func (c CheckinDAO) FindUserActivityByDay(userID int, activityID int, day string) *po.UserActivity { + var userActivity po.UserActivity + err := c.Tx.Take(&userActivity, "user_id=? and activity_id=? and day=?", userID, activityID, day).Error + if err == gorm.ErrRecordNotFound { + return nil + } + if err != nil { + panic(err) + } + return &userActivity +} +func (c CheckinDAO) CreateCheckin(userID int, activityID int, image string) (*po.UserActivity, error) { + if activity := c.FindActivityByID(activityID); activity == nil { + return nil, errors.New("activityID不存在") + } + userActivity := po.UserActivity{ + UserID: userID, + UserActivityForm: po.UserActivityForm{ + ActivityID: activityID, + Image: image, + }, + Status: 1, + Day: time.Now().Format("20060102"), + } + err := c.Tx.Create(&userActivity).Error + if err != nil { + panic(err) + } + return &userActivity, nil +} +func (c CheckinDAO) FindActivityByID(id int) *po.Activity { + var activity po.Activity + err := c.Tx.Take(&activity, id).Error + if err == gorm.ErrRecordNotFound { + return nil + } + if err != nil { + panic(err) + } + return &activity +} +func (c CheckinDAO) FindUserActivityByID(id int) *po.UserActivity { + var userActivity po.UserActivity + err := c.Tx.Take(&userActivity, id).Error + if err == gorm.ErrRecordNotFound { + return nil + } + if err != nil { + panic(err) + } + return &userActivity +} +func (c CheckinDAO) SaveUserActivity(userActivity *po.UserActivity) { + err := c.Tx.Save(userActivity).Error + if err != nil { + panic(err) + } +} +func (c CheckinDAO) FindAllActivities() []po.Activity { + var arr []po.Activity + err := c.Tx.Find(&arr).Error + if err != nil { + panic(err) + } + return arr +} +func getUserActivityExtendedSQL(where string) string { + return "select ua.*,a.category activity_category," + + "a.name activity_name from user_activity ua left join activities a on a.id=ua.activity_id " + where + + " order by created_at desc" +} +func (c CheckinDAO) FindAllUserActivities() []dto.UserActivityExtended { + var arr []dto.UserActivityExtended + err := c.Tx.Raw(getUserActivityExtendedSQL("")).Find(&arr).Error + if err != nil { + panic(err) + } + return arr +} +func (c CheckinDAO) FindUserActivityExtendedByID(id int) *dto.UserActivityExtended { + var userActivityExtended dto.UserActivityExtended + err := c.Tx.Raw(getUserActivityExtendedSQL("where id=?"), id).Take(&userActivityExtended).Error + if err == gorm.ErrRecordNotFound { + return nil + } + if err != nil { + panic(err) + } + return &userActivityExtended +} diff --git a/dao/dao.go b/dao/dao.go new file mode 100644 index 0000000..e0c12f9 --- /dev/null +++ b/dao/dao.go @@ -0,0 +1,22 @@ +package dao + +import ( + "fmt" + "gorm.io/driver/mysql" + "gorm.io/gorm" +) + +var DB *gorm.DB + +func Setup() { + db, err := gorm.Open(mysql.Open(fmt.Sprintf("%v:%v@tcp(%v:%v)/%v?charset=utf8mb4&parseTime=True&loc=Local", + "integrity", + "integrity", + "202.194.15.130", + 3306, + "integrity")), &gorm.Config{}) + if err != nil { + panic(err) + } + DB = db +} diff --git a/dao/user.go b/dao/user.go new file mode 100644 index 0000000..dbbfea7 --- /dev/null +++ b/dao/user.go @@ -0,0 +1,51 @@ +package dao + +import ( + "gorm.io/gorm" + "integrity-checkin/data/po" +) + +type UserDAO struct { + Tx *gorm.DB +} + +func (u UserDAO) FindUserByUsername(username string) *po.User { + var user po.User + err := u.Tx.Take(&user, "username=?", username).Error + if err == gorm.ErrRecordNotFound { + return nil + } + if err != nil { + panic(err) + } + return &user +} +func (u UserDAO) FindUserByID(id int) *po.User { + var user po.User + err := u.Tx.Take(&user, id).Error + if err == gorm.ErrRecordNotFound { + return nil + } + if err != nil { + panic(err) + } + return &user +} +func (u UserDAO) SaveUser(user *po.User) { + err := u.Tx.Save(user).Error + if err != nil { + panic(err) + } +} +func (u UserDAO) CreateUser(username string, realName string) *po.User { + user := po.User{ + Username: username, + RealName: realName, + RoleID: 1, + } + err := u.Tx.Create(&user).Error + if err != nil { + panic(err) + } + return &user +} diff --git a/data/dto/user.go b/data/dto/user.go new file mode 100644 index 0000000..e14e358 --- /dev/null +++ b/data/dto/user.go @@ -0,0 +1,17 @@ +package dto + +import "integrity-checkin/data/po" + +type UserClaims struct { + UserID int `json:"user_id"` + RoleID int `json:"role_id"` +} +type SDULoginData struct { + RealName string + Username string +} +type UserActivityExtended struct { + po.UserActivity + ActivityCategory string `json:"activity_category"` + ActivityName string `json:"activity_name"` +} diff --git a/data/itf/itf.go b/data/itf/itf.go new file mode 100644 index 0000000..4d067f5 --- /dev/null +++ b/data/itf/itf.go @@ -0,0 +1,5 @@ +package itf + +type IValidator interface { + Validate() error +} diff --git a/data/po/checkin.go b/data/po/checkin.go new file mode 100644 index 0000000..eae1b25 --- /dev/null +++ b/data/po/checkin.go @@ -0,0 +1,44 @@ +package po + +import ( + "errors" + "strings" + "time" +) + +type Activity struct { + ID int `json:"id"` + Category string `json:"category"` + Name string `json:"name"` + Credit int `json:"credit"` +} + +func (a Activity) TableName() string { + return "activities" +} + +type UserActivity struct { + ID int `json:"id"` + UserID int `json:"user_id"` + UserActivityForm + Status int `json:"status"` + Day string `json:"day"` + CreatedAt time.Time `json:"created_at"` + UpdatedAt time.Time `json:"updated_at"` +} + +func (u UserActivity) TableName() string { + return "user_activity" +} + +type UserActivityForm struct { + ActivityID int `json:"activity_id" form:"activity_id" binding:"required"` + Image string `json:"image" form:"image" binding:"required,url"` +} + +func (u UserActivityForm) Validate() error { + if !strings.HasPrefix(u.Image, "https://m.360buyimg.com") { + return errors.New("不是京东图床链接") + } + return nil +} diff --git a/data/po/user.go b/data/po/user.go new file mode 100644 index 0000000..e81caa3 --- /dev/null +++ b/data/po/user.go @@ -0,0 +1,9 @@ +package po + +type User struct { + ID int `json:"id"` + Username string `json:"username"` + RealName string `json:"real_name"` + Credit int `json:"credit"` + RoleID int `json:"role_id"` +} diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..338db62 --- /dev/null +++ b/go.mod @@ -0,0 +1,36 @@ +module integrity-checkin + +go 1.19 + +require ( + github.com/PuerkitoBio/goquery v1.8.0 // indirect + github.com/andybalholm/cascadia v1.3.1 // indirect + github.com/ekyoung/gin-nice-recovery v0.0.0-20160510022553-1654dca486db // indirect + github.com/gin-contrib/sse v0.1.0 // indirect + github.com/gin-gonic/gin v1.8.1 // indirect + github.com/go-errors/errors v1.4.2 // indirect + github.com/go-playground/locales v0.14.0 // indirect + github.com/go-playground/universal-translator v0.18.0 // indirect + github.com/go-playground/validator/v10 v10.11.1 // indirect + github.com/go-resty/resty/v2 v2.7.0 // indirect + github.com/go-sql-driver/mysql v1.6.0 // indirect + github.com/goccy/go-json v0.9.11 // indirect + github.com/golang-jwt/jwt v3.2.2+incompatible // indirect + github.com/jinzhu/inflection v1.0.0 // indirect + github.com/jinzhu/now v1.1.5 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/leodido/go-urn v1.2.1 // indirect + github.com/mattn/go-isatty v0.0.16 // indirect + github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/pelletier/go-toml/v2 v2.0.6 // indirect + github.com/ugorji/go/codec v1.2.7 // indirect + golang.org/x/crypto v0.3.0 // indirect + golang.org/x/net v0.2.0 // indirect + golang.org/x/sys v0.2.0 // indirect + golang.org/x/text v0.4.0 // indirect + google.golang.org/protobuf v1.28.1 // indirect + gopkg.in/yaml.v2 v2.4.0 // indirect + gorm.io/driver/mysql v1.4.4 // indirect + gorm.io/gorm v1.24.2 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..b3af8d4 --- /dev/null +++ b/go.sum @@ -0,0 +1,113 @@ +github.com/PuerkitoBio/goquery v1.8.0 h1:PJTF7AmFCFKk1N6V6jmKfrNH9tV5pNE6lZMkG0gta/U= +github.com/PuerkitoBio/goquery v1.8.0/go.mod h1:ypIiRMtY7COPGk+I/YbZLbxsxn9g5ejnI2HSMtkjZvI= +github.com/andybalholm/cascadia v1.3.1 h1:nhxRkql1kdYCc8Snf7D5/D3spOX+dBgjA6u8x004T2c= +github.com/andybalholm/cascadia v1.3.1/go.mod h1:R4bJ1UQfqADjvDa4P6HZHLh/3OxWWEqc0Sk8XGwHqvA= +github.com/creack/pty v1.1.9/go.mod h1:oKZEueFk5CKHvIhNR5MUki03XCEU+Q6VDXinZuGJ33E= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/ekyoung/gin-nice-recovery v0.0.0-20160510022553-1654dca486db h1:oZ4U9IqO8NS+61OmGTBi8vopzqTRxwQeogyBHdrhjbc= +github.com/ekyoung/gin-nice-recovery v0.0.0-20160510022553-1654dca486db/go.mod h1:Pk7/9x6tyChFTkahDvLBQMlvdsWvfC+yU8HTT5VD314= +github.com/gin-contrib/sse v0.1.0 h1:Y/yl/+YNO8GZSjAhjMsSuLt29uWRFHdHYUb5lYOV9qE= +github.com/gin-contrib/sse v0.1.0/go.mod h1:RHrZQHXnP2xjPF+u1gW/2HnVO7nvIa9PG3Gm+fLHvGI= +github.com/gin-gonic/gin v1.8.1 h1:4+fr/el88TOO3ewCmQr8cx/CtZ/umlIRIs5M4NTNjf8= +github.com/gin-gonic/gin v1.8.1/go.mod h1:ji8BvRH1azfM+SYow9zQ6SZMvR8qOMZHmsCuWR9tTTk= +github.com/go-errors/errors v1.4.2 h1:J6MZopCL4uSllY1OfXM374weqZFFItUbrImctkmUxIA= +github.com/go-errors/errors v1.4.2/go.mod h1:sIVyrIiJhuEF+Pj9Ebtd6P/rEYROXFi3BopGUQ5a5Og= +github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4= +github.com/go-playground/locales v0.14.0 h1:u50s323jtVGugKlcYeyzC0etD1HifMjqmJqb8WugfUU= +github.com/go-playground/locales v0.14.0/go.mod h1:sawfccIbzZTqEDETgFXqTho0QybSa7l++s0DH+LDiLs= +github.com/go-playground/universal-translator v0.18.0 h1:82dyy6p4OuJq4/CByFNOn/jYrnRPArHwAcmLoJZxyho= +github.com/go-playground/universal-translator v0.18.0/go.mod h1:UvRDBj+xPUEGrFYl+lu/H90nyDXpg0fqeB/AQUGNTVA= +github.com/go-playground/validator/v10 v10.11.1 h1:prmOlTVv+YjZjmRmNSF3VmspqJIxJWXmqUsHwfTRRkQ= +github.com/go-playground/validator/v10 v10.11.1/go.mod h1:i+3WkQ1FvaUjjxh1kSvIA4dMGDBiPU55YFDl0WbKdWU= +github.com/go-resty/resty/v2 v2.7.0 h1:me+K9p3uhSmXtrBZ4k9jcEAfJmuC8IivWHwaLZwPrFY= +github.com/go-resty/resty/v2 v2.7.0/go.mod h1:9PWDzw47qPphMRFfhsyk0NnSgvluHcljSMVIq3w7q0I= +github.com/go-sql-driver/mysql v1.6.0 h1:BCTh4TKNUYmOmMUcQ3IipzF5prigylS7XXjEkfCHuOE= +github.com/go-sql-driver/mysql v1.6.0/go.mod h1:DCzpHaOWr8IXmIStZouvnhqoel9Qv2LBy8hT2VhHyBg= +github.com/goccy/go-json v0.9.11 h1:/pAaQDLHEoCq/5FFmSKBswWmK6H0e8g4159Kc/X/nqk= +github.com/goccy/go-json v0.9.11/go.mod h1:6MelG93GURQebXPDq3khkgXZkazVtN9CRI+MGFi0w8I= +github.com/golang-jwt/jwt v3.2.2+incompatible h1:IfV12K8xAKAnZqdXVzCZ+TOjboZ2keLg81eXfW3O+oY= +github.com/golang-jwt/jwt v3.2.2+incompatible/go.mod h1:8pz2t5EyA70fFQQSrl6XZXzqecmYZeUEB8OUGHkxJ+I= +github.com/golang/protobuf v1.5.0/go.mod h1:FsONVRAS9T7sI+LIUmWTfcYkHO4aIWwzhcaSAoJOfIk= +github.com/google/go-cmp v0.5.5/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/jinzhu/inflection v1.0.0 h1:K317FqzuhWc8YvSVlFMCCUb36O/S9MCKRDI7QkRKD/E= +github.com/jinzhu/inflection v1.0.0/go.mod h1:h+uFLlag+Qp1Va5pdKtLDYj+kHp5pxUVkryuEj+Srlc= +github.com/jinzhu/now v1.1.4/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/jinzhu/now v1.1.5 h1:/o9tlHleP7gOFmsnYNz3RGnqzefHA47wQpKrrdTIwXQ= +github.com/jinzhu/now v1.1.5/go.mod h1:d3SSVoowX0Lcu0IBviAWJpolVfI5UJVZZ7cO71lE/z8= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo= +github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI= +github.com/kr/pretty v0.3.0/go.mod h1:640gp4NfQd8pI5XOwp5fnNeVWj67G7CFk/SaSQn7NBk= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= +github.com/leodido/go-urn v1.2.1 h1:BqpAaACuzVSgi/VLzGZIobT2z4v53pjosyNd9Yv6n/w= +github.com/leodido/go-urn v1.2.1/go.mod h1:zt4jvISO2HfUBqxjfIshjdMTYS56ZS/qv49ictyFfxY= +github.com/mattn/go-isatty v0.0.16 h1:bq3VjFmv/sOjHtdEhmkEV4x1AJtvUvOJ2PFAZ5+peKQ= +github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/pelletier/go-toml/v2 v2.0.6 h1:nrzqCb7j9cDFj2coyLNLaZuJTLjWjlaz6nvTvIwycIU= +github.com/pelletier/go-toml/v2 v2.0.6/go.mod h1:eumQOmlWiOPt5WriQQqoM5y18pDHwha2N+QD+EUNTek= +github.com/pkg/diff v0.0.0-20210226163009-20ebb0f2a09e/go.mod h1:pJLUxLENpZxwdsKMEsNbx1VGcRFpLqf3715MtcvvzbA= +github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4= +github.com/rogpeppe/go-internal v1.6.1/go.mod h1:xXDCJY+GAPziupqXw64V24skbSoqbTEfhy4qGm1nDQc= +github.com/rogpeppe/go-internal v1.8.0/go.mod h1:WmiCO8CzOY8rg0OYDC4/i/2WRWAB6poM+XZ2dLUbcbE= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.6.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/ugorji/go v1.2.7 h1:qYhyWUUd6WbiM+C6JZAUkIJt/1WrjzNHY9+KCIjVqTo= +github.com/ugorji/go v1.2.7/go.mod h1:nF9osbDWLy6bDVv/Rtoh6QgnvNDpmCalQV5urGCCS6M= +github.com/ugorji/go/codec v1.2.7 h1:YPXUKf7fYbp/y8xloBqZOw2qaVggbfwMlI8WM3wZUJ0= +github.com/ugorji/go/codec v1.2.7/go.mod h1:WGN1fab3R1fzQlVQTkfxVtIBhWDRqOviHU95kRgeqEY= +golang.org/x/crypto v0.0.0-20211215153901-e495a2d5b3d3/go.mod h1:IxCIyHEi3zRg3s0A5j5BB6A9Jmi73HwBIUl50j+osU4= +golang.org/x/crypto v0.3.0 h1:a06MkbcxBrEFc0w0QIZWXrH/9cCX6KJyWbBOIwAn+7A= +golang.org/x/crypto v0.3.0/go.mod h1:hebNnKkNXi2UzZN1eVRvBB7co0a+JxK6XbPiWVs/3J4= +golang.org/x/net v0.0.0-20210916014120-12bc252f5db8/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/net v0.0.0-20211029224645-99673261e6eb/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/net v0.0.0-20211112202133-69e39bad7dc2/go.mod h1:9nx3DQGgdP8bBQD5qxJ1jj9UTztislL4KSBs9R2vV5Y= +golang.org/x/net v0.2.0 h1:sZfSu1wtKLGlWI4ZZayP0ck9Y73K1ynO6gqzTdBVdPU= +golang.org/x/net v0.2.0/go.mod h1:KqCZLdyyvdV855qA2rE3GC2aiw5xGR5TEjj8smXukLY= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210423082822-04245dca01da/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20210806184541-e5e7981a1069/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.2.0 h1:ljd4t30dBnAvMZaQCevtY0xLLD0A+bRZXbgLMLU1F/A= +golang.org/x/sys v0.2.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/text v0.3.6/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/text v0.4.0 h1:BrVqGRd7+k1DiOgtnFvAkoQEWQvBc25ouMJM6429SFg= +golang.org/x/text v0.4.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/protobuf v1.26.0-rc.1/go.mod h1:jlhhOSvTdKEhbULTjvd4ARK9grFBp09yW+WbY/TyQbw= +google.golang.org/protobuf v1.28.1 h1:d0NfwRgPtno5B1Wa6L2DAG+KivqkdutMf1UhdNx175w= +google.golang.org/protobuf v1.28.1/go.mod h1:HV8QOd/L58Z+nl8r43ehVNZIU/HEI6OcFqwMG9pJV4I= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q= +gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= +gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.0-20210107192922-496545a6307b/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gorm.io/driver/mysql v1.4.4 h1:MX0K9Qvy0Na4o7qSC/YI7XxqUw5KDw01umqgID+svdQ= +gorm.io/driver/mysql v1.4.4/go.mod h1:BCg8cKI+R0j/rZRQxeKis/forqRwRSYOR8OM3Wo6hOM= +gorm.io/gorm v1.23.8/go.mod h1:l2lP/RyAtc1ynaTjFksBde/O8v9oOGIApu2/xRitmZk= +gorm.io/gorm v1.24.2 h1:9wR6CFD+G8nOusLdvkZelOEhpJVwwHzpQOUM+REd6U0= +gorm.io/gorm v1.24.2/go.mod h1:DVrVomtaYTbqs7gB/x2uVvqnXzv0nqjB396B8cG4dBA= diff --git a/main.go b/main.go new file mode 100644 index 0000000..823d03a --- /dev/null +++ b/main.go @@ -0,0 +1,35 @@ +package main + +import ( + "fmt" + nice "github.com/ekyoung/gin-nice-recovery" + "github.com/gin-gonic/gin" + "integrity-checkin/dao" + "integrity-checkin/pkg/app" + "integrity-checkin/pkg/httpclient" + "integrity-checkin/router" + "integrity-checkin/service" + "math/rand" + "time" +) + +func main() { + Setup() + httpclient.Setup() + engine := gin.New() + engine.Use(gin.Logger(), + nice.Recovery(func(c *gin.Context, err interface{}) { + aw := app.NewWrapper(c) + aw.Error("内部错误，请重试：" + fmt.Sprintf("%v", err)).SendJSON() + })) + dao.Setup() + service.Setup() + router.Setup(engine) + err := engine.Run(fmt.Sprintf(":%v", "8964")) + if err != nil { + panic(err) + } +} +func Setup() { + rand.Seed(time.Now().UnixNano()) +} diff --git a/middleware/cors.go b/middleware/cors.go new file mode 100644 index 0000000..fb0156f --- /dev/null +++ b/middleware/cors.go @@ -0,0 +1,17 @@ +package middleware + +import "github.com/gin-gonic/gin" + +func Cors() gin.HandlerFunc { + return func(c *gin.Context) { + c.Writer.Header().Set("Access-Control-Allow-Origin", "*") + c.Writer.Header().Set("Access-Control-Allow-Credentials", "true") + c.Writer.Header().Set("Access-Control-Allow-Headers", "*") + c.Writer.Header().Set("Access-Control-Allow-Methods", "GET, HEAD, POST, PUT, DELETE, OPTIONS, PATCH") + if c.Request.Method == "OPTIONS" { + c.AbortWithStatus(204) + return + } + c.Next() + } +} diff --git a/middleware/jwt.go b/middleware/jwt.go new file mode 100644 index 0000000..dd5d47c --- /dev/null +++ b/middleware/jwt.go @@ -0,0 +1,31 @@ +package middleware + +import ( + "fmt" + "github.com/gin-gonic/gin" + "integrity-checkin/pkg/app" + "integrity-checkin/pkg/util" + "strings" +) + +func JWT(minRoleID int) gin.HandlerFunc { + return func(c *gin.Context) { + aw := app.NewWrapper(c) + authHeader := c.GetHeader("Authorization") + if strings.HasPrefix(authHeader, "Bearer ") { + authHeader = authHeader[7:] + } + claims, err := util.ParseJWT(authHeader) + if err != nil { + aw.Error("该接口需要登录").SendJSON() + aw.Ctx.Abort() + return + } + if claims.RoleID < minRoleID { + aw.Error(fmt.Sprintf("该接口需要角色大于或等于：%v（请尝试退出后重登录）", minRoleID)).SendJSON() + aw.Ctx.Abort() + return + } + aw.Ctx.Set("userClaims", claims) + } +} diff --git a/pkg/app/handler_func.go b/pkg/app/handler_func.go new file mode 100644 index 0000000..5d92b76 --- /dev/null +++ b/pkg/app/handler_func.go @@ -0,0 +1,11 @@ +package app + +import "github.com/gin-gonic/gin" + +func HandlerFunc(handler func(aw *Wrapper) Result) func(c *gin.Context) { + return func(c *gin.Context) { + aw := NewWrapper(c) + res := handler(aw) + c.JSON(200, res) + } +} diff --git a/pkg/app/wrapper.go b/pkg/app/wrapper.go new file mode 100644 index 0000000..e6ea3e9 --- /dev/null +++ b/pkg/app/wrapper.go @@ -0,0 +1,72 @@ +package app + +import ( + "github.com/gin-gonic/gin" + "integrity-checkin/pkg/util" +) + +type Result struct { + Code int `json:"code"` + Msg string `json:"msg,omitempty"` + Data interface{} `json:"data,omitempty"` + wrapper *Wrapper +} + +func (r Result) SendJSON() { + r.wrapper.Ctx.JSON(200, r) +} + +type Wrapper struct { + Ctx *gin.Context +} + +func NewWrapper(c *gin.Context) *Wrapper { + return &Wrapper{Ctx: c} +} + +func (w Wrapper) OK() Result { + return Result{ + Code: 0, + Msg: "", + Data: nil, + wrapper: &w, + } +} +func (w Wrapper) Success(data interface{}) Result { + return Result{ + Code: 0, + Msg: "", + Data: data, + wrapper: &w, + } +} +func (w Wrapper) Error(msg string) Result { + return Result{ + Code: -1, + Msg: msg, + Data: nil, + wrapper: &w, + } +} +func (w Wrapper) ErrorWithCode(code int, msg string) Result { + return Result{ + Code: code, + Msg: msg, + Data: nil, + wrapper: &w, + } +} +func (w Wrapper) GetIP() string { + return w.Ctx.ClientIP() +} +func (w Wrapper) ExtractUserClaims() *util.UserClaims { + raw, exist := w.Ctx.Get("userClaims") + if !exist { + panic("userClaims not exists") + } + uc, ok := raw.(*util.UserClaims) + if !ok { + panic("userClaims failed to convert") + } + return uc +} diff --git a/pkg/httpclient/resty.go b/pkg/httpclient/resty.go new file mode 100644 index 0000000..c9102bd --- /dev/null +++ b/pkg/httpclient/resty.go @@ -0,0 +1,67 @@ +package httpclient + +import ( + "errors" + "github.com/go-resty/resty/v2" + "integrity-checkin/data/dto" + "regexp" + "time" +) + +var Client *resty.Client + +func Setup() { + Client = resty.New().SetTimeout(10 * time.Second) +} +func SDULogin(username string, password string) (*dto.SDULoginData, error) { + resp, err := Client.R().SetFormData(map[string]string{ + "username": username, + "password": password, + }).Post("https://pass.sdu.edu.cn/cas/restlet/tickets") + if err != nil { + return nil, err + } + result1 := string(resp.Body()) + if resp.StatusCode()/100 != 2 { + return nil, errors.New(result1) + } + resp, err = Client.R().SetBody("service=https://service.sdu.edu.cn/tp_up/view?m=up"). + Post("https://pass.sdu.edu.cn/cas/restlet/tickets/" + result1) + if err != nil { + return nil, err + } + result2 := string(resp.Body()) + if resp.StatusCode()/100 != 2 { + return nil, errors.New(result2) + } + resp, err = Client.R().SetQueryParams(map[string]string{ + "ticket": result2, + "service": "https://service.sdu.edu.cn/tp_up/view?m=up", + }).Get("https://pass.sdu.edu.cn/cas/serviceValidate") + if err != nil { + return nil, err + } + result3 := string(resp.Body()) + if resp.StatusCode()/100 != 2 { + return nil, errors.New(result3) + } + realName := "" + username = "" + if arr := regexp.MustCompile("(.*?)<").FindStringSubmatch(result3); len(arr) >= 2 { + realName = arr[1] + } else { + return nil, errors.New(result3) + } + if arr := regexp.MustCompile("(.*?)<").FindStringSubmatch(result3); len(arr) >= 2 { + username = arr[1] + } else { + return nil, errors.New(result3) + } + if realName == "" || username == "" { + return nil, errors.New(result3) + } + return &dto.SDULoginData{ + RealName: realName, + Username: username, + }, nil +} diff --git a/pkg/util/permission.go b/pkg/util/permission.go new file mode 100644 index 0000000..c089f81 --- /dev/null +++ b/pkg/util/permission.go @@ -0,0 +1,42 @@ +package util + +import ( + "github.com/golang-jwt/jwt" +) + +type UserClaims struct { + UserID int `json:"user_id"` + RoleID int `json:"role_id"` + jwt.StandardClaims +} + +const ( + JWTSecret = "fee516aa-8f5b-4248-a80c-bf467d40fd62" +) + +func ParseJWT(token string) (*UserClaims, error) { + tokenClaims, err := jwt.ParseWithClaims(token, &UserClaims{}, func(token *jwt.Token) (interface{}, error) { + return []byte(JWTSecret), nil + }) + if err != nil { + return nil, err + } + if tokenClaims != nil { + if claims, ok := tokenClaims.Claims.(*UserClaims); ok && tokenClaims.Valid { + return claims, nil + } + } + return nil, err +} +func GenerateJWT(userID int, roleID int) string { + token := jwt.NewWithClaims(jwt.SigningMethodHS256, UserClaims{ + UserID: userID, + RoleID: roleID, + StandardClaims: jwt.StandardClaims{}, + }) + str, err := token.SignedString([]byte(JWTSecret)) + if err != nil { + panic(err) + } + return str +} diff --git a/router/router.go b/router/router.go new file mode 100644 index 0000000..4eafe07 --- /dev/null +++ b/router/router.go @@ -0,0 +1,27 @@ +package router + +import ( + "github.com/gin-gonic/gin" + "integrity-checkin/middleware" + "integrity-checkin/pkg/app" + "integrity-checkin/service" +) + +func Setup(engine *gin.Engine) { + user := engine.Group("/user") + { + hub := service.ExUserService + // 测试方法，获取signed jwt + user.POST("/login", app.HandlerFunc(hub.Login)) + } + checkin := engine.Group("/checkin") + { + hub := service.ExCheckinService + checkin.GET("/list_activities", app.HandlerFunc(hub.ListActivities)) + checkin.Use(middleware.JWT(1)) + checkin.POST("/submit", app.HandlerFunc(hub.Submit)) + checkin.POST("/set_status", middleware.JWT(2), app.HandlerFunc(hub.SetStatus)) + checkin.GET("/get_user_activity", middleware.JWT(2), app.HandlerFunc(hub.GetUserActivity)) + checkin.GET("/list_user_activities", middleware.JWT(2), app.HandlerFunc(hub.ListUserActivities)) + } +} diff --git a/service/checkin.go b/service/checkin.go new file mode 100644 index 0000000..2246eb1 --- /dev/null +++ b/service/checkin.go @@ -0,0 +1,73 @@ +package service + +import ( + "integrity-checkin/data/po" + "integrity-checkin/pkg/app" + "strconv" + "time" +) + +type CheckinService struct { +} + +func (c CheckinService) Submit(aw *app.Wrapper) app.Result { + uc := aw.ExtractUserClaims() + var req po.UserActivityForm + if err := aw.Ctx.ShouldBind(&req); err != nil { + return aw.Error(err.Error()) + } + if err := req.Validate(); err != nil { + return aw.Error(err.Error()) + } + today := time.Now().Format("20060102") + if checkinDAO.FindUserActivityByDay(uc.UserID, req.ActivityID, today) != nil { + return aw.Error("您今天已参加过该项目，请明天再来") + } + userActivity, err := checkinDAO.CreateCheckin(uc.UserID, req.ActivityID, req.Image) + if err != nil { + return aw.Error(err.Error()) + } + return aw.Success(userActivity) +} +func (c CheckinService) SetStatus(aw *app.Wrapper) app.Result { + type SetStatusReq struct { + ID int `form:"id" binding:"required"` + Status int `form:"status" binding:"required"` + Credit int `form:"credit" binding:"required,gte=1"` + } + var req SetStatusReq + if err := aw.Ctx.ShouldBind(&req); err != nil { + return aw.Error(err.Error()) + } + userActivity := checkinDAO.FindUserActivityByID(req.ID) + if userActivity == nil { + return aw.Error("打卡不存在") + } + if userActivity.Status != 1 { + return aw.Error("已经审核过此打卡") + } + if req.Status == 1 { + return aw.Error("只能是 2通过，3拒绝") + } + if req.Status == 2 { // pass + user := userDAO.FindUserByID(userActivity.UserID) + user.Credit += req.Credit + userDAO.SaveUser(user) + } + userActivity.Status = req.Status + checkinDAO.SaveUserActivity(userActivity) + return aw.Success(userActivity) +} +func (c CheckinService) ListActivities(aw *app.Wrapper) app.Result { + return aw.Success(checkinDAO.FindAllActivities()) +} +func (c CheckinService) ListUserActivities(aw *app.Wrapper) app.Result { + return aw.Success(checkinDAO.FindAllUserActivities()) +} +func (c CheckinService) GetUserActivity(aw *app.Wrapper) app.Result { + id, err := strconv.Atoi(aw.Ctx.Query("id")) + if err != nil { + return aw.Error(err.Error()) + } + return aw.Success(checkinDAO.FindUserActivityByID(id)) +} diff --git a/service/service.go b/service/service.go new file mode 100644 index 0000000..572608b --- /dev/null +++ b/service/service.go @@ -0,0 +1,21 @@ +package service + +import "integrity-checkin/dao" + +var ( + userDAO dao.UserDAO + checkinDAO dao.CheckinDAO +) + +var ( + ExUserService UserService + ExCheckinService CheckinService +) + +func Setup() { + userDAO = dao.UserDAO{Tx: dao.DB} + checkinDAO = dao.CheckinDAO{Tx: dao.DB} + + ExUserService = UserService{} + ExCheckinService = CheckinService{} +} diff --git a/service/user.go b/service/user.go new file mode 100644 index 0000000..fe5248d --- /dev/null +++ b/service/user.go @@ -0,0 +1,30 @@ +package service + +import ( + "integrity-checkin/pkg/app" + "integrity-checkin/pkg/httpclient" + "integrity-checkin/pkg/util" +) + +type UserService struct { +} + +func (u UserService) Login(aw *app.Wrapper) app.Result { + type LoginReq struct { + Username string `form:"username" binding:"required"` + Password string `form:"password" binding:"required"` + } + var req LoginReq + if err := aw.Ctx.ShouldBind(&req); err != nil { + return aw.Error(err.Error()) + } + sduLoginData, err := httpclient.SDULogin(req.Username, req.Password) + if err != nil { + return aw.Error(err.Error()) + } + user := userDAO.FindUserByUsername(sduLoginData.Username) + if user == nil { + user = userDAO.CreateUser(sduLoginData.Username, sduLoginData.RealName) + } + return aw.Success(util.GenerateJWT(user.ID, user.RoleID)) +}