You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
109 lines
2.6 KiB
109 lines
2.6 KiB
package main
|
|
|
|
import (
|
|
"OpenIM/internal/tools"
|
|
"OpenIM/pkg/common/cmd"
|
|
"context"
|
|
"fmt"
|
|
"github.com/spf13/cobra"
|
|
"os"
|
|
)
|
|
|
|
var seqCmd = &cobra.Command{
|
|
Use: "seq",
|
|
Short: "seq operation",
|
|
RunE: func(cmdLines *cobra.Command, args []string) error {
|
|
msgTool, err := tools.InitMsgTool()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
userID, _ := cmdLines.Flags().GetString("userID")
|
|
superGroupID, _ := cmdLines.Flags().GetString("superGroupID")
|
|
fixAll, _ := cmdLines.Flags().GetBool("fixAll")
|
|
ctx := context.Background()
|
|
switch {
|
|
case cmdLines.Parent() == getCmd:
|
|
switch {
|
|
case userID != "":
|
|
msgTool.ShowUserSeqs(ctx, userID)
|
|
case superGroupID != "":
|
|
msgTool.ShowSuperGroupSeqs(ctx, superGroupID)
|
|
}
|
|
case cmdLines.Parent() == fixCmd:
|
|
switch {
|
|
case userID != "":
|
|
_, _, err = msgTool.GetAndFixUserSeqs(ctx, userID)
|
|
case superGroupID != "":
|
|
err = msgTool.FixGroupSeq(ctx, userID)
|
|
case fixAll:
|
|
err = msgTool.FixAllSeq(ctx)
|
|
}
|
|
}
|
|
return err
|
|
},
|
|
}
|
|
|
|
var msgCmd = &cobra.Command{
|
|
Use: "msg",
|
|
Short: "msg operation",
|
|
RunE: func(cmdLines *cobra.Command, args []string) error {
|
|
msgTool, err := tools.InitMsgTool()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
userID, _ := cmdLines.Flags().GetString("userID")
|
|
superGroupID, _ := cmdLines.Flags().GetString("superGroupID")
|
|
clearAll, _ := cmdLines.Flags().GetBool("clearAll")
|
|
ctx := context.Background()
|
|
switch {
|
|
case cmdLines.Parent() == getCmd:
|
|
switch {
|
|
case userID != "":
|
|
msgTool.ShowUserSeqs(ctx, userID)
|
|
case superGroupID != "":
|
|
msgTool.ShowSuperGroupSeqs(ctx, superGroupID)
|
|
}
|
|
case cmdLines.Parent() == clearCmd:
|
|
switch {
|
|
case userID != "":
|
|
msgTool.ClearUsersMsg(ctx, []string{userID})
|
|
case superGroupID != "":
|
|
msgTool.ClearSuperGroupMsg(ctx, []string{superGroupID})
|
|
case clearAll:
|
|
msgTool.AllUserClearMsgAndFixSeq()
|
|
}
|
|
}
|
|
return nil
|
|
},
|
|
}
|
|
|
|
var getCmd = &cobra.Command{
|
|
Use: "get",
|
|
Short: "get operation",
|
|
}
|
|
|
|
var fixCmd = &cobra.Command{
|
|
Use: "fix",
|
|
Short: "fix seq operation",
|
|
}
|
|
|
|
var clearCmd = &cobra.Command{
|
|
Use: "clear",
|
|
Short: "clear operation",
|
|
}
|
|
|
|
func main() {
|
|
cmd.RootCmd.PersistentFlags().StringP("userID", "u", "", "openIM userID")
|
|
cmd.RootCmd.PersistentFlags().StringP("groupID", "u", "", "openIM superGroupID")
|
|
seqCmd.Flags().BoolP("fixAll", "c", false, "openIM fix all seqs")
|
|
msgCmd.Flags().BoolP("clearAll", "c", false, "openIM clear all timeout msgs")
|
|
cmd.RootCmd.AddCommand(getCmd, fixCmd, clearCmd)
|
|
getCmd.AddCommand(seqCmd, msgCmd)
|
|
fixCmd.AddCommand(seqCmd)
|
|
clearCmd.AddCommand(msgCmd)
|
|
if err := cmd.RootCmd.Execute(); err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
}
|