chore: add code

Author: qloog

go.mod

@@ -3,41 +3,50 @@ module github.com/1024casts/fastim
 go 1.13
 
 require (
-	github.com/1024casts/snake v1.2.1
+	github.com/1024casts/snake v1.2.2
 	github.com/alicebob/gopher-json v0.0.0-20200520072559-a9ecdc9d1d3a // indirect
 	github.com/fsnotify/fsnotify v1.4.9
 	github.com/gin-contrib/pprof v1.3.0
 	github.com/gin-gonic/gin v1.6.3
 	github.com/go-openapi/spec v0.19.9 // indirect
 	github.com/go-openapi/swag v0.19.9 // indirect
-	github.com/go-playground/validator/v10 v10.3.0 // indirect
+	github.com/go-playground/validator/v10 v10.4.0 // indirect
 	github.com/go-redis/redis v6.15.9+incompatible
+	github.com/golang/snappy v0.0.2 // indirect
 	github.com/google/pprof v0.0.0-20200708004538-1a94d8640e99 // indirect
 	github.com/google/uuid v1.1.2 // indirect
 	github.com/jinzhu/gorm v1.9.16
+	github.com/lestrrat-go/file-rotatelogs v2.4.0+incompatible // indirect
 	github.com/lestrrat-go/strftime v1.0.3 // indirect
 	github.com/lexkong/log v0.0.0-20180607165131-972f9cd951fc
-	github.com/magiconair/properties v1.8.2 // indirect
+	github.com/magiconair/properties v1.8.4 // indirect
 	github.com/mailru/easyjson v0.7.6 // indirect
 	github.com/mitchellh/mapstructure v1.3.3 // indirect
-	github.com/pelletier/go-toml v1.8.0 // indirect
+	github.com/pelletier/go-toml v1.8.1 // indirect
 	github.com/pkg/errors v0.9.1
-	github.com/prometheus/common v0.13.0 // indirect
-	github.com/qiniu/api.v7 v7.2.5+incompatible
-	github.com/spf13/afero v1.3.4 // indirect
+	github.com/prometheus/common v0.14.0 // indirect
+	github.com/prometheus/procfs v0.2.0 // indirect
+	github.com/qiniu/api.v7 v0.0.0-20190520053455-bea02cd22bf4
+	github.com/qiniu/api.v7/v7 v7.4.1
+	github.com/spf13/afero v1.4.0 // indirect
 	github.com/spf13/jwalterweatherman v1.1.0 // indirect
-	github.com/spf13/pflag v1.0.5 // indirect
+	github.com/spf13/pflag v1.0.5
 	github.com/spf13/viper v1.7.1
 	github.com/swaggo/gin-swagger v1.2.0
 	github.com/swaggo/swag v1.6.7 // indirect
 	github.com/teris-io/shortid v0.0.0-20171029131806-771a37caa5cf
+	github.com/ugorji/go v1.1.9 // indirect
 	github.com/willf/pad v0.0.0-20200313202418-172aa767f2a4 // indirect
 	github.com/yuin/gopher-lua v0.0.0-20200816102855-ee81675732da // indirect
-	golang.org/x/crypto v0.0.0-20200820211705-5c72a883971a // indirect
-	golang.org/x/net v0.0.0-20200822124328-c89045814202 // indirect
-	golang.org/x/sys v0.0.0-20200828194041-157a740278f4 // indirect
-	golang.org/x/tools v0.0.0-20200828161849-5deb26317202 // indirect
+	go.uber.org/multierr v1.6.0 // indirect
+	go.uber.org/zap v1.16.0 // indirect
+	golang.org/x/crypto v0.0.0-20200930160638-afb6bcd081ae // indirect
+	golang.org/x/net v0.0.0-20200930145003-4acb6c075d10 // indirect
+	golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f // indirect
+	golang.org/x/tools v0.0.0-20201001104356-43ebab892c4c // indirect
 	google.golang.org/appengine v1.6.6 // indirect
+	google.golang.org/genproto v0.0.0-20200930140634-01fc692af84b // indirect
+	google.golang.org/grpc v1.32.0 // indirect
 	google.golang.org/protobuf v1.25.0 // indirect
-	gopkg.in/ini.v1 v1.60.2 // indirect
+	gopkg.in/ini.v1 v1.61.0 // indirect
 )

handler/v1/user/login.go

@@ -0,0 +1,55 @@
+package user
+
+import (
+	"github.com/1024casts/fastim/handler"
+	"github.com/1024casts/fastim/internal/model"
+	"github.com/1024casts/fastim/internal/service"
+	"github.com/1024casts/fastim/pkg/errno"
+	"github.com/gin-gonic/gin"
+	"github.com/lexkong/log"
+)
+
+// PhoneLogin 手机登录接口
+// @Summary 用户登录接口
+// @Description 仅限手机登录
+// @Tags 用户
+// @Produce  json
+// @Param req body PhoneLoginCredentials true "phone"
+// @Success 200 {string} json "{"code":0,"message":"OK","data":{"token":"eyJhbGciOiJIUzI1NiIsInR5cCI6Ik"}}"
+// @Router /users/login [post]
+func PhoneLogin(c *gin.Context) {
+	log.Info("Phone Login function called.")
+
+	// Binding the data with the u struct.
+	var req PhoneLoginCredentials
+	if err := c.Bind(&req); err != nil {
+		log.Warnf("phone login bind param err: %v", err)
+		handler.SendResponse(c, errno.ErrBind, nil)
+		return
+	}
+
+	log.Infof("req %#v", req)
+	// check param
+	if req.Phone == 0 || req.VerifyCode == 0 {
+		log.Warn("phone login bind param is empty")
+		handler.SendResponse(c, errno.ErrParam, nil)
+		return
+	}
+
+	// 验证校验码
+	if !service.VCodeService.CheckLoginVCode(req.Phone, req.VerifyCode) {
+		handler.SendResponse(c, errno.ErrVerifyCode, nil)
+		return
+	}
+
+	// 登录
+	t, err := service.UserSvc.PhoneLogin(c, req.Phone, req.VerifyCode)
+	if err != nil {
+		handler.SendResponse(c, errno.ErrVerifyCode, nil)
+		return
+	}
+
+	handler.SendResponse(c, nil, model.Token{
+		Token: t,
+	})
+}

handler/v1/user/user.go

@@ -0,0 +1,62 @@
+package user
+
+import (
+	"github.com/1024casts/fastim/internal/model"
+)
+
+// CreateRequest 创建用户请求
+type CreateRequest struct {
+	Username string `json:"username"`
+	Password string `json:"password"`
+}
+
+// CreateResponse 创建用户响应
+type CreateResponse struct {
+	Username string `json:"username"`
+}
+
+// RegisterRequest 注册
+type RegisterRequest struct {
+	Username        string `json:"username" form:"username"`
+	Email           string `json:"email" form:"email"`
+	Password        string `json:"password" form:"password"`
+	ConfirmPassword string `json:"confirm_password" form:"confirm_password"`
+}
+
+// LoginCredentials 默认登录方式-邮箱
+type LoginCredentials struct {
+	Email    string `json:"email" form:"email"`
+	Password string `json:"password" form:"password"`
+}
+
+// PhoneLoginCredentials 手机登录
+type PhoneLoginCredentials struct {
+	Phone      int64 `json:"phone" form:"phone" binding:"required" example:"13010002000"`
+	VerifyCode int   `json:"verify_code" form:"verify_code" binding:"required" example:"120110"`
+}
+
+// UpdateRequest 更新请求
+type UpdateRequest struct {
+	Avatar string `json:"avatar"`
+	Sex    int    `json:"sex"`
+}
+
+// FollowRequest 关注请求
+type FollowRequest struct {
+	UserID uint64 `json:"user_id"`
+}
+
+// ListResponse 通用列表resp
+type ListResponse struct {
+	TotalCount uint64      `json:"total_count"`
+	HasMore    int         `json:"has_more"`
+	PageKey    string      `json:"page_key"`
+	PageValue  int         `json:"page_value"`
+	Items      interface{} `json:"items"`
+}
+
+// SwaggerListResponse 文档
+type SwaggerListResponse struct {
+	TotalCount uint64           `json:"totalCount"`
+	UserList   []model.UserInfo `json:"userList"`
+}

internal/dao/user_dao.go

@@ -1,11 +1,17 @@
 package dao
 
-import "github.com/1024casts/fastim/internal/model"
+import (
+	"context"
+
+	"github.com/1024casts/fastim/internal/model"
+	"github.com/pkg/errors"
+)
 
 type UserDao interface {
 	CreateUser(user model.UserBaseModel) (id uint64, err error)
 	GetUserById(id uint64) (*model.UserBaseModel, error)
 	GetUsersByIds(ids []uint64) ([]*model.UserBaseModel, error)
+	GetUserByPhone(ctx context.Context, phone int64) (*model.UserBaseModel, error)
 }
 
 type userDao struct{}
@@ -38,3 +44,14 @@ func (repo *userDao) GetUsersByIds(ids []uint64) ([]*model.UserBaseModel, error)
 
 	return users, result.Error
 }
+
+// GetUserByPhone 根据手机号获取用户
+func (repo *userDao) GetUserByPhone(ctx context.Context, phone int64) (*model.UserBaseModel, error) {
+	user := model.UserBaseModel{}
+	err := model.DB.Where("phone = ?", phone).First(&user).Error
+	if err != nil {
+		return nil, errors.Wrap(err, "[user_repo] get user err by phone")
+	}
+
+	return &user, nil
+}

internal/idl/user.go

@@ -1,13 +1,13 @@
 package idl
 
 import (
-	"github.com/1024casts/banhui/model"
-	"github.com/1024casts/banhui/util"
+	"github.com/1024casts/fastim/internal/model"
+	"github.com/1024casts/fastim/pkg/utils"
 )
 
 type TransUserInput struct {
-	CurUser       *model.UserModel
-	User          *model.UserModel
+	CurUser       *model.UserBaseModel
+	User          *model.UserBaseModel
 	Point         int
 	ActivityCount int
 	IsFollow      int
@@ -21,14 +21,14 @@ func TransUser(input *TransUserInput) *model.UserInfo {
 	user := input.User
 
 	avatar := ""
-	if user != nil && user.Id > 0 {
+	if user != nil && user.ID > 0 {
 		avatar = user.Avatar
 	}
 
 	return &model.UserInfo{
-		Id:             user.Id,
+		Id:             user.ID,
 		Username:       user.Username,
-		Avatar:         util.GetAvatarUrl(avatar),
+		Avatar:         utils.GetAvatarUrl(avatar),
 		Sex:            user.Sex,
 		PostCount:      user.PostCount,
 		CommentCount:   user.CommentCount,

internal/model/user_base.go

@@ -7,7 +7,7 @@ import (
 
 type UserBaseModel struct {
 	BaseModel
-	Phone          int       `gorm:"column:phone" json:"phone"`
+	Phone          int64     `gorm:"column:phone" json:"phone"`
 	Username       string    `gorm:"column:username" json:"username"`
 	Avatar         string    `gorm:"column:avatar" json:"avatar"`
 	Password       string    `gorm:"column:password" json:"password"`
@@ -56,3 +56,8 @@ type UserList struct {
 	Lock  *sync.Mutex
 	IdMap map[uint64]*UserInfo
 }
+
+// Token represents a JSON web token.
+type Token struct {
+	Token string `json:"token"`
+}

internal/service/sms_service.go

@@ -0,0 +1,67 @@
+package service
+
+import (
+	"github.com/pkg/errors"
+	"github.com/qiniu/api.v7/v7/auth"
+	"github.com/qiniu/api.v7/v7/sms"
+	"github.com/spf13/viper"
+)
+
+// ServiceSms 短信服务
+// 使用七牛云
+// 直接初始化，可以避免在使用时再实例化
+var ServiceSms = NewSmsService()
+
+// ISmsService 短信服务接口定义
+type ISmsService interface {
+	Send(phoneNumber string, verifyCode int) error
+	_sendViaQiNiu(phoneNumber string, verifyCode int) error
+}
+
+// smsService 校验码服务，生成校验码和获得校验码
+type smsService struct{}
+
+// NewSmsService 实例化一个sms
+func NewSmsService() ISmsService {
+	return &smsService{}
+}
+
+// Send 发送短信
+func (srv *smsService) Send(phoneNumber string, verifyCode int) error {
+	// 校验参数的正确性
+	if phoneNumber == "" || verifyCode == 0 {
+		return errors.New("param phone or verify_code error")
+	}
+
+	// 调用第三方发送服务
+	return srv._sendViaQiNiu(phoneNumber, verifyCode)
+}
+
+// _sendViaQiNiu 调用七牛短信服务
+func (srv *smsService) _sendViaQiNiu(phoneNumber string, verifyCode int) error {
+	accessKey := viper.GetString("qiniu.access_key")
+	secretKey := viper.GetString("qiniu.secret_key")
+
+	mac := auth.New(accessKey, secretKey)
+	manager := sms.NewManager(mac)
+
+	args := sms.MessagesRequest{
+		SignatureID: viper.GetString("qiniu.signature_id"),
+		TemplateID:  viper.GetString("qiniu.template_id"),
+		Mobiles:     []string{phoneNumber},
+		Parameters: map[string]interface{}{
+			"code": verifyCode,
+		},
+	}
+
+	ret, err := manager.SendMessage(args)
+	if err != nil {
+		return errors.Wrap(err, "send sms message error")
+	}
+
+	if len(ret.JobID) == 0 {
+		return errors.New("send sms message job id error")
+	}
+
+	return nil
+}