mirror of https://github.com/veypi/OneAuth.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
84 lines
2.0 KiB
Go
84 lines
2.0 KiB
Go
3 years ago
|
package sub
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/olivere/elastic/v7"
|
||
|
|
"github.com/urfave/cli/v2"
|
||
|
|
"github.com/veypi/OneAuth/cfg"
|
||
|
|
"github.com/veypi/OneAuth/models"
|
||
|
|
"github.com/veypi/OneAuth/models/index"
|
||
|
|
"github.com/veypi/utils/log"
|
||
|
|
"reflect"
|
||
|
|
)
|
||
|
|
|
||
|
|
var File = &cli.Command{
|
||
|
|
Name: "file",
|
||
|
|
Subcommands: []*cli.Command{
|
||
|
|
{
|
||
|
|
Name: "list",
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
&cli.UintFlag{
|
||
|
|
Name: "id",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Action: runFileList,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
Name: "history",
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
&cli.StringFlag{
|
||
|
|
Name: "id",
|
||
|
|
},
|
||
|
|
&cli.UintFlag{Name: "from"},
|
||
|
|
&cli.UintFlag{Name: "size"},
|
||
|
|
&cli.UintFlag{Name: "uid", Usage: "user_id"},
|
||
|
|
},
|
||
|
|
Action: runFileHistory,
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
func runFileList(c *cli.Context) error {
|
||
|
|
id := c.Uint("id")
|
||
|
|
log.Info().Msgf("try to get file of %d", id)
|
||
|
|
res, err := index.File.Search().
|
||
|
|
Query(elastic.NewTermQuery("OwnerID", id)).
|
||
|
|
Do(cfg.Ctx)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
fmt.Printf("%-32s %-40s %-6s %6s\n", "file_id", "path", "count", "size")
|
||
|
|
for _, c := range res.Each(reflect.TypeOf(&models.File{})) {
|
||
|
|
f := c.(*models.File)
|
||
|
|
fmt.Printf("%-32s %-40s %-6d %6d\n", f.ID(), f.Path, f.Count, f.Size)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
func runFileHistory(c *cli.Context) error {
|
||
|
|
id := c.String("id")
|
||
|
|
uid := c.Uint("uid")
|
||
|
|
from := int(c.Uint("from"))
|
||
|
|
size := int(c.Uint("size"))
|
||
|
|
if size <= from {
|
||
|
|
size = from + 10
|
||
|
|
}
|
||
|
|
log.Info().Msgf("try to get history[%d, %d] of %s", from, size, id)
|
||
|
|
query := index.History.Search().Sort("CreatedAt", false).From(from).Size(size)
|
||
|
|
if id != "" {
|
||
|
|
query = query.Query(elastic.NewTermQuery("FileID", id))
|
||
|
|
}
|
||
|
|
if uid > 0 {
|
||
|
|
query = query.Query(elastic.NewTermQuery("UserID", uid))
|
||
|
|
}
|
||
|
|
res, err := query.Do(cfg.Ctx)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
fmt.Printf(" %-24s %-10s %-20s %5s %s\n", "time", "action", "ip", "user", "note")
|
||
|
|
for index, c := range res.Each(reflect.TypeOf(&models.History{})) {
|
||
|
|
f := c.(*models.History)
|
||
|
|
fmt.Printf("%-5d %-24s %-10s %-20s %5s %s\n", index, f.CreatedAt.Format(cfg.CFG.TimeFormat), f.Action, f.IP, f.ActorID, f.Tag)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|