-
Notifications
You must be signed in to change notification settings - Fork 0
/
preview_cmd.go
89 lines (78 loc) · 2.29 KB
/
preview_cmd.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
// SPDX-License-Identifier: Apache-2.0
// Copyright © 2021 Wrangle Ltd
package main
import (
"encoding/hex"
"fmt"
"strings"
"github.com/gdamore/tcell/v2"
"github.com/rivo/tview"
"github.com/spf13/cobra"
"github.com/wrgl/core/pkg/diff"
"github.com/wrgl/core/pkg/objects"
"github.com/wrgl/core/pkg/ref"
"github.com/wrgl/core/pkg/widgets"
)
func newPreviewCmd() *cobra.Command {
cmd := &cobra.Command{
Use: "preview COMMIT",
Short: "Shows commit data in a table view",
Example: strings.Join([]string{
` wrgl preview my-branch`,
` wrgl preview 1a2ed6248c7243cdaaecb98ac12213a7`,
` wrgl preview my-file.csv`,
}, "\n"),
Args: cobra.ExactArgs(1),
RunE: func(cmd *cobra.Command, args []string) error {
cleanup, err := setupDebugLog(cmd)
if err != nil {
return err
}
defer cleanup()
cStr := args[0]
rd := getRepoDir(cmd)
quitIfRepoDirNotExist(cmd, rd)
db, err := rd.OpenObjectsStore()
if err != nil {
return err
}
rs := rd.OpenRefStore()
defer db.Close()
_, hash, commit, err := ref.InterpretCommitName(db, rs, cStr, false)
if err != nil {
return err
}
if commit == nil {
return fmt.Errorf("commit \"%s\" not found", cStr)
}
tbl, err := objects.GetTable(db, commit.Table)
if err != nil {
return fmt.Errorf("GetTable: %v", err)
}
return previewTable(cmd, db, hex.EncodeToString(hash), commit, tbl)
},
}
return cmd
}
func previewTable(cmd *cobra.Command, db objects.Store, hash string, commit *objects.Commit, tbl *objects.Table) error {
app := tview.NewApplication().EnableMouse(true)
// create title bar
titleBar := tview.NewTextView().SetDynamicColors(true)
fmt.Fprintf(titleBar, "[yellow]%s[white] ([teal]%d[white] x [teal]%d[white])", hash, tbl.RowsCount, len(tbl.Columns))
// create table
rowReader, err := diff.NewTableReader(db, tbl)
if err != nil {
return err
}
tv := widgets.NewPreviewTable(rowReader, int(tbl.RowsCount), tbl.Columns, tbl.PK)
usageBar := widgets.DataTableUsage()
flex := tview.NewFlex().SetDirection(tview.FlexRow).
AddItem(titleBar, 1, 1, false).
AddItem(tv, 0, 1, true).
AddItem(usageBar, 0, 1, false)
app.SetBeforeDrawFunc(func(screen tcell.Screen) bool {
usageBar.BeforeDraw(screen, flex)
return false
})
return app.SetRoot(flex, true).SetFocus(flex).Run()
}