mirror of
https://github.com/ddworken/hishtory.git
synced 2024-11-26 10:13:44 +01:00
97 lines
2.9 KiB
Go
97 lines
2.9 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/ddworken/hishtory/client/data"
|
|
"github.com/ddworken/hishtory/client/hctx"
|
|
"github.com/ddworken/hishtory/client/lib"
|
|
"github.com/ddworken/hishtory/shared"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var GROUP_ID_MANAGEMENT string = "group_id_management"
|
|
|
|
var redactCmd = &cobra.Command{
|
|
Use: "redact",
|
|
Aliases: []string{"delete"},
|
|
Short: "Query for matching commands and remove them from your shell history",
|
|
Long: "This removes history entries on the current machine and on all remote machines. Supports the same query format as 'hishtory query'.",
|
|
GroupID: GROUP_ID_MANAGEMENT,
|
|
DisableFlagParsing: true,
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
ctx := hctx.MakeContext()
|
|
lib.CheckFatalError(lib.RetrieveAdditionalEntriesFromRemote(ctx))
|
|
lib.CheckFatalError(lib.ProcessDeletionRequests(ctx))
|
|
query := strings.Join(args, " ")
|
|
lib.CheckFatalError(redact(ctx, query, os.Getenv("HISHTORY_REDACT_FORCE") != ""))
|
|
},
|
|
}
|
|
|
|
func redact(ctx *context.Context, query string, force bool) error {
|
|
tx, err := lib.MakeWhereQueryFromSearch(ctx, hctx.GetDb(ctx), query)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
var historyEntries []*data.HistoryEntry
|
|
res := tx.Find(&historyEntries)
|
|
if res.Error != nil {
|
|
return res.Error
|
|
}
|
|
if force {
|
|
fmt.Printf("Permanently deleting %d entries\n", len(historyEntries))
|
|
} else {
|
|
fmt.Printf("This will permanently delete %d entries, are you sure? [y/N]", len(historyEntries))
|
|
reader := bufio.NewReader(os.Stdin)
|
|
resp, err := reader.ReadString('\n')
|
|
if err != nil {
|
|
return fmt.Errorf("failed to read response: %v", err)
|
|
}
|
|
if strings.TrimSpace(resp) != "y" {
|
|
fmt.Printf("Aborting delete per user response of %#v\n", strings.TrimSpace(resp))
|
|
return nil
|
|
}
|
|
}
|
|
tx, err = lib.MakeWhereQueryFromSearch(ctx, hctx.GetDb(ctx), query)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
res = tx.Delete(&data.HistoryEntry{})
|
|
if res.Error != nil {
|
|
return res.Error
|
|
}
|
|
if res.RowsAffected != int64(len(historyEntries)) {
|
|
return fmt.Errorf("DB deleted %d rows, when we only expected to delete %d rows, something may have gone wrong", res.RowsAffected, len(historyEntries))
|
|
}
|
|
err = deleteOnRemoteInstances(ctx, historyEntries)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func deleteOnRemoteInstances(ctx *context.Context, historyEntries []*data.HistoryEntry) error {
|
|
config := hctx.GetConf(ctx)
|
|
if config.IsOffline {
|
|
return nil
|
|
}
|
|
|
|
var deletionRequest shared.DeletionRequest
|
|
deletionRequest.SendTime = time.Now()
|
|
deletionRequest.UserId = data.UserId(config.UserSecret)
|
|
|
|
for _, entry := range historyEntries {
|
|
deletionRequest.Messages.Ids = append(deletionRequest.Messages.Ids, shared.MessageIdentifier{Date: entry.EndTime, DeviceId: entry.DeviceId})
|
|
}
|
|
return lib.SendDeletionRequest(deletionRequest)
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(redactCmd)
|
|
}
|