Initial Commit

Initial Commit

Author: st0le

README.md

@@ -0,0 +1,43 @@
+# kv
+
+Dumb Program to store simple Key-Value notes on persistent storage. Integrates with your command line pipeline so you can pull/store values at any time.
+
+## Get the code
+
+`go get -u github.com/st0le/kv`
+
+## Build 
+
+`go build`
+
+## Instructions
+
+You can invoke `kv` in 4 modes. Ideally copy the binary into a location that is in your PATH.
+
+1) list all keys
+
+    `kv`
+
+2) get a single key
+
+    `kv <key>`
+
+3) set a value 
+
+    `kv <key> <value>`
+
+4) pipe contents to a key
+
+    `echo test | kv <key>`
+
+The values are stored in files named by the key in `$HOME\.kv`. Of course, keys will be unique. Files are not encrypted, I wouldn't use it for anything sensitive like your password, apikeys, etc 
+
+## Tip for Windows Powershell Users
+
+Use 
+
+`New-Alias -Name kv -Value "kv.exe"`
+
+To avoid typing ".\kv.exe" each time.
+
+

go.mod

@@ -0,0 +1,3 @@
+module github.com/st0le/kv
+
+go 1.15

main.go

@@ -0,0 +1,94 @@
+package main
+
+import (
+	"fmt"
+	"io"
+	"io/ioutil"
+	"log"
+	"os"
+	"os/user"
+	"path/filepath"
+)
+
+func getStorageDirectory() string {
+	usr, _ := user.Current()
+	path := filepath.Join(usr.HomeDir, ".kv")
+	return path
+}
+
+func ensureDirectory() string {
+	path := getStorageDirectory()
+	return path
+}
+
+func list() {
+	path := getStorageDirectory()
+	files, err := ioutil.ReadDir(path)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	for _, file := range files {
+		fmt.Println(file.Name())
+	}
+}
+
+func set(key, value string) {
+	keyFile := filepath.Join(ensureDirectory(), key)
+
+	f, err := os.Create(keyFile)
+
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	defer f.Close()
+	if value == "" {
+		io.Copy(f, os.Stdin)
+	} else {
+		f.WriteString(value)
+	}
+}
+
+func get(key string) {
+	keyFile := filepath.Join(getStorageDirectory(), key)
+	f, err := os.Open(keyFile)
+
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	defer f.Close()
+	io.Copy(os.Stdout, f)
+}
+
+func main() {
+
+	var args = os.Args
+
+	fi, _ := os.Stdin.Stat()
+
+	if (fi.Mode() & os.ModeCharDevice) == 0 {
+		// if there is a piped stdin, assume its a set command.
+		if len(args) < 2 {
+			log.Fatal("specify a key")
+		} else {
+			set(args[1], "")
+		}
+
+	} else {
+
+		switch len(args) {
+		case 1:
+			// list
+			list()
+		case 2:
+			// get
+			get(args[1])
+		case 3:
+			// set
+			set(args[1], args[2])
+		}
+	}
+
+}