Compare commits

...

5 Commits

Author SHA1 Message Date
5d90305b95 a lot 2025-08-25 13:40:51 -07:00
17f34afb2e make stuff sorted and dates n stuff 2025-08-18 20:34:35 -07:00
56ca9d0cb8 prevent duplicates in cache and display 2025-08-18 14:56:01 -07:00
2aa18e77e3 caching (not ready) 2025-08-17 14:53:39 -07:00
33938a31e8 loading and decryption works 2025-08-17 10:57:54 -07:00
3 changed files with 247 additions and 72 deletions

2
go.mod
View File

@@ -3,8 +3,10 @@ module miningtcup.me/notes-cli
go 1.24.6
require (
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e // indirect
github.com/dustin/go-humanize v1.0.1 // indirect
github.com/google/uuid v1.6.0 // indirect
github.com/manifoldco/promptui v0.9.0 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect
github.com/ncruces/go-strftime v0.1.9 // indirect
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect

7
go.sum
View File

@@ -1,7 +1,13 @@
github.com/chzyer/logex v1.1.10/go.mod h1:+Ywpsq7O8HXn0nuIou7OrIPyXbp3wmkHB+jjWRnGsAI=
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e h1:fY5BOSpyZCqRo5OhCuC+XN+r/bBCmeuuJtjz+bCNIf8=
github.com/chzyer/readline v0.0.0-20180603132655-2972be24d48e/go.mod h1:nSuG5e5PlCu98SY8svDHJxuZscDgtXS6KTTbou5AhLI=
github.com/chzyer/test v0.0.0-20180213035817-a1ea475d72b1/go.mod h1:Q3SI9o4m/ZMnBNeIyt5eFwwo7qiLfzFZmjNmxjkiQlU=
github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/manifoldco/promptui v0.9.0 h1:3V4HzJk1TtXW1MTZMP7mdlwbBpIinw3HztaIlYthEiA=
github.com/manifoldco/promptui v0.9.0/go.mod h1:ka04sppxSGFAtxX0qhlYQjISsg9mR4GWtQEhdbn6Pgg=
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
github.com/ncruces/go-strftime v0.1.9 h1:bY0MQC28UADQmHmaF5dgpLmImcShSi2kHU9XLdhx/f4=
@@ -10,6 +16,7 @@ github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec h1:W09IVJc94
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec/go.mod h1:qqbHyh8v60DhA7CoWK5oRCqLrMHRGoxYCSS9EjAz6Eo=
golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b h1:M2rDM6z3Fhozi9O7NWsxAkg/yqS/lQJ6PmkyIV3YP+o=
golang.org/x/exp v0.0.0-20250620022241-b7579e27df2b/go.mod h1:3//PLf8L/X+8b4vuAfHzxeRUl04Adcb341+IGKfnqS8=
golang.org/x/sys v0.0.0-20181122145206-62eef0e2fa9b/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.34.0 h1:H5Y5sJ2L2JRdyv7ROF1he/lPdvFsd0mJHFw2ThKHxLA=
golang.org/x/sys v0.34.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=

310
main.go
View File

@@ -1,27 +1,27 @@
package main
import (
// "database/sql"
"bufio"
"crypto/aes"
"crypto/cipher"
"crypto/pbkdf2"
"crypto/sha256"
"encoding/base64"
"database/sql"
"encoding/json"
"errors"
"fmt"
"io"
"io/fs"
"log"
"net/http"
"net/url"
"os/signal"
"syscall"
"os"
"os/signal"
"slices"
"strconv"
"strings"
"sync"
"syscall"
"time"
"golang.org/x/term"
"github.com/manifoldco/promptui"
_ "modernc.org/sqlite"
)
@@ -31,19 +31,89 @@ var (
password string
dataDir string
dataPath string
cacheDir string
cachePath string
scanner bufio.Scanner
)
func check(err error, format string, a ...any) bool {
if err != nil {
log.Fatalf(format, a, err)
return false
}
return true
type Note struct {
Body string `json:"body"`
Color int64 `json:"color"`
Date string `json:"date"`
ID string `json:"id"`
Title string `json:"title"`
}
func ReadPassword() (string, error) {
func scanInt(promptFormat string, maxNum int64, minNum int64, promptA ...any) (choice int64, err error) {
validateInt := func(input string) error {
parsedInt, err := strconv.ParseInt(input, 10, 64)
if parsedInt > maxNum || parsedInt < minNum {
return fmt.Errorf("Out of bounds (must be between [inclusive] %d and %d)", minNum, maxNum)
}
return err
}
prompt := promptui.Prompt{
Label: fmt.Sprintf(promptFormat, promptA...),
Validate: validateInt,
}
choiceString, err := prompt.Run()
if err != nil {
return
}
choice, err = strconv.ParseInt(choiceString, 10, 64)
if err != nil {
return
}
return
}
func scanString(template string, promptFormat string, promptA ...any) (choice string, err error) {
prompt := promptui.Prompt{
Label: fmt.Sprintf(promptFormat, promptA...),
Default: template,
}
choice, err = prompt.Run()
if err != nil {
return
}
return
}
func scanConfirm(promptFormat string, promptA ...any) (choice string, err error) {
prompt := promptui.Prompt{
Label: fmt.Sprintf(promptFormat, promptA...),
IsConfirm: true,
}
choice, err = prompt.Run()
if err != nil {
return
}
return
}
func scanSelect(selection []string, template string, promptFormat string, promptA ...any) (choice int, err error) {
prompt := promptui.Select{
Label: fmt.Sprintf(promptFormat, promptA...),
Items: selection,
}
choice, _, err = prompt.Run()
if err != nil {
return
}
return
}
func readPassword() (string, error) {
stdin := int(syscall.Stdin)
oldState, err := term.GetState(stdin)
if err != nil {
@@ -79,14 +149,25 @@ func main() {
if len(cachePath) < 1 {
cachePath = home + "/.cache"
}
cacheDir = cachePath + "/notes"
cachePath += "/notes/notes.db"
dataDir = dataPath + "/notes"
dataPath += "/notes/userinfo.json"
// db, err := sql.Open("sqlite", cachePath)
// if err != nil {
// log.Fatal("Error opening sqlite database for notes cache at ", cachePath, ": ", err)
// }
err := os.Mkdir(cacheDir, 0700)
if !errors.Is(err, os.ErrExist) {
check(err, "Error creating cache directory")
}
db, err := sql.Open("sqlite", cachePath)
check(err, "Error opening cache database at '%s'", cachePath)
check(db.Ping(), "Ping of cache database failed!")
db.Exec(`
CREATE TABLE notes (
id TEXT PRIMARY KEY,
data TEXT
);
`)
args := os.Args[1:]
var command string
if len(args) < 1 {
@@ -95,40 +176,142 @@ func main() {
command = args[0]
}
switch command {
case "l", "ls", "list", "search", "find":
case "ls":
userDataBytes, err := os.ReadFile(dataPath)
if errors.Is(err, fs.ErrNotExist) {
login()
}
check(err, "Error reading user data at '%s': %s", dataPath)
check(err, "Error reading user data at '%s'", dataPath)
var userData []string
err = json.Unmarshal(userDataBytes, &userData)
check(err, "Error unmarshalling user data: %s")
check(err, "Error unmarshalling user data")
serverAddress, username, password = userData[0], userData[1], userData[2]
var notes []Note
// fzf := exec.Command("fzf", "--ansi")
// pipe, err := fzf.StdinPipe()
// check(err, "Error getting stdin pipe for fzf")
// check(fzf.Start(), "Error starting fzf")
var wg sync.WaitGroup
response, err := http.Get(fmt.Sprintf("https://%s/get?user=%s", serverAddress, url.QueryEscape(username)))
check(err, "Error getting notes from '%s' as '%s': %s", serverAddress, username)
defer response.Body.Close()
body, err := io.ReadAll(response.Body)
check(err, "Error reading response body while getting notes: %s")
var jsonBody [][2]string
err = json.Unmarshal(body, &jsonBody)
check(err, "Error unmarshalling response body: %s")
decryptedBody := make([]string, len(jsonBody))
for i := range len(jsonBody) {
decryptedBody[i] = decrypt(jsonBody[i][1], password)
jsonDecryptedBody := json.Marshal(decryptedBody)
decryptedBody[i]
cacheJsonBody, err := load(db)
check(err, "Error loading data from cache")
for i := range cacheJsonBody {
go list(cacheJsonBody[i], &notes, &wg)
wg.Add(1)
}
fetchJsonBody, err := fetch()
if err == nil {
for i := range fetchJsonBody {
if !slices.Contains(cacheJsonBody, fetchJsonBody[i]) {
go list(fetchJsonBody[i], &notes, &wg)
wg.Add(1)
}
}
}
fmt.Println(decryptedBody)
wg.Wait()
slices.SortFunc(notes, func(a, b Note) int {
return strings.Compare(b.Date, a.Date)
})
for i := range notes {
timestamp, err := strconv.ParseInt(notes[i].Date, 10, 64)
if err != nil {
fmt.Println("Error parsing Unix timestamp", err)
continue
}
date := time.Unix(timestamp/1000, 0).Format("Mon, Jan 01, 06")
r, g, b, _ := decodeColorInt(notes[i].Color)
check(err, "Error decoding color int %d", notes[i].Color)
formattedNote := strings.TrimSpace(
fmt.Sprintf(
"[%02d] %s | \033[1;38;2;%d;%d;%dm%s\033[0m | %s\n",
i+1,
date,
r,
g,
b,
strings.TrimSpace(notes[i].Title),
strings.ReplaceAll(
notes[i].Body,
"\n",
"/",
),
),
)
terminalWidth, _, err := term.GetSize(int(syscall.Stdin))
check(err, "Error getting terminal size for stdin")
fmt.Printf("%.*s\n", terminalWidth, formattedNote)
}
// pipe.Close()
cache(db, fetchJsonBody)
// fzf.Wait()
fmt.Println()
if fetchJsonBody == nil {
fmt.Println("\033[1;31mNotes couldn't be loaded from the server.\033[0m")
fmt.Println("A connection to the server is required to create, edit, or delete notes.")
fmt.Println("\033[1mTroubleshooting steps:\033[0m")
fmt.Println(` * Check your internet or network connection.
* Make sure the server address is correct.
* Make sure the server is running the latest version of noteserver.
* Check the server's internet or network connection.
`)
return
}
chosenOperationIndex, err := scanSelect([]string{"\033[32mCreate\033[0m", "\033[31mDelete\033[0m", "\033[33mEdit\033[0m"}, "", "Select operation")
check(err, "Error scanning selection")
switch chosenOperationIndex {
case 0:
create(Note{})
fmt.Println("Success!")
case 1:
chosenNoteIndex, err := scanInt(
"Select note to \033[31mdelete\033[37m by index (1-%d)",
int64(len(notes)),
1,
len(notes),
)
check(err, "Error scanning integer")
chosenNoteIndex -= 1
scanConfirm("Really delete '%s'", notes[chosenNoteIndex].Title)
del(notes[chosenNoteIndex].ID, db)
case 2:
chosenNoteIndex, err := scanInt(
"Select note to \033[33medit\033[37m by index (1-%d)",
int64(len(notes)),
1,
len(notes),
)
check(err, "Error scanning integer")
chosenNoteIndex -= 1
create(notes[chosenNoteIndex])
del(notes[chosenNoteIndex].ID, db)
}
default:
login()
}
}
func cache(db *sql.DB, jsonData [][2]string) {
for i := range jsonData {
var id string
err := db.QueryRow("SELECT id FROM notes WHERE id = ?", jsonData[i][0]).Scan(&id)
if err != sql.ErrNoRows {
check(err, "Error scanning cache database at %s for duplicates", cachePath)
continue
}
stmt, err := db.Prepare("INSERT INTO notes VALUES(?, ?)")
check(err, "Error preparing to insert note into cache database at %s", cachePath)
defer stmt.Close()
_, err = stmt.Exec(jsonData[i][0], jsonData[i][1])
check(err, "Error executing insert note into cache database at %s", cachePath)
}
}
func login() {
fmt.Println("Welcome to Notes! To get started, choose your server and user.")
fmt.Print("Server Address [notes.miningtcup.me]: ")
@@ -139,49 +322,32 @@ func login() {
fmt.Print("Username: ")
scanner.Scan()
username = scanner.Text()
username = url.QueryEscape(scanner.Text())
fmt.Print("Password: ")
passwordBytes, err := ReadPassword()
password = string(passwordBytes)
check(err, "Error reading user input (password): %s")
passwordBytes, err := readPassword()
check(err, "Error reading user input (password)")
password = url.QueryEscape(string(passwordBytes))
jsonData, err := json.Marshal([]string{serverAddress, username, password})
check(err, "Error marshalling server address and credentials: %s")
check(err, "Error marshalling server address and credentials")
err = os.Mkdir(dataDir, 0700)
if !errors.Is(err, fs.ErrExist) {
check(err, "Error creating data directory at '%s': %s", dataDir)
check(err, "Error creating data directory at '%s'", dataDir)
}
err = os.WriteFile(dataPath, jsonData, 0600)
check(err, "Error writing to '%s': ", dataPath)
check(err, "Error writing to '%s'", dataPath)
fmt.Println("\nAwesome! To get started, run 'notes add' or 'notes list'")
}
func decrypt(data string, password string) string {
dataBytes, err := base64.StdEncoding.DecodeString(data)
check(err, "Error decoding base64 data: %s")
if len(dataBytes) < 32 {
fmt.Printf("Not enough data (got %d bytes, expected at least 32)\n", len(dataBytes))
return ""
func check(err error, format string, a ...any) bool {
if err != nil {
if writeErr := os.WriteFile("error.log", fmt.Appendf([]byte{}, format+": %s", a, err), 0600); writeErr != nil {
log.Fatalf("\033[31mError writing error '%s' to error.log:\033[0m %s\n", err, writeErr)
}
log.Fatalf("\033[31m"+format+":\033[0m %s\n", a, err)
return false
}
iv := dataBytes[0:16]
salt := dataBytes[16:32]
ciphertext := dataBytes[32:]
if len(ciphertext)%aes.BlockSize != 0 {
fmt.Printf("Ciphertext length %d is not a multiple of block size %d\n", len(ciphertext), aes.BlockSize)
return ""
}
key, err := pbkdf2.Key(sha256.New, password, salt, 65536, 32)
check(err, "Error deriving pbkdf2-sha256 key from password: %s")
block, err := aes.NewCipher(key)
mode := cipher.NewCBCDecrypter(block, iv)
plaintext := make([]byte, len(ciphertext))
mode.CryptBlocks(plaintext, ciphertext)
return string(plaintext)
return true
}