package api import ( "context" "fmt" "github.com/OpenIMSDK/Open-IM-Server/pkg/a2r" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/config" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/constant" "github.com/OpenIMSDK/Open-IM-Server/pkg/common/mcontext" "github.com/OpenIMSDK/Open-IM-Server/pkg/discoveryregistry" "github.com/OpenIMSDK/Open-IM-Server/pkg/errs" "github.com/OpenIMSDK/Open-IM-Server/pkg/proto/third" "github.com/gin-gonic/gin" "net/http" "strconv" ) var _ context.Context // 解决goland编辑器bug func NewThird(c discoveryregistry.SvcDiscoveryRegistry) *Third { return &Third{c: c} } type Third struct { c discoveryregistry.SvcDiscoveryRegistry } func (o *Third) client() (third.ThirdClient, error) { conn, err := o.c.GetConn(config.Config.RpcRegisterName.OpenImThirdName) if err != nil { return nil, err } return third.NewThirdClient(conn), nil } func (o *Third) ApplyPut(c *gin.Context) { a2r.Call(third.ThirdClient.ApplyPut, o.client, c) } func (o *Third) GetPut(c *gin.Context) { a2r.Call(third.ThirdClient.GetPut, o.client, c) } func (o *Third) ConfirmPut(c *gin.Context) { a2r.Call(third.ThirdClient.ConfirmPut, o.client, c) } func (o *Third) GetSignalInvitationInfo(c *gin.Context) { a2r.Call(third.ThirdClient.GetSignalInvitationInfo, o.client, c) } func (o *Third) GetSignalInvitationInfoStartApp(c *gin.Context) { a2r.Call(third.ThirdClient.GetSignalInvitationInfoStartApp, o.client, c) } func (o *Third) FcmUpdateToken(c *gin.Context) { a2r.Call(third.ThirdClient.FcmUpdateToken, o.client, c) } func (o *Third) SetAppBadge(c *gin.Context) { a2r.Call(third.ThirdClient.SetAppBadge, o.client, c) } func (o *Third) GetURL(c *gin.Context) { if c.Request.Method == http.MethodPost { a2r.Call(third.ThirdClient.GetUrl, o.client, c) return } operationID := c.Query("operationID") if operationID == "" { c.String(http.StatusBadRequest, "operationID is empty") return } name := c.Query("name") if name == "" { c.String(http.StatusBadRequest, "name is empty") return } expires, err := strconv.ParseInt(c.Query("expires"), 10, 64) if err != nil { c.String(http.StatusBadRequest, fmt.Sprintf("expires is invalid: %s", err.Error())) return } attachment, _ := strconv.ParseBool(c.Query("attachment")) client, err := o.client() if err != nil { c.String(http.StatusInternalServerError, err.Error()) return } c.Set(constant.OperationID, operationID) resp, err := client.GetUrl(mcontext.SetOperationID(c, operationID), &third.GetUrlReq{Name: name, Expires: expires, Attachment: attachment}) if err != nil { if errs.ErrArgs.Is(err) { c.String(http.StatusBadRequest, err.Error()) return } if errs.ErrRecordNotFound.Is(err) { c.String(http.StatusNotFound, err.Error()) return } c.String(http.StatusInternalServerError, err.Error()) return } c.Redirect(http.StatusTemporaryRedirect, resp.Url) }