package api import ( "context" "egg-im/internal/business/app" comm "egg-im/internal/business/comm" "egg-im/internal/business/comm/utils" "egg-im/internal/business/domain/user/repo" friend2 "egg-im/internal/logic/domain/friend" "egg-im/internal/logic/domain/group/model" repo2 "egg-im/internal/logic/domain/group/repo" "egg-im/internal/logic/domain/message/service" "egg-im/pkg/grpclib" "egg-im/pkg/pb" "encoding/json" "errors" "strconv" "time" ) type BusinessExtServer struct{} func (s *BusinessExtServer) ComplainGroup(ctx context.Context, req *pb.ComplainGroupReq) (*pb.Empty, error) { now := time.Now() userId, _, err := grpclib.GetCtxData(ctx) if err != nil { return nil, err } imageList, _ := json.Marshal(req.ImageList) groupComplain := model.GroupComplain{ GroupId: req.GroupId, UserId: userId, ComplainType: int(req.ComplainType), Text: req.Text, ImageList: string(imageList), Status: 0, CreateTime: now, UpdateTime: now, } return new(pb.Empty), repo2.GroupComplainDao.Save(&groupComplain) } func (s *BusinessExtServer) IsFriends(ctx context.Context, req *pb.IsFriendsReq) (*pb.IsFriendsResp, error) { user, err := repo.UserRepo.GetByPhoneNumber(req.UserPhone) if err != nil { return nil, err } if user == nil { return nil, errors.New("未查询到自身用户信息") } userFriend, err := repo.UserRepo.GetByPhoneNumber(req.FriendPhone) if err != nil { return nil, err } if userFriend == nil { return nil, errors.New("未查询到好友用户信息") } friend, err := friend2.FriendRepo.Get(user.Id, userFriend.Id) if err != nil { return nil, err } var isFriend = int64(1) if friend == nil || friend.Status == friend2.FriendStatusApply { isFriend = 2 } return &pb.IsFriendsResp{ User: &pb.User{ UserId: userFriend.Id, Nickname: userFriend.Nickname, Sex: userFriend.Sex, AvatarUrl: userFriend.AvatarUrl, IsAutoAddedFriends: int64(userFriend.IsAutoAddedFriends), }, IsFriend: isFriend, }, err } func (s *BusinessExtServer) EmoticonList(ctx context.Context, empty *pb.Empty) (*pb.EmoticonListResp, error) { list, err := service.EmoticonList() if err != nil { return nil, err } return &pb.EmoticonListResp{Emoticons: list}, err } func (s *BusinessExtServer) SignIn(ctx context.Context, req *pb.SignInReq) (*pb.SignInResp, error) { utils.FilePutContents("sign_in", utils.SerializeStr(map[string]interface{}{ "args": req, })) isNew, userId, token, err := app.AuthApp.SignIn(ctx, req.PhoneNumber, req.Code, req.DeviceId, req.PushAlia, req.Nickname, req.AvatarUrl) if err != nil { return nil, err } return &pb.SignInResp{ IsNew: isNew, UserId: userId, Token: token, }, nil } func (s *BusinessExtServer) GetUser(ctx context.Context, req *pb.GetUserReq) (*pb.GetUserResp, error) { var user *pb.User if req.Phone != "" { useInfo, err := repo.UserRepo.GetByPhoneNumber(req.Phone) user = &pb.User{ UserId: useInfo.Id, Nickname: useInfo.Nickname, Sex: useInfo.Sex, AvatarUrl: useInfo.AvatarUrl, IsAutoAddedFriends: int64(useInfo.IsAutoAddedFriends), PhoneNumber: useInfo.PhoneNumber, } if err != nil { return nil, err } } else { userId, _, err := grpclib.GetCtxData(ctx) if err != nil { return nil, err } if req.UserId != 0 { userId = req.UserId } user, err = app.UserApp.Get(ctx, userId) } return &pb.GetUserResp{User: user}, nil } func (s *BusinessExtServer) UpdateUser(ctx context.Context, req *pb.UpdateUserReq) (*pb.Empty, error) { userId, _, err := grpclib.GetCtxData(ctx) if err != nil { return nil, err } return new(pb.Empty), app.UserApp.Update(ctx, userId, req) } func (s *BusinessExtServer) SearchUser(ctx context.Context, req *pb.SearchUserReq) (*pb.SearchUserResp, error) { users, err := app.UserApp.Search(ctx, req.Key) return &pb.SearchUserResp{Users: users}, err } func (s *BusinessExtServer) CloudUploadFile(ctx context.Context, req *pb.CloudUploadFileReq) (*pb.CloudUploadFileResp, error) { userId, _, err := grpclib.GetCtxData(ctx) if err != nil { return nil, err } upload, err := comm.CloudUpload.FileReqUpload(ctx, strconv.FormatInt(userId, 10), req.Dir, req.FileName, utils.StrToInt64(req.FileSize)) data := utils.SerializeStr(upload) var result map[string]string utils.Unserialize([]byte(data), &result) return &pb.CloudUploadFileResp{ Method: result["method"], Host: result["host"], Key: result["key"], Token: result["token"], }, err }