init commit

Author: Sora233

.gitignore

@@ -0,0 +1,6 @@
+/.idea
+
+/buntdb-cli
+/vendor
+/.coverage
+/coverage.html

Makefile

@@ -0,0 +1,21 @@
+flags = "-X main.COMMIT=`git rev-parse --short HEAD` -X 'main.GOVERSION=`go version`'"
+package = buntdb-cli
+cov = .coverage
+SRC = $(wildcard db/*.go ./*.go)
+
+build : $(package)
+test : $(cov)
+cov: $(cov)
+	go tool cover -html=$(cov)
+
+$(package): $(SRC)
+	go build -ldflags $(flags)
+
+.DELETE_ON_ERROR:
+$(cov): $(SRC)
+	go test -cover -coverprofile=$(cov) ./...
+
+
+.PHONY : clean
+clean :
+	-rm -f $(package) $(cov)

README.md

@@ -0,0 +1,3 @@
+# Buntdb-cli
+
+### WIP

completer.go

@@ -0,0 +1,54 @@
+package main
+
+import (
+	"github.com/c-bata/go-prompt"
+	"strings"
+)
+
+func buntdbCompleter(d prompt.Document) []prompt.Suggest {
+	if d.TextBeforeCursor() == "" {
+		return []prompt.Suggest{}
+	}
+	args := strings.Split(d.TextBeforeCursor(), " ")
+	if len(args) == 1 {
+		// input command
+		return cmdCompleter(args[0], true)
+	} else {
+		return optionCompleter(args[0], args[1:])
+	}
+}
+
+func cmdCompleter(cmd string, help bool) []prompt.Suggest {
+	cmds := []prompt.Suggest{
+		{Text: "get", Description: "get command"},
+		{Text: "set", Description: "set command"},
+		{Text: "del", Description: "del command"},
+		{Text: "show", Description: "show info"},
+		{Text: "scan", Description: "iterate keys"},
+		{Text: "use", Description: "change db"},
+	}
+	if help {
+		cmds = append(cmds, prompt.Suggest{Text: "help", Description: "show help for command"})
+	}
+	return prompt.FilterHasPrefix(cmds, cmd, true)
+}
+
+func optionCompleter(cmd string, args []string) []prompt.Suggest {
+	last := args[len(args)-1]
+	switch cmd {
+	case "get":
+	case "set":
+	case "del":
+	case "show":
+		return []prompt.Suggest{
+			{Text: "index"},
+			{Text: "db"},
+		}
+	case "scan":
+	case "help":
+		return cmdCompleter(last, false)
+	default:
+		return []prompt.Suggest{}
+	}
+	return []prompt.Suggest{}
+}

db/buntdb.go

@@ -0,0 +1,44 @@
+package db
+
+import (
+	"errors"
+	"github.com/tidwall/buntdb"
+)
+
+var db *buntdb.DB
+
+var dbpath string
+
+func InitBuntDB(filename string) error {
+	buntDB, err := buntdb.Open(filename)
+	if err != nil {
+		return err
+	}
+	db = buntDB
+	db.SetConfig(buntdb.Config{
+		SyncPolicy: buntdb.Always,
+	})
+	dbpath = filename
+	return nil
+}
+
+func GetClient() (*buntdb.DB, error) {
+	if db == nil {
+		return nil, errors.New("not initialized")
+	}
+	return db, nil
+}
+
+func GetDbPath() string {
+	return dbpath
+}
+
+func Close() error {
+	if db != nil {
+		if err := db.Close(); err != nil {
+			return err
+		}
+		db = nil
+	}
+	return nil
+}

db/buntdb_test.go

@@ -0,0 +1,37 @@
+package db
+
+import (
+	"github.com/stretchr/testify/assert"
+	"os"
+	"testing"
+)
+
+var testDb = ".test"
+
+func TestInitBuntDB(t *testing.T) {
+	assert.Nil(t, InitBuntDB(testDb))
+	assert.Nil(t, Close())
+	os.Remove(testDb)
+}
+
+func TestGetClient(t *testing.T) {
+	db, err := GetClient()
+	assert.NotNil(t, err)
+	assert.Nil(t, db)
+
+	InitBuntDB(testDb)
+	defer os.Remove(testDb)
+	defer Close()
+
+	db, err = GetClient()
+	assert.Nil(t, err)
+	assert.NotNil(t, db)
+
+	assert.Equal(t, GetDbPath(), testDb)
+}
+
+func TestGetDbPath(t *testing.T) {
+	InitBuntDB(testDb)
+	defer os.Remove(testDb)
+	defer Close()
+}

executor.go

@@ -0,0 +1,18 @@
+package main
+
+import (
+	"fmt"
+	"runtime"
+	"strings"
+)
+
+func buntdbExecutor(s string) {
+	s = strings.TrimSpace(s)
+	if s == "" {
+		return
+	} else if s == "quit" || s == "exit" {
+		fmt.Println("bye")
+		runtime.Goexit()
+	}
+	fmt.Println("got", s)
+}

go.mod

@@ -0,0 +1,13 @@
+module github.com/Sora233/buntdb-cli
+
+go 1.15
+
+require (
+	github.com/alecthomas/kong v0.2.12
+	github.com/c-bata/go-prompt v0.2.5
+	github.com/mattn/go-colorable v0.1.8 // indirect
+	github.com/pkg/errors v0.9.1 // indirect
+	github.com/stretchr/testify v1.2.2
+	github.com/tidwall/buntdb v1.1.6
+	golang.org/x/sys v0.0.0-20210108172913-0df2131ae363 // indirect
+)

go.sum

@@ -0,0 +1,57 @@
+github.com/alecthomas/kong v0.2.12 h1:X3kkCOXGUNzLmiu+nQtoxWqj4U2a39MpSJR3QdQXOwI=
+github.com/alecthomas/kong v0.2.12/go.mod h1:kQOmtJgV+Lb4aj+I2LEn40cbtawdWJ9Y8QLq+lElKxE=
+github.com/c-bata/go-prompt v0.2.5 h1:3zg6PecEywxNn0xiqcXHD96fkbxghD+gdB2tbsYfl+Y=
+github.com/c-bata/go-prompt v0.2.5/go.mod h1:vFnjEGDIIA/Lib7giyE4E9c50Lvl8j0S+7FVlAwDAVw=
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/mattn/go-colorable v0.1.4/go.mod h1:U0ppj6V5qS13XJ6of8GYAs25YV2eR4EVcfRqFIhoBtE=
+github.com/mattn/go-colorable v0.1.7 h1:bQGKb3vps/j0E9GfJQ03JyhRuxsvdAanXlT9BTw3mdw=
+github.com/mattn/go-colorable v0.1.7/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=
+github.com/mattn/go-colorable v0.1.8 h1:c1ghPdyEDarC70ftn0y+A/Ee++9zz8ljHG1b13eJ0s8=
+github.com/mattn/go-colorable v0.1.8/go.mod h1:u6P/XSegPjTcexA+o6vUJrdnUu04hMope9wVRipJSqc=
+github.com/mattn/go-isatty v0.0.8/go.mod h1:Iq45c/XA43vh69/j3iqttzPXn0bhXyGjM0Hdxcsrc5s=
+github.com/mattn/go-isatty v0.0.10/go.mod h1:qgIWMr58cqv1PHHyhnkY9lrL7etaEgOFcMEpPG5Rm84=
+github.com/mattn/go-isatty v0.0.12 h1:wuysRhFDzyxgEmMf5xjvJ2M9dZoWAXNNr5LSBS7uHXY=
+github.com/mattn/go-isatty v0.0.12/go.mod h1:cbi8OIDigv2wuxKPP5vlRcQ1OAZbq2CE4Kysco4FUpU=
+github.com/mattn/go-runewidth v0.0.6/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=
+github.com/mattn/go-runewidth v0.0.9 h1:Lm995f3rfxdpd6TSmuVCHVb/QhupuXlYr8sCI/QdE+0=
+github.com/mattn/go-runewidth v0.0.9/go.mod h1:H031xJmbD/WCDINGzjvQ9THkh0rPKHF+m2gUSrubnMI=
+github.com/mattn/go-tty v0.0.3 h1:5OfyWorkyO7xP52Mq7tB36ajHDG5OHrmBGIS/DtakQI=
+github.com/mattn/go-tty v0.0.3/go.mod h1:ihxohKRERHTVzN+aSVRwACLCeqIoZAWpoICkkvrWyR0=
+github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I=
+github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
+github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
+github.com/pkg/term v1.1.0 h1:xIAAdCMh3QIAy+5FrE8Ad8XoDhEU4ufwbaSozViP9kk=
+github.com/pkg/term v1.1.0/go.mod h1:E25nymQcrSllhX42Ok8MRm1+hyBdHY0dCeiKZ9jpNGw=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
+github.com/stretchr/testify v1.2.2 h1:bSDNvY7ZPG5RlJ8otE/7V6gMiyenm9RtJ7IUVIAoJ1w=
+github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
+github.com/tidwall/btree v0.3.0 h1:LcwmLI5kv+AaH/xnBgOuKfbu5eLBWVdWTpD2o+qSRdU=
+github.com/tidwall/btree v0.3.0/go.mod h1:huei1BkDWJ3/sLXmO+bsCNELL+Bp2Kks9OLyQFkzvA8=
+github.com/tidwall/buntdb v1.1.6 h1:GMBmFMOjk4F32KzQ8hA6HKxYpVzMxW1Tc74G7Pgt+U8=
+github.com/tidwall/buntdb v1.1.6/go.mod h1:DqVwIsd7QCInPKRxgr03w165FBuvjLClDy8DrJ+sXJE=
+github.com/tidwall/gjson v1.6.7 h1:Mb1M9HZCRWEcXQ8ieJo7auYyyiSux6w9XN3AdTpxJrE=
+github.com/tidwall/gjson v1.6.7/go.mod h1:zeFuBCIqD4sN/gmqBzZ4j7Jd6UcA2Fc56x7QFsv+8fI=
+github.com/tidwall/grect v0.0.0-20161006141115-ba9a043346eb h1:5NSYaAdrnblKByzd7XByQEJVT8+9v0W/tIY0Oo4OwrE=
+github.com/tidwall/grect v0.0.0-20161006141115-ba9a043346eb/go.mod h1:lKYYLFIr9OIgdgrtgkZ9zgRxRdvPYsExnYBsEAd8W5M=
+github.com/tidwall/match v1.0.3 h1:FQUVvBImDutD8wJLN6c5eMzWtjgONK9MwIBCOrUJKeE=
+github.com/tidwall/match v1.0.3/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM=
+github.com/tidwall/pretty v1.0.2 h1:Z7S3cePv9Jwm1KwS0513MRaoUe3S01WPbLNV40pwWZU=
+github.com/tidwall/pretty v1.0.2/go.mod h1:XNkn88O1ChpSDQmQeStsy+sBenx6DDtFZJxhVysOjyk=
+github.com/tidwall/rtree v0.0.0-20201103190202-0d877048965d h1:RfyBHwIzDt48a28uBzW+W+6rvTJzuYH9OpXBru+CDeo=
+github.com/tidwall/rtree v0.0.0-20201103190202-0d877048965d/go.mod h1:/h+UnNGt0IhNNJLkGikcdcJqm66zGD/uJGMRxK/9+Ao=
+github.com/tidwall/tinyqueue v0.0.0-20180302190814-1e39f5511563 h1:Otn9S136ELckZ3KKDyCkxapfufrqDqwmGjcHfAyXRrE=
+github.com/tidwall/tinyqueue v0.0.0-20180302190814-1e39f5511563/go.mod h1:mLqSmt7Dv/CNneF2wfcChfN1rvapyQr01LGKnKex0DQ=
+golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+golang.org/x/sys v0.0.0-20190222072716-a9d3bda3a223/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
+golang.org/x/sys v0.0.0-20191008105621-543471e840be/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20191120155948-bd437916bb0e/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200116001909-b77594299b42/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200223170610-d5e6a3e2c0ae/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200909081042-eff7692f9009/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20200918174421-af09f7315aff h1:1CPUrky56AcgSpxz/KfgzQWzfG09u5YOL8MvPYBlrL8=
+golang.org/x/sys v0.0.0-20200918174421-af09f7315aff/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=
+golang.org/x/sys v0.0.0-20210108172913-0df2131ae363 h1:wHn06sgWHMO1VsQ8F+KzDJx/JzqfsNLnc+oEi07qD7s=
+golang.org/x/sys v0.0.0-20210108172913-0df2131ae363/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs=

main.go

@@ -0,0 +1,39 @@
+package main
+
+import (
+	"github.com/Sora233/buntdb-cli/db"
+	"github.com/alecthomas/kong"
+	"github.com/c-bata/go-prompt"
+	"io/ioutil"
+	"os"
+	"path"
+)
+
+var CLI struct {
+	Path    string  `arg:"" optional:"" help:"buntudb file path, default a tempfile"`
+	Version version `optional:"" short:"v" help:"print version"`
+}
+
+func main() {
+	kong.Parse(&CLI, kong.UsageOnError(), kong.Name("buntdb-cli"))
+	defer os.Exit(0)
+
+	if CLI.Path == "" {
+		f, err := ioutil.TempFile("", "buntdb-cli")
+		if err != nil {
+		}
+		f.Close()
+		CLI.Path = f.Name()
+		defer os.Remove(f.Name())
+	}
+	db.InitBuntDB(CLI.Path)
+	defer db.Close()
+
+	p := prompt.New(
+		buntdbExecutor,
+		buntdbCompleter,
+		prompt.OptionTitle("buntdb-cli: an interactive buntdb shell client"),
+		prompt.OptionPrefix(path.Base(CLI.Path)+">"),
+	)
+	p.Run()
+}

version.go

@@ -0,0 +1,26 @@
+package main
+
+import (
+	"fmt"
+	"github.com/alecthomas/kong"
+	"io"
+)
+
+// COMMIT is set when build
+var COMMIT = ""
+var GOVERSION = ""
+
+type version bool
+
+// BeforeApply impl a kong hook
+func (v version) BeforeApply(ctx *kong.Context) error {
+	if GOVERSION == "" {
+		GOVERSION = "Unknown"
+	}
+	if COMMIT == "" {
+		COMMIT = "Unknown"
+	}
+	io.WriteString(ctx.Stdout, fmt.Sprintf("COMMIT: %v\nGO VERSION: %v\n", COMMIT, GOVERSION))
+	ctx.Exit(0)
+	return nil
+}