Lilian Jónsdóttir
9b87c02744
fix typos, less short and more idomatic variable names, package comments, etc.
504 lines
13 KiB
Go
504 lines
13 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"io/fs"
|
|
"os"
|
|
"path/filepath"
|
|
"slices"
|
|
"time"
|
|
|
|
"git.burning.moe/celediel/gt/internal/filemode"
|
|
"git.burning.moe/celediel/gt/internal/files"
|
|
"git.burning.moe/celediel/gt/internal/filter"
|
|
"git.burning.moe/celediel/gt/internal/interactive"
|
|
"git.burning.moe/celediel/gt/internal/interactive/modes"
|
|
|
|
"github.com/adrg/xdg"
|
|
"github.com/charmbracelet/log"
|
|
"github.com/urfave/cli/v2"
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
const (
|
|
appname string = "gt"
|
|
appdesc string = "xdg trash cli"
|
|
appversion string = "v0.0.1"
|
|
executePerm = fs.FileMode(0755)
|
|
)
|
|
|
|
var (
|
|
loglvl string
|
|
fltr *filter.Filter
|
|
onArg, beforeArg, afterArg string
|
|
globArg, patternArg string
|
|
unGlobArg, unPatternArg string
|
|
modeArg, minArg, maxArg string
|
|
filesOnlyArg, dirsOnlyArg bool
|
|
hiddenArg, noInterArg bool
|
|
askconfirm, all bool
|
|
workdir, ogdir cli.Path
|
|
recursive bool
|
|
termwidth int
|
|
termheight int
|
|
|
|
trashDir = filepath.Join(xdg.DataHome, "Trash")
|
|
|
|
beforeAll = func(_ *cli.Context) error {
|
|
// setup log
|
|
log.SetReportTimestamp(true)
|
|
log.SetTimeFormat(time.TimeOnly)
|
|
if level, err := log.ParseLevel(loglvl); err == nil {
|
|
log.SetLevel(level)
|
|
// Some extra info for debug level
|
|
if log.GetLevel() == log.DebugLevel {
|
|
log.SetReportCaller(true)
|
|
}
|
|
} else {
|
|
log.Errorf("unknown log level '%s' (possible values: debug, info, warn, error, fatal, default: warn)", loglvl)
|
|
}
|
|
|
|
// read the term height and width for tables
|
|
width, height, e := term.GetSize(int(os.Stdout.Fd()))
|
|
if e != nil {
|
|
width = 80
|
|
height = 24
|
|
}
|
|
termwidth = width
|
|
termheight = height
|
|
|
|
// ensure trash directories exist
|
|
if _, e := os.Stat(trashDir); os.IsNotExist(e) {
|
|
if err := os.Mkdir(trashDir, executePerm); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if _, e := os.Stat(filepath.Join(trashDir, "info")); os.IsNotExist(e) {
|
|
if err := os.Mkdir(filepath.Join(trashDir, "info"), executePerm); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if _, e := os.Stat(filepath.Join(trashDir, "files")); os.IsNotExist(e) {
|
|
if err := os.Mkdir(filepath.Join(trashDir, "files"), executePerm); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// action launches interactive mode if run without args, or trashes files as args.
|
|
action = func(ctx *cli.Context) error {
|
|
var (
|
|
err error
|
|
)
|
|
|
|
if fltr == nil {
|
|
md, e := filemode.Parse(modeArg)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
fltr, err = filter.New(onArg, beforeArg, afterArg, globArg, patternArg, unGlobArg, unPatternArg, filesOnlyArg, dirsOnlyArg, false, minArg, maxArg, md)
|
|
}
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(ctx.Args().Slice()) == 0 {
|
|
// no ags, so do interactive mode
|
|
var (
|
|
infiles files.Files
|
|
selected files.Files
|
|
mode modes.Mode
|
|
err error
|
|
)
|
|
infiles, err = files.FindTrash(trashDir, ogdir, fltr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(infiles) <= 0 {
|
|
var msg string
|
|
if fltr.Blank() {
|
|
msg = "trash is empty"
|
|
} else {
|
|
msg = "no files to show"
|
|
}
|
|
fmt.Fprint(os.Stdout, msg)
|
|
return nil
|
|
}
|
|
selected, mode, err = interactive.Select(infiles, termwidth, termheight, false, false, false, workdir, modes.Interactive)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
switch mode {
|
|
case modes.Cleaning:
|
|
for _, file := range selected {
|
|
log.Debugf("gonna clean %s", file.Name())
|
|
}
|
|
if err := files.ConfirmClean(askconfirm, selected); err != nil {
|
|
return err
|
|
}
|
|
case modes.Restoring:
|
|
for _, file := range selected {
|
|
log.Debugf("gonna restore %s", file.Name())
|
|
}
|
|
if err := files.ConfirmRestore(askconfirm, selected); err != nil {
|
|
return err
|
|
}
|
|
case modes.Interactive:
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("got bad mode %s", mode)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// args, so try to trash files
|
|
var filesToTrash files.Files
|
|
for _, arg := range ctx.Args().Slice() {
|
|
file, e := files.NewDisk(arg)
|
|
if e != nil {
|
|
log.Fatalf("cannot trash '%s': No such file or directory", arg)
|
|
}
|
|
filesToTrash = append(filesToTrash, file)
|
|
}
|
|
return files.ConfirmTrash(askconfirm, filesToTrash, trashDir)
|
|
}
|
|
|
|
beforeCommands = func(ctx *cli.Context) (err error) {
|
|
// setup filter
|
|
if fltr == nil {
|
|
md, e := filemode.Parse(modeArg)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
fltr, err = filter.New(onArg, beforeArg, afterArg, globArg, patternArg, unGlobArg, unPatternArg, filesOnlyArg, dirsOnlyArg, false, minArg, maxArg, md, ctx.Args().Slice()...)
|
|
}
|
|
log.Debugf("filter: %s", fltr.String())
|
|
return
|
|
}
|
|
|
|
beforeTrash = func(_ *cli.Context) (err error) {
|
|
if fltr == nil {
|
|
md, e := filemode.Parse(modeArg)
|
|
if e != nil {
|
|
return e
|
|
}
|
|
fltr, err = filter.New(onArg, beforeArg, afterArg, globArg, patternArg, unGlobArg, unPatternArg, filesOnlyArg, dirsOnlyArg, !hiddenArg, minArg, maxArg, md)
|
|
}
|
|
log.Debugf("filter: %s", fltr.String())
|
|
return
|
|
}
|
|
|
|
after = func(_ *cli.Context) error {
|
|
return nil
|
|
}
|
|
|
|
doTrash = &cli.Command{
|
|
Name: "trash",
|
|
Aliases: []string{"tr"},
|
|
Usage: "Trash a file or files",
|
|
Flags: slices.Concat(trashFlags, filterFlags),
|
|
Before: beforeTrash,
|
|
Action: func(ctx *cli.Context) error {
|
|
var filesToTrash files.Files
|
|
var selectall bool
|
|
for _, arg := range ctx.Args().Slice() {
|
|
file, e := files.NewDisk(arg)
|
|
if e != nil || workdir != "" {
|
|
log.Debugf("%s wasn't really a file", arg)
|
|
fltr.AddFileName(arg)
|
|
continue
|
|
}
|
|
filesToTrash = append(filesToTrash, file)
|
|
selectall = true
|
|
}
|
|
|
|
// if none of the args were files, then process find files based on filter
|
|
if len(filesToTrash) == 0 {
|
|
fls, err := files.FindDisk(workdir, recursive, fltr)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if len(fls) == 0 {
|
|
fmt.Fprintf(os.Stdout, "no files to trash")
|
|
return nil
|
|
}
|
|
filesToTrash = append(filesToTrash, fls...)
|
|
selectall = !fltr.Blank()
|
|
}
|
|
|
|
selected, _, err := interactive.Select(filesToTrash, termwidth, termheight, false, selectall, false, workdir, modes.Trashing)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(selected) <= 0 {
|
|
return nil
|
|
}
|
|
|
|
return files.ConfirmTrash(askconfirm, selected, trashDir)
|
|
},
|
|
}
|
|
|
|
doList = &cli.Command{
|
|
Name: "list",
|
|
Aliases: []string{"ls"},
|
|
Usage: "List trashed files",
|
|
Flags: slices.Concat(listFlags, alreadyintrashFlags, filterFlags),
|
|
Before: beforeCommands,
|
|
Action: func(_ *cli.Context) error {
|
|
log.Debugf("searching in directory %s for files", trashDir)
|
|
|
|
// look for files
|
|
fls, err := files.FindTrash(trashDir, ogdir, fltr)
|
|
|
|
var msg string
|
|
log.Debugf("filter '%s' is blank? %t in %s", fltr, fltr.Blank(), ogdir)
|
|
if fltr.Blank() && ogdir == "" {
|
|
msg = "trash is empty"
|
|
} else {
|
|
msg = "no files to show"
|
|
}
|
|
|
|
if len(fls) == 0 {
|
|
fmt.Fprint(os.Stdout, msg)
|
|
return nil
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
// display them
|
|
_, _, err = interactive.Select(fls, termwidth, termheight, true, false, noInterArg, workdir, modes.Listing)
|
|
|
|
return err
|
|
},
|
|
}
|
|
|
|
doRestore = &cli.Command{
|
|
Name: "restore",
|
|
Aliases: []string{"re"},
|
|
Usage: "Restore a trashed file or files",
|
|
Flags: slices.Concat(cleanRestoreFlags, alreadyintrashFlags, filterFlags),
|
|
Before: beforeCommands,
|
|
Action: func(_ *cli.Context) error {
|
|
log.Debugf("searching in directory %s for files", trashDir)
|
|
|
|
// look for files
|
|
fls, err := files.FindTrash(trashDir, ogdir, fltr)
|
|
if len(fls) == 0 {
|
|
fmt.Fprintf(os.Stdout, "no files to restore")
|
|
return nil
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
selected, _, err := interactive.Select(fls, termwidth, termheight, false, all, all, workdir, modes.Restoring)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(selected) <= 0 {
|
|
return nil
|
|
}
|
|
|
|
return files.ConfirmRestore(askconfirm || all, selected)
|
|
},
|
|
}
|
|
|
|
doClean = &cli.Command{
|
|
Name: "clean",
|
|
Aliases: []string{"cl"},
|
|
Usage: "Clean files from trash",
|
|
Flags: slices.Concat(cleanRestoreFlags, alreadyintrashFlags, filterFlags),
|
|
Before: beforeCommands,
|
|
Action: func(_ *cli.Context) error {
|
|
fls, err := files.FindTrash(trashDir, ogdir, fltr)
|
|
if len(fls) == 0 {
|
|
fmt.Fprintf(os.Stdout, "no files to clean")
|
|
return nil
|
|
} else if err != nil {
|
|
return err
|
|
}
|
|
|
|
selected, _, err := interactive.Select(fls, termwidth, termheight, false, all, all, workdir, modes.Cleaning)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(selected) <= 0 {
|
|
return nil
|
|
}
|
|
|
|
return files.ConfirmClean(askconfirm, selected)
|
|
},
|
|
}
|
|
|
|
globalFlags = []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "log",
|
|
Usage: "log level",
|
|
Value: "warn",
|
|
Aliases: []string{"l"},
|
|
Destination: &loglvl,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "confirm",
|
|
Usage: "ask for confirmation before executing any action",
|
|
Value: false,
|
|
Aliases: []string{"c"},
|
|
DisableDefaultText: true,
|
|
Destination: &askconfirm,
|
|
},
|
|
}
|
|
|
|
filterFlags = []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: "match",
|
|
Usage: "operate on files matching regex `PATTERN`",
|
|
Aliases: []string{"m"},
|
|
Destination: &patternArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "glob",
|
|
Usage: "operate on files matching `GLOB`",
|
|
Aliases: []string{"g"},
|
|
Destination: &globArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "not-match",
|
|
Usage: "operate on files not matching regex `PATTERN`",
|
|
Aliases: []string{"M"},
|
|
Destination: &unPatternArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "not-glob",
|
|
Usage: "operate on files not matching `GLOB`",
|
|
Aliases: []string{"G"},
|
|
Destination: &unGlobArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "on",
|
|
Usage: "operate on files modified on `DATE`",
|
|
Aliases: []string{"O"},
|
|
Destination: &onArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "after",
|
|
Usage: "operate on files modified before `DATE`",
|
|
Aliases: []string{"A"},
|
|
Destination: &afterArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "before",
|
|
Usage: "operate on files modified after `DATE`",
|
|
Aliases: []string{"B"},
|
|
Destination: &beforeArg,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "files-only",
|
|
Usage: "operate on files only",
|
|
Aliases: []string{"F"},
|
|
DisableDefaultText: true,
|
|
Destination: &filesOnlyArg,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "dirs-only",
|
|
Usage: "operate on directories only",
|
|
Aliases: []string{"D"},
|
|
DisableDefaultText: true,
|
|
Destination: &dirsOnlyArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "min-size",
|
|
Usage: "operate on files larger than `SIZE`",
|
|
Aliases: []string{"N"},
|
|
Destination: &minArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "max-size",
|
|
Usage: "operate on files smaller than `SIZE`",
|
|
Aliases: []string{"X"},
|
|
Destination: &maxArg,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: "mode",
|
|
Usage: "operate on files matching mode `MODE`",
|
|
Aliases: []string{"x"},
|
|
Destination: &modeArg,
|
|
},
|
|
}
|
|
|
|
trashFlags = []cli.Flag{
|
|
&cli.BoolFlag{
|
|
Name: "recursive",
|
|
Usage: "operate on files recursively",
|
|
Aliases: []string{"r"},
|
|
Destination: &recursive,
|
|
Value: false,
|
|
DisableDefaultText: true,
|
|
},
|
|
&cli.PathFlag{
|
|
Name: "work-dir",
|
|
Usage: "operate on files in this `DIRECTORY`",
|
|
Aliases: []string{"w"},
|
|
Destination: &workdir,
|
|
},
|
|
&cli.BoolFlag{
|
|
Name: "hidden",
|
|
Usage: "operate on hidden files",
|
|
Aliases: []string{"H"},
|
|
DisableDefaultText: true,
|
|
Destination: &hiddenArg,
|
|
},
|
|
}
|
|
|
|
alreadyintrashFlags = []cli.Flag{
|
|
&cli.PathFlag{
|
|
Name: "original-path",
|
|
Usage: "operate on files trashed from this `DIRECTORY`",
|
|
Aliases: []string{"o"},
|
|
Destination: &ogdir,
|
|
},
|
|
}
|
|
|
|
listFlags = []cli.Flag{
|
|
&cli.BoolFlag{
|
|
Name: "non-interactive",
|
|
Usage: "list files and quit",
|
|
Aliases: []string{"n"},
|
|
Destination: &noInterArg,
|
|
DisableDefaultText: true,
|
|
},
|
|
}
|
|
|
|
cleanRestoreFlags = []cli.Flag{
|
|
&cli.BoolFlag{
|
|
Name: "all",
|
|
Usage: "operate on all files in trash",
|
|
Aliases: []string{"a"},
|
|
Destination: &all,
|
|
DisableDefaultText: true,
|
|
},
|
|
}
|
|
)
|
|
|
|
func main() {
|
|
app := &cli.App{
|
|
Name: appname,
|
|
Usage: appdesc,
|
|
Version: appversion,
|
|
Before: beforeAll,
|
|
After: after,
|
|
Action: action,
|
|
Commands: []*cli.Command{doTrash, doList, doRestore, doClean},
|
|
Flags: globalFlags,
|
|
EnableBashCompletion: true,
|
|
UseShortOptionHandling: true,
|
|
}
|
|
|
|
if err := app.Run(os.Args); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|