mirror of
https://github.com/sbrow/envr.git
synced 2026-06-27 18:48:33 -04:00
Compare commits
2 Commits
f0b12582ba
...
9ebe789a67
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9ebe789a67 | ||
| f1b3129f7d |
@@ -1,16 +1,11 @@
|
|||||||
# Changelog
|
# Changelog
|
||||||
|
|
||||||
## [0.4.0](https://github.com/sbrow/envr/compare/v0.3.0...v0.4.0) (2026-06-18)
|
## [0.4.0](https://github.com/sbrow/envr/compare/v0.3.0...v0.4.0) (2026-06-17)
|
||||||
|
|
||||||
|
|
||||||
### Features
|
### Features
|
||||||
|
|
||||||
* Removed runtime git dependency. ([12574e1](https://github.com/sbrow/envr/commit/12574e123bdedba3aca813143e906ec5e0b95719))
|
* Removed runtime git dependency. ([f1b3129](https://github.com/sbrow/envr/commit/f1b3129f7dc26d177cd9b5bb7b0807d9871a91c5))
|
||||||
|
|
||||||
|
|
||||||
### Bug Fixes
|
|
||||||
|
|
||||||
* Fixed memory leaks in the db. ([5059572](https://github.com/sbrow/envr/commit/5059572951b3ec20b3d2027032a9c3be5cb14dba))
|
|
||||||
|
|
||||||
|
|
||||||
### Performance Improvements
|
### Performance Improvements
|
||||||
|
|||||||
27
README.md
27
README.md
@@ -12,13 +12,14 @@ the tool [of your choosing](#backup-options).
|
|||||||
|
|
||||||
## Features
|
## Features
|
||||||
|
|
||||||
- **Encrypted Storage**: All `.env` files are encrypted using your ssh key and
|
- 🔐 **Encrypted Storage**: All `.env` files are encrypted using your ssh key and
|
||||||
[libsodium](https://github.com/jedisct1/libsodium).
|
[libsodium](https://github.com/jedisct1/libsodium) encryption.
|
||||||
- **Automatic Sync**: Update the database with one command, which can easily
|
- 🔄 **Automatic Sync**: Update the database with one command, which can easily
|
||||||
be run on a cron.
|
be run on a cron.
|
||||||
- **Smart Scanning**: Automatically discover and import `.env` files in your
|
- 🔍 **Smart Scanning**: Automatically discover and import `.env` files in your
|
||||||
home directory.
|
home directory.
|
||||||
- **Rename Detection**: Automatically find and updates renamed/moved
|
- ✨ **Interactive CLI**: User-friendly prompts for file selection and management.
|
||||||
|
- 🗂️ **Rename Detection**: Automatically finds and updates renamed/moved
|
||||||
repositories.
|
repositories.
|
||||||
|
|
||||||
## TODOS
|
## TODOS
|
||||||
@@ -27,10 +28,13 @@ repositories.
|
|||||||
- [x] Allow configuration of ssh key.
|
- [x] Allow configuration of ssh key.
|
||||||
- [x] Allow multiple ssh keys.
|
- [x] Allow multiple ssh keys.
|
||||||
|
|
||||||
## Installation
|
## Prerequisites
|
||||||
|
|
||||||
You will need an SSH key pair for encryption and decryption. You can generate one
|
- An SSH key pair (for encryption/decryption)
|
||||||
with `ssh-keygen -t ed25519`. It will be saved to `~/.ssh/id_ed25519`.
|
- The following binaries:
|
||||||
|
- [fd](https://github.com/sharkdp/fd)
|
||||||
|
|
||||||
|
## Installation
|
||||||
|
|
||||||
### With Odin
|
### With Odin
|
||||||
|
|
||||||
@@ -91,12 +95,7 @@ The configuration file is created during initialization:
|
|||||||
],
|
],
|
||||||
"scan": {
|
"scan": {
|
||||||
"matcher": "\\.env",
|
"matcher": "\\.env",
|
||||||
"exclude": [
|
"exclude": "*.envrc",
|
||||||
"*\\.envrc",
|
|
||||||
"\\.local/",
|
|
||||||
"node_modules",
|
|
||||||
"vendor"
|
|
||||||
],
|
|
||||||
"include": "~"
|
"include": "~"
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
19
TODOS.md
19
TODOS.md
@@ -2,13 +2,11 @@
|
|||||||
|
|
||||||
1. Consider giving db its own allocator
|
1. Consider giving db its own allocator
|
||||||
|
|
||||||
27. Commands are still leaking.
|
2. **db.odin:324-327** — Map iteration (`remote_set`) is non-deterministic. Same file can produce different JSON on each backup, causing spurious DB diffs. Sort remotes before storing.
|
||||||
|
|
||||||
2. Generate md and man pages again.
|
3. **db.odin:135, 250** — String interpolation into SQL (`VACUUM INTO '%s'`, `ATTACH DATABASE '%s'`). Currently safe because input is controlled, but fragile.
|
||||||
|
|
||||||
3. **db.odin:324-327** — Map iteration (`remote_set`) is non-deterministic. Same file can produce different JSON on each backup, causing spurious DB diffs. Sort remotes before storing.
|
4. **features.odin:30-41** — `find_binary` uses `strings.join` instead of `filepath.join`, uses `os.stat` instead of checking executability, hardcodes `:` as PATH separator (wrong on Windows).
|
||||||
|
|
||||||
4. Make sure official path separators are used when appropriate, rather than '/'.
|
|
||||||
|
|
||||||
5. **cmd_restore.odin:20-30 & cmd_remove.odin:19-29** — Identical path-resolution block copy-pasted. `is_abs` guard is redundant since `filepath.abs` is a no-op on absolute paths. Extract a helper.
|
5. **cmd_restore.odin:20-30 & cmd_remove.odin:19-29** — Identical path-resolution block copy-pasted. `is_abs` guard is redundant since `filepath.abs` is a no-op on absolute paths. Extract a helper.
|
||||||
|
|
||||||
@@ -20,7 +18,7 @@
|
|||||||
|
|
||||||
11. **db.odin:352-353** — `hex.encode` error ignored. `string(hex_bytes)` aliases the byte slice.
|
11. **db.odin:352-353** — `hex.encode` error ignored. `string(hex_bytes)` aliases the byte slice.
|
||||||
|
|
||||||
12. **cmd_sync.odin:80, cmd_list.odin:33** — `make([]string, 2)` for table rows never freed. Leaks per row. Defer to memory pass.
|
12. **cmd_sync.odin:80, cmd_list.odin:33, cmd_deps.odin:9** — `make([]string, 2)` for table rows never freed. Leaks per row. Defer to memory pass.
|
||||||
|
|
||||||
13. **cmd_list.odin** — Non-TTY branch builds `ListEntry` structs and marshals JSON separately. Now that `render_json_rows` (issue 1) accepts an `io.Writer` and uses `json.marshal`, unify both branches to use it. Note: will change JSON keys from `"directory"/"path"` to `"Directory"/"Path"`.
|
13. **cmd_list.odin** — Non-TTY branch builds `ListEntry` structs and marshals JSON separately. Now that `render_json_rows` (issue 1) accepts an `io.Writer` and uses `json.marshal`, unify both branches to use it. Note: will change JSON keys from `"directory"/"path"` to `"Directory"/"Path"`.
|
||||||
|
|
||||||
@@ -40,11 +38,7 @@
|
|||||||
|
|
||||||
23. procedures should be ordered by use, main at the top, then in the order they are called from main.
|
23. procedures should be ordered by use, main at the top, then in the order they are called from main.
|
||||||
|
|
||||||
24. Shell completion
|
24. [x] Remove git dependency.
|
||||||
|
|
||||||
25. Bring back windows support / cross-compilation.
|
|
||||||
|
|
||||||
26. Test all cmds / terminal branches.
|
|
||||||
|
|
||||||
## Double-check AI output
|
## Double-check AI output
|
||||||
|
|
||||||
@@ -53,6 +47,7 @@
|
|||||||
- [x] cmd_backup.odin
|
- [x] cmd_backup.odin
|
||||||
- [x] cmd_check.odin
|
- [x] cmd_check.odin
|
||||||
- [ ] cmd_check_test.odin
|
- [ ] cmd_check_test.odin
|
||||||
|
- [x] cmd_deps.odin
|
||||||
- [ ] cmd_edit_config.odin
|
- [ ] cmd_edit_config.odin
|
||||||
- [x] cmd_init.odin
|
- [x] cmd_init.odin
|
||||||
- [x] cmd_list.odin
|
- [x] cmd_list.odin
|
||||||
@@ -71,6 +66,8 @@
|
|||||||
- [ ] db.odin
|
- [ ] db.odin
|
||||||
- [ ] db_integration_test.odin
|
- [ ] db_integration_test.odin
|
||||||
- [ ] db_test.odin
|
- [ ] db_test.odin
|
||||||
|
- [x] features.odin
|
||||||
|
- [x] features_test.odin
|
||||||
- [x] main.odin
|
- [x] main.odin
|
||||||
- [x] prompt.odin
|
- [x] prompt.odin
|
||||||
- [ ] scan.odin
|
- [ ] scan.odin
|
||||||
|
|||||||
87
WINDOWS.md
Normal file
87
WINDOWS.md
Normal file
@@ -0,0 +1,87 @@
|
|||||||
|
# Windows Compatibility Guide
|
||||||
|
|
||||||
|
This document outlines Windows compatibility issues and solutions for the envr project.
|
||||||
|
|
||||||
|
## Critical Issues
|
||||||
|
|
||||||
|
### 1. Path Handling Bug (MUST FIX)
|
||||||
|
|
||||||
|
**File:** `app/env_file.go:209`
|
||||||
|
|
||||||
|
**Issue:** Uses `path.Join` instead of `filepath.Join`, which won't work correctly on Windows due to different path separators.
|
||||||
|
|
||||||
|
**Current code:**
|
||||||
|
```go
|
||||||
|
f.Path = path.Join(newDir, path.Base(f.Path))
|
||||||
|
```
|
||||||
|
|
||||||
|
**Fixed code:**
|
||||||
|
```go
|
||||||
|
f.Path = filepath.Join(newDir, filepath.Base(f.Path))
|
||||||
|
```
|
||||||
|
|
||||||
|
## External Dependencies
|
||||||
|
|
||||||
|
The application relies on external tools that need to be installed separately on Windows:
|
||||||
|
|
||||||
|
### Required Tools
|
||||||
|
|
||||||
|
1. **fd** - Fast file finder
|
||||||
|
- Install via: `winget install sharkdp.fd` or `choco install fd`
|
||||||
|
- Alternative: `scoop install fd`
|
||||||
|
|
||||||
|
## Minor Compatibility Notes
|
||||||
|
|
||||||
|
### File Permissions
|
||||||
|
- Unix file permissions (`0755`, `0644`) are used throughout the codebase
|
||||||
|
- These are safely ignored on Windows - no changes needed
|
||||||
|
|
||||||
|
### Editor Configuration
|
||||||
|
**File:** `cmd/edit_config.go:20-24`
|
||||||
|
|
||||||
|
**Issue:** Relies on `$EDITOR` environment variable which is less common on Windows.
|
||||||
|
|
||||||
|
**Current behavior:** Fails if `$EDITOR` is not set
|
||||||
|
|
||||||
|
**Recommended improvement:** Add fallback detection for Windows editors:
|
||||||
|
```go
|
||||||
|
editor := os.Getenv("EDITOR")
|
||||||
|
if editor == "" {
|
||||||
|
if runtime.GOOS == "windows" {
|
||||||
|
editor = "notepad.exe" // or "code.exe" for VS Code
|
||||||
|
} else {
|
||||||
|
fmt.Println("Error: $EDITOR environment variable is not set")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
## Installation Instructions for Windows
|
||||||
|
|
||||||
|
1. Install required dependencies:
|
||||||
|
```powershell
|
||||||
|
winget install sharkdp.fd
|
||||||
|
```
|
||||||
|
|
||||||
|
2. Fix the path handling bug in `app/env_file.go:209`
|
||||||
|
|
||||||
|
3. Build and run as normal:
|
||||||
|
```powershell
|
||||||
|
go build
|
||||||
|
.\envr.exe init
|
||||||
|
```
|
||||||
|
|
||||||
|
## Testing on Windows
|
||||||
|
|
||||||
|
After applying the critical path fix, the core functionality should work correctly on Windows. The application has been designed with cross-platform compatibility in mind, using:
|
||||||
|
|
||||||
|
- `filepath` package for path operations (mostly)
|
||||||
|
- `os.UserHomeDir()` for home directory detection
|
||||||
|
- Standard Go file operations
|
||||||
|
|
||||||
|
## Summary
|
||||||
|
|
||||||
|
- **1 critical bug** must be fixed for Windows compatibility
|
||||||
|
- **2 external tools** need to be installed
|
||||||
|
- **1 minor enhancement** recommended for better Windows UX
|
||||||
|
- Overall architecture is Windows-compatible
|
||||||
69
cli.odin
69
cli.odin
@@ -54,12 +54,20 @@ key somewhere, otherwise your data could be lost forever.`,
|
|||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
delete_command :: proc(cmd: ^Command) {
|
||||||
|
delete(cmd.args)
|
||||||
|
delete(cmd.flags)
|
||||||
|
delete(cmd.bool_set)
|
||||||
|
bufio.writer_destroy(cmd.out_buf)
|
||||||
|
free(cmd.out_buf)
|
||||||
|
}
|
||||||
|
|
||||||
// Caller is responsible for calling delete_command(cmd).
|
// Caller is responsible for calling delete_command(cmd).
|
||||||
// FIXME: Works in kinda a wonky and awkward way.
|
// FIXME: Works in kinda a wonky and awkward way.
|
||||||
parse_args :: proc(args: []string, out: io.Stream, err: io.Stream) -> (cmd: Command, ok: bool) {
|
parse_args :: proc(args: []string, out: io.Stream, err: io.Stream) -> (cmd: Command, ok: bool) {
|
||||||
{
|
{
|
||||||
cmd.out_buf = new(bufio.Writer)
|
cmd.out_buf = new(bufio.Writer)
|
||||||
bufio.writer_init(cmd.out_buf, out, allocator = context.allocator)
|
bufio.writer_init(cmd.out_buf, out)
|
||||||
cmd.out = bufio.writer_to_writer(cmd.out_buf)
|
cmd.out = bufio.writer_to_writer(cmd.out_buf)
|
||||||
cmd.err = err
|
cmd.err = err
|
||||||
}
|
}
|
||||||
@@ -122,12 +130,27 @@ parse_args :: proc(args: []string, out: io.Stream, err: io.Stream) -> (cmd: Comm
|
|||||||
return cmd, true
|
return cmd, true
|
||||||
}
|
}
|
||||||
|
|
||||||
print_command_help :: proc(cmd: ^Command) {
|
has_flag :: proc(cmd: ^Command, name: string) -> bool {
|
||||||
ok := write_command_help(cmd.name, cmd.out)
|
_, ok := cmd.flags[name]
|
||||||
if !ok {
|
if ok {
|
||||||
fmt.wprintf(cmd.err, "Unknown command: %s\n", cmd.name)
|
return true
|
||||||
write_usage(cmd.out)
|
|
||||||
}
|
}
|
||||||
|
_, ok2 := cmd.bool_set[name]
|
||||||
|
return ok2
|
||||||
|
}
|
||||||
|
|
||||||
|
find_command :: proc(name: string) -> (CommandInfo, bool) {
|
||||||
|
for c in COMMANDS {
|
||||||
|
if c.name == name {
|
||||||
|
return c, true
|
||||||
|
}
|
||||||
|
for a in c.aliases {
|
||||||
|
if a == name {
|
||||||
|
return c, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return CommandInfo{}, false
|
||||||
}
|
}
|
||||||
|
|
||||||
write_command_help :: proc(name: string, w: io.Writer) -> bool {
|
write_command_help :: proc(name: string, w: io.Writer) -> bool {
|
||||||
@@ -160,18 +183,12 @@ write_command_help :: proc(name: string, w: io.Writer) -> bool {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
find_command :: proc(name: string) -> (CommandInfo, bool) {
|
print_command_help :: proc(cmd: ^Command) {
|
||||||
for c in COMMANDS {
|
ok := write_command_help(cmd.name, cmd.out)
|
||||||
if c.name == name {
|
if !ok {
|
||||||
return c, true
|
fmt.wprintf(cmd.err, "Unknown command: %s\n", cmd.name)
|
||||||
}
|
write_usage(cmd.out)
|
||||||
for a in c.aliases {
|
|
||||||
if a == name {
|
|
||||||
return c, true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return CommandInfo{}, false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: command args should be shown in usage.
|
// TODO: command args should be shown in usage.
|
||||||
@@ -246,21 +263,3 @@ Use "envr [command] --help" for more information about a command.
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
has_flag :: proc(cmd: ^Command, name: string) -> bool {
|
|
||||||
_, ok := cmd.flags[name]
|
|
||||||
if ok {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
_, ok2 := cmd.bool_set[name]
|
|
||||||
return ok2
|
|
||||||
}
|
|
||||||
|
|
||||||
delete_command :: proc(cmd: ^Command) {
|
|
||||||
bufio.writer_flush(cmd.out_buf)
|
|
||||||
delete(cmd.args)
|
|
||||||
delete(cmd.flags)
|
|
||||||
delete(cmd.bool_set)
|
|
||||||
bufio.writer_destroy(cmd.out_buf)
|
|
||||||
free(cmd.out_buf)
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|||||||
@@ -54,8 +54,6 @@ cmd_check :: proc(cmd: ^Command) {
|
|||||||
if !list_ok {
|
if !list_ok {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer delete(db_files)
|
|
||||||
defer for &file in db_files {delete_envfile(&file)}
|
|
||||||
|
|
||||||
not_backed := find_unbacked(files_in_path[:], db_files[:])
|
not_backed := find_unbacked(files_in_path[:], db_files[:])
|
||||||
|
|
||||||
@@ -63,23 +61,13 @@ cmd_check :: proc(cmd: ^Command) {
|
|||||||
if len(files_in_path) == 0 {
|
if len(files_in_path) == 0 {
|
||||||
fmt.wprintln(cmd.out, "No .env files found in the specified directory.", flush = false)
|
fmt.wprintln(cmd.out, "No .env files found in the specified directory.", flush = false)
|
||||||
} else {
|
} else {
|
||||||
fmt.wprintln(
|
fmt.wprintln(cmd.out, "✓ All .env files in the directory are backed up.", flush = false)
|
||||||
cmd.out,
|
|
||||||
"✓ All .env files in the directory are backed up.",
|
|
||||||
flush = false,
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
fmt.wprintf(
|
fmt.wprintf(cmd.out, "Found %d .env file(s) that are not backed up:\n", len(not_backed), flush = false)
|
||||||
cmd.out,
|
|
||||||
"Found %d .env file(s) that are not backed up:\n",
|
|
||||||
len(not_backed),
|
|
||||||
flush = false,
|
|
||||||
)
|
|
||||||
for file in not_backed {
|
for file in not_backed {
|
||||||
fmt.wprintf(cmd.out, " %s\n", file, flush = false)
|
fmt.wprintf(cmd.out, " %s\n", file, flush = false)
|
||||||
}
|
}
|
||||||
fmt.wprintln(cmd.out, "\nRun 'envr sync' to back up these files.", flush = false)
|
fmt.wprintln(cmd.out, "\nRun 'envr sync' to back up these files.", flush = false)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -26,7 +26,6 @@ cmd_list :: proc(cmd: ^Command) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
defer delete(rows)
|
defer delete(rows)
|
||||||
defer for &row in rows {delete_envfile(&row)}
|
|
||||||
|
|
||||||
if terminal.is_terminal(os.stdout) {
|
if terminal.is_terminal(os.stdout) {
|
||||||
headers := []string{"Directory", "Path"}
|
headers := []string{"Directory", "Path"}
|
||||||
@@ -35,7 +34,7 @@ cmd_list :: proc(cmd: ^Command) {
|
|||||||
for row in rows {
|
for row in rows {
|
||||||
dir_str := strings.concatenate({row.Dir, "/"}, context.temp_allocator)
|
dir_str := strings.concatenate({row.Dir, "/"}, context.temp_allocator)
|
||||||
filename := filepath.base(row.Path)
|
filename := filepath.base(row.Path)
|
||||||
row_slice := make([]string, 2, context.temp_allocator)
|
row_slice := make([]string, 2)
|
||||||
row_slice[0] = dir_str
|
row_slice[0] = dir_str
|
||||||
row_slice[1] = filename
|
row_slice[1] = filename
|
||||||
append(&table_rows, row_slice)
|
append(&table_rows, row_slice)
|
||||||
|
|||||||
171
config.odin
171
config.odin
@@ -25,6 +25,14 @@ Config :: struct {
|
|||||||
config_path: string `json:"-"`,
|
config_path: string `json:"-"`,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
default_config_path :: proc(home: string, allocator := context.allocator) -> string {
|
||||||
|
path, err := filepath.join([]string{home, ".envr", "config.json"}, allocator)
|
||||||
|
if err != nil {
|
||||||
|
panic("Ran out of memory when building config path")
|
||||||
|
}
|
||||||
|
return path
|
||||||
|
}
|
||||||
|
|
||||||
load_config :: proc(config_path: string) -> (Config, bool) {
|
load_config :: proc(config_path: string) -> (Config, bool) {
|
||||||
data, read_err := os.read_entire_file_from_path(config_path, context.allocator)
|
data, read_err := os.read_entire_file_from_path(config_path, context.allocator)
|
||||||
if read_err != nil {
|
if read_err != nil {
|
||||||
@@ -45,14 +53,6 @@ load_config :: proc(config_path: string) -> (Config, bool) {
|
|||||||
return cfg, true
|
return cfg, true
|
||||||
}
|
}
|
||||||
|
|
||||||
default_config_path :: proc(home: string, allocator := context.allocator) -> string {
|
|
||||||
path, err := filepath.join([]string{home, ".envr", "config.json"}, allocator)
|
|
||||||
if err != nil {
|
|
||||||
panic("Ran out of memory when building config path")
|
|
||||||
}
|
|
||||||
return path
|
|
||||||
}
|
|
||||||
|
|
||||||
delete_config :: proc(cfg: ^Config) {
|
delete_config :: proc(cfg: ^Config) {
|
||||||
for key in cfg.Keys {
|
for key in cfg.Keys {
|
||||||
delete(key.Private)
|
delete(key.Private)
|
||||||
@@ -73,73 +73,13 @@ delete_config :: proc(cfg: ^Config) {
|
|||||||
delete(cfg.ScanConfig.Include)
|
delete(cfg.ScanConfig.Include)
|
||||||
}
|
}
|
||||||
|
|
||||||
save_config :: proc(cfg: Config, force: bool = false) -> bool {
|
envr_dir :: proc(config_path: string) -> string {
|
||||||
config_dir := envr_dir(cfg.config_path)
|
return filepath.dir(config_path)
|
||||||
|
|
||||||
if !os.exists(config_dir) {
|
|
||||||
mkdir_err := os.make_directory(config_dir)
|
|
||||||
if mkdir_err != nil {
|
|
||||||
fmt.printf("Error creating %s directory: %v\n", config_dir, mkdir_err)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if os.exists(cfg.config_path) && !force {
|
|
||||||
info, stat_err := os.stat(cfg.config_path, context.allocator)
|
|
||||||
if stat_err == nil {
|
|
||||||
defer os.file_info_delete(info, context.allocator)
|
|
||||||
if info.size > 0 {
|
|
||||||
fmt.println("Config file already exists. Run again with --force to reinitialize.")
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
data, marshal_err := json.marshal(cfg, {pretty = true, use_spaces = true, spaces = 2})
|
|
||||||
if marshal_err != nil {
|
|
||||||
fmt.printf("Error marshaling config: %v\n", marshal_err)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
defer delete(data)
|
|
||||||
|
|
||||||
write_err := os.write_entire_file(cfg.config_path, data)
|
|
||||||
if write_err != nil {
|
|
||||||
fmt.printf("Error writing config: %v\n", write_err)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
return true
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Caller is responsible for calling delete_config()
|
data_path :: proc(config_path: string) -> string {
|
||||||
new_config :: proc(
|
path, _ := filepath.join([]string{envr_dir(config_path), "data.envr"})
|
||||||
private_key_paths: []string,
|
return path
|
||||||
cfg_path: string = "~/.envr/config.json",
|
|
||||||
) -> Config {
|
|
||||||
keys := make([dynamic]SshKeyPair, 0, len(private_key_paths))
|
|
||||||
for priv in private_key_paths {
|
|
||||||
// TODO: Is this bad?
|
|
||||||
priv_key := strings.clone(priv)
|
|
||||||
pub, _ := strings.concatenate([]string{priv_key, ".pub"})
|
|
||||||
append(&keys, SshKeyPair{Private = priv_key, Public = pub})
|
|
||||||
}
|
|
||||||
|
|
||||||
exclude := make([dynamic]string, 0, 4)
|
|
||||||
append(&exclude, strings.clone("*\\.envrc"))
|
|
||||||
append(&exclude, strings.clone("\\.local/"))
|
|
||||||
append(&exclude, strings.clone("node_modules"))
|
|
||||||
append(&exclude, strings.clone("vendor"))
|
|
||||||
|
|
||||||
include := make([dynamic]string, 0, 1)
|
|
||||||
append(&include, strings.clone("~"))
|
|
||||||
|
|
||||||
scan_cfg := ScanConfig {
|
|
||||||
Matcher = strings.clone("\\.env"),
|
|
||||||
Exclude = exclude,
|
|
||||||
Include = include,
|
|
||||||
}
|
|
||||||
|
|
||||||
return Config{Keys = keys, ScanConfig = scan_cfg, config_path = cfg_path}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
find_ssh_private_keys :: proc() -> (keys: [dynamic]string, ok: bool) {
|
find_ssh_private_keys :: proc() -> (keys: [dynamic]string, ok: bool) {
|
||||||
@@ -188,11 +128,73 @@ find_ssh_private_keys :: proc() -> (keys: [dynamic]string, ok: bool) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
find_git_roots :: proc(cfg: Config) -> (roots: [dynamic]string, ok: bool) {
|
// Caller is responsible for calling delete_config()
|
||||||
paths := search_paths(cfg)
|
new_config :: proc(
|
||||||
findr.find_repos(paths[:], &roots, os.get_processor_core_count())
|
private_key_paths: []string,
|
||||||
ok = true
|
cfg_path: string = "~/.envr/config.json",
|
||||||
return
|
) -> Config {
|
||||||
|
keys := make([dynamic]SshKeyPair, 0, len(private_key_paths))
|
||||||
|
for priv in private_key_paths {
|
||||||
|
// TODO: Is this bad?
|
||||||
|
priv_key := strings.clone(priv)
|
||||||
|
pub, _ := strings.concatenate([]string{priv_key, ".pub"})
|
||||||
|
append(&keys, SshKeyPair{Private = priv_key, Public = pub})
|
||||||
|
}
|
||||||
|
|
||||||
|
exclude := make([dynamic]string, 0, 4)
|
||||||
|
append(&exclude, strings.clone("*\\.envrc"))
|
||||||
|
append(&exclude, strings.clone("\\.local/"))
|
||||||
|
append(&exclude, strings.clone("node_modules"))
|
||||||
|
append(&exclude, strings.clone("vendor"))
|
||||||
|
|
||||||
|
include := make([dynamic]string, 0, 1)
|
||||||
|
append(&include, strings.clone("~"))
|
||||||
|
|
||||||
|
scan_cfg := ScanConfig {
|
||||||
|
Matcher = strings.clone("\\.env"),
|
||||||
|
Exclude = exclude,
|
||||||
|
Include = include,
|
||||||
|
}
|
||||||
|
|
||||||
|
return Config{Keys = keys, ScanConfig = scan_cfg, config_path = cfg_path}
|
||||||
|
}
|
||||||
|
|
||||||
|
save_config :: proc(cfg: Config, force: bool = false) -> bool {
|
||||||
|
config_dir := envr_dir(cfg.config_path)
|
||||||
|
|
||||||
|
if !os.exists(config_dir) {
|
||||||
|
mkdir_err := os.make_directory(config_dir)
|
||||||
|
if mkdir_err != nil {
|
||||||
|
fmt.printf("Error creating %s directory: %v\n", config_dir, mkdir_err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if os.exists(cfg.config_path) && !force {
|
||||||
|
info, stat_err := os.stat(cfg.config_path, context.allocator)
|
||||||
|
if stat_err == nil {
|
||||||
|
defer os.file_info_delete(info, context.allocator)
|
||||||
|
if info.size > 0 {
|
||||||
|
fmt.println("Config file already exists. Run again with --force to reinitialize.")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
data, marshal_err := json.marshal(cfg, {pretty = true, use_spaces = true, spaces = 2})
|
||||||
|
if marshal_err != nil {
|
||||||
|
fmt.printf("Error marshaling config: %v\n", marshal_err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
defer delete(data)
|
||||||
|
|
||||||
|
write_err := os.write_entire_file(cfg.config_path, data)
|
||||||
|
if write_err != nil {
|
||||||
|
fmt.printf("Error writing config: %v\n", write_err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
search_paths :: proc(cfg: Config) -> (paths: [dynamic]string) {
|
search_paths :: proc(cfg: Config) -> (paths: [dynamic]string) {
|
||||||
@@ -216,13 +218,10 @@ search_paths :: proc(cfg: Config) -> (paths: [dynamic]string) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
envr_dir :: proc(config_path: string) -> string {
|
find_git_roots :: proc(cfg: Config) -> (roots: [dynamic]string, ok: bool) {
|
||||||
return filepath.dir(config_path)
|
paths := search_paths(cfg)
|
||||||
}
|
findr.find_repos(paths[:], &roots, os.get_processor_core_count())
|
||||||
|
ok = true
|
||||||
// User is responsible for freeing the path
|
return
|
||||||
data_path :: proc(config_path: string, allocator := context.allocator) -> string {
|
|
||||||
path, _ := filepath.join([]string{envr_dir(config_path), "data.envr"}, allocator)
|
|
||||||
return path
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
123
crypto.odin
123
crypto.odin
@@ -2,7 +2,6 @@ package main
|
|||||||
|
|
||||||
import "core:fmt"
|
import "core:fmt"
|
||||||
import "core:mem"
|
import "core:mem"
|
||||||
import "core:os"
|
|
||||||
|
|
||||||
MAGIC :: "ENVR"
|
MAGIC :: "ENVR"
|
||||||
MAGIC_BYTES := [4]u8{u8('E'), u8('N'), u8('V'), u8('R')}
|
MAGIC_BYTES := [4]u8{u8('E'), u8('N'), u8('V'), u8('R')}
|
||||||
@@ -21,19 +20,80 @@ RecipientEntry :: struct {
|
|||||||
EncryptedKey: [CRYPTO_SECRETBOX_KEY_BYTES + CRYPTO_BOX_MAC_BYTES]u8,
|
EncryptedKey: [CRYPTO_SECRETBOX_KEY_BYTES + CRYPTO_BOX_MAC_BYTES]u8,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
sodium_initialized: bool
|
||||||
|
|
||||||
|
ensure_sodium :: proc() -> bool {
|
||||||
|
if sodium_initialized {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
rc := sodium_init()
|
||||||
|
if rc < 0 {
|
||||||
|
fmt.println("Error: libsodium initialization failed")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
sodium_initialized = true
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
X25519Keypair :: struct {
|
X25519Keypair :: struct {
|
||||||
Public: [CRYPTO_BOX_PUBLICKEY_BYTES]u8,
|
Public: [CRYPTO_BOX_PUBLICKEY_BYTES]u8,
|
||||||
Private: [CRYPTO_BOX_SECRETKEY_BYTES]u8,
|
Private: [CRYPTO_BOX_SECRETKEY_BYTES]u8,
|
||||||
}
|
}
|
||||||
|
|
||||||
@(init)
|
ssh_to_x25519 :: proc(keys: []SshKeyPair) -> (pairs: []X25519Keypair, ok: bool) {
|
||||||
init_sodium :: proc "contextless" () {
|
if len(keys) == 0 {
|
||||||
if sodium_init() < 0 {
|
return
|
||||||
os.exit(1)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pairs = make([]X25519Keypair, len(keys))
|
||||||
|
|
||||||
|
for i in 0 ..< len(keys) {
|
||||||
|
ssh_kp, parse_ok := parse_ssh_private_key(keys[i].Private)
|
||||||
|
if !parse_ok {
|
||||||
|
fmt.printf("Error: failed to parse SSH private key: %s\n", keys[i].Private)
|
||||||
|
delete(pairs)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ssh_pub, pub_ok := parse_ssh_public_key(keys[i].Public)
|
||||||
|
if !pub_ok {
|
||||||
|
fmt.printf("Error: failed to parse SSH public key: %s\n", keys[i].Public)
|
||||||
|
delete(pairs)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
pk_rc := crypto_sign_ed25519_pk_to_curve25519(&pairs[i].Public[0], &ssh_pub[0])
|
||||||
|
if pk_rc != 0 {
|
||||||
|
fmt.println("Error: failed to convert ed25519 public key to curve25519")
|
||||||
|
delete(pairs)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ed25519_sk: [64]u8
|
||||||
|
for j in 0 ..< 32 {
|
||||||
|
ed25519_sk[j] = ssh_kp.Private[j]
|
||||||
|
}
|
||||||
|
for j in 0 ..< 32 {
|
||||||
|
ed25519_sk[32 + j] = ssh_kp.Public[j]
|
||||||
|
}
|
||||||
|
|
||||||
|
sk_rc := crypto_sign_ed25519_sk_to_curve25519(&pairs[i].Private[0], &ed25519_sk[0])
|
||||||
|
if sk_rc != 0 {
|
||||||
|
fmt.println("Error: failed to convert ed25519 private key to curve25519")
|
||||||
|
delete(pairs)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
ok = true
|
||||||
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
encrypt :: proc(plaintext: []u8, keys: []SshKeyPair) -> (ciphertext: []u8, ok: bool) {
|
encrypt :: proc(plaintext: []u8, keys: []SshKeyPair) -> (ciphertext: []u8, ok: bool) {
|
||||||
|
if !ensure_sodium() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
x25519_pairs, pairs_ok := ssh_to_x25519(keys)
|
x25519_pairs, pairs_ok := ssh_to_x25519(keys)
|
||||||
if !pairs_ok {
|
if !pairs_ok {
|
||||||
return
|
return
|
||||||
@@ -133,6 +193,10 @@ encrypt :: proc(plaintext: []u8, keys: []SshKeyPair) -> (ciphertext: []u8, ok: b
|
|||||||
}
|
}
|
||||||
|
|
||||||
decrypt :: proc(ciphertext: []u8, keys: []SshKeyPair) -> (plaintext: []u8, ok: bool) {
|
decrypt :: proc(ciphertext: []u8, keys: []SshKeyPair) -> (plaintext: []u8, ok: bool) {
|
||||||
|
if !ensure_sodium() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
if len(ciphertext) < HEADER_SIZE {
|
if len(ciphertext) < HEADER_SIZE {
|
||||||
fmt.println("Error: ciphertext too short (header)")
|
fmt.println("Error: ciphertext too short (header)")
|
||||||
return
|
return
|
||||||
@@ -272,52 +336,3 @@ decrypt :: proc(ciphertext: []u8, keys: []SshKeyPair) -> (plaintext: []u8, ok: b
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
ssh_to_x25519 :: proc(keys: []SshKeyPair) -> (pairs: []X25519Keypair, ok: bool) {
|
|
||||||
if len(keys) == 0 {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
pairs = make([]X25519Keypair, len(keys))
|
|
||||||
|
|
||||||
for i in 0 ..< len(keys) {
|
|
||||||
ssh_kp, parse_ok := parse_ssh_private_key(keys[i].Private)
|
|
||||||
if !parse_ok {
|
|
||||||
fmt.printf("Error: failed to parse SSH private key: %s\n", keys[i].Private)
|
|
||||||
delete(pairs)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
ssh_pub, pub_ok := parse_ssh_public_key(keys[i].Public)
|
|
||||||
if !pub_ok {
|
|
||||||
fmt.printf("Error: failed to parse SSH public key: %s\n", keys[i].Public)
|
|
||||||
delete(pairs)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
pk_rc := crypto_sign_ed25519_pk_to_curve25519(&pairs[i].Public[0], &ssh_pub[0])
|
|
||||||
if pk_rc != 0 {
|
|
||||||
fmt.println("Error: failed to convert ed25519 public key to curve25519")
|
|
||||||
delete(pairs)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
ed25519_sk: [64]u8
|
|
||||||
for j in 0 ..< 32 {
|
|
||||||
ed25519_sk[j] = ssh_kp.Private[j]
|
|
||||||
}
|
|
||||||
for j in 0 ..< 32 {
|
|
||||||
ed25519_sk[32 + j] = ssh_kp.Public[j]
|
|
||||||
}
|
|
||||||
|
|
||||||
sk_rc := crypto_sign_ed25519_sk_to_curve25519(&pairs[i].Private[0], &ed25519_sk[0])
|
|
||||||
if sk_rc != 0 {
|
|
||||||
fmt.println("Error: failed to convert ed25519 private key to curve25519")
|
|
||||||
delete(pairs)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
ok = true
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|||||||
380
db.odin
380
db.odin
@@ -51,85 +51,43 @@ delete_envfile :: proc(f: ^EnvFile) {
|
|||||||
delete(f.contents)
|
delete(f.contents)
|
||||||
}
|
}
|
||||||
|
|
||||||
db_open :: proc(cfg_path: string) -> (database: Db, ok: bool) {
|
|
||||||
database.cfg = load_config(cfg_path) or_return
|
|
||||||
|
|
||||||
{
|
db_open :: proc(cfg_path: string) -> (Db, bool) {
|
||||||
db: ^rawptr
|
cfg, ok := load_config(cfg_path)
|
||||||
rc := sqlite.db_open(":memory:", &db)
|
if !ok {
|
||||||
if rc != sqlite.OK {
|
return Db{}, false
|
||||||
fmt.printf("Error opening in-memory database: %s\n", sqlite.db_errmsg(db))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
create_sql: cstring = "CREATE TABLE IF NOT EXISTS envr_env_files (path TEXT PRIMARY KEY NOT NULL, remotes TEXT, sha256 TEXT NOT NULL, contents TEXT NOT NULL)"
|
|
||||||
rc = sqlite.db_exec(db, create_sql, nil, nil, nil)
|
|
||||||
if rc != sqlite.OK {
|
|
||||||
fmt.printf("Error creating table: %s\n", sqlite.db_errmsg(db))
|
|
||||||
sqlite.db_close(db)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
database.db = db
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: Use different allocators?
|
data_path := data_path(cfg.config_path)
|
||||||
data_path := data_path(database.cfg.config_path, context.temp_allocator)
|
_, stat_err := os.stat(data_path, context.allocator)
|
||||||
if os.exists(data_path) {
|
|
||||||
if ok = db_restore_from_encrypted(&database, data_path); !ok {
|
|
||||||
sqlite.db_close(database.db)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
// DB was created
|
|
||||||
database.changed = true
|
|
||||||
}
|
|
||||||
|
|
||||||
return database, true
|
db: ^rawptr
|
||||||
}
|
rc := sqlite.db_open(":memory:", &db)
|
||||||
|
|
||||||
db_restore_from_encrypted :: proc(db: ^Db, data_path: string) -> bool {
|
|
||||||
encrypted_data, read_err := os.read_entire_file_from_path(data_path, context.allocator)
|
|
||||||
defer delete(encrypted_data)
|
|
||||||
if read_err != nil {
|
|
||||||
fmt.printf("Error reading encrypted database: %v\n", read_err)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
plaintext, dec_ok := decrypt(encrypted_data, db.cfg.Keys[:])
|
|
||||||
if !dec_ok {
|
|
||||||
fmt.println("Error: decryption failed")
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
defer delete(plaintext)
|
|
||||||
|
|
||||||
n := i64(len(plaintext))
|
|
||||||
buf := sqlite.malloc64(n)
|
|
||||||
if buf == nil {
|
|
||||||
fmt.println("Error: failed to allocate buffer for deserialization")
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
copy(buf[:len(plaintext)], plaintext)
|
|
||||||
|
|
||||||
rc := sqlite.deserialize(
|
|
||||||
db.db,
|
|
||||||
"main",
|
|
||||||
buf,
|
|
||||||
n,
|
|
||||||
n,
|
|
||||||
sqlite.DESERIALIZE_FREEONCLOSE | sqlite.DESERIALIZE_RESIZEABLE,
|
|
||||||
)
|
|
||||||
if rc != sqlite.OK {
|
if rc != sqlite.OK {
|
||||||
sqlite.free(buf)
|
fmt.printf("Error opening in-memory database: %s\n", sqlite.db_errmsg(db))
|
||||||
fmt.printf("Error deserializing database: %s\n", sqlite.db_errmsg(db.db))
|
return Db{}, false
|
||||||
return false
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return true
|
create_sql: cstring = "CREATE TABLE IF NOT EXISTS envr_env_files (path TEXT PRIMARY KEY NOT NULL, remotes TEXT, sha256 TEXT NOT NULL, contents TEXT NOT NULL)"
|
||||||
|
rc = sqlite.db_exec(db, create_sql, nil, nil, nil)
|
||||||
|
if rc != sqlite.OK {
|
||||||
|
fmt.printf("Error creating table: %s\n", sqlite.db_errmsg(db))
|
||||||
|
sqlite.db_close(db)
|
||||||
|
return Db{}, false
|
||||||
|
}
|
||||||
|
|
||||||
|
if stat_err == nil {
|
||||||
|
if !db_restore_from_encrypted(db, cfg) {
|
||||||
|
sqlite.db_close(db)
|
||||||
|
return Db{}, false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return Db{db = db, cfg = cfg, changed = stat_err != nil}, true
|
||||||
}
|
}
|
||||||
|
|
||||||
db_close :: proc(d: ^Db) {
|
db_close :: proc(d: ^Db) {
|
||||||
defer sqlite.db_close(d.db)
|
defer sqlite.db_close(d.db)
|
||||||
defer delete_config(&d.cfg)
|
|
||||||
|
|
||||||
if d.changed {
|
if d.changed {
|
||||||
rc := sqlite.db_exec(d.db, "VACUUM", nil, nil, nil)
|
rc := sqlite.db_exec(d.db, "VACUUM", nil, nil, nil)
|
||||||
@@ -168,6 +126,13 @@ db_close :: proc(d: ^Db) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Caller is responsible for calling:
|
||||||
|
// ```odin
|
||||||
|
// delete(results)
|
||||||
|
// for &result in results {
|
||||||
|
// delete(&result)
|
||||||
|
// }
|
||||||
|
// ```
|
||||||
db_list :: proc(d: ^Db, allocator := context.allocator) -> (results: [dynamic]EnvFile, ok: bool) {
|
db_list :: proc(d: ^Db, allocator := context.allocator) -> (results: [dynamic]EnvFile, ok: bool) {
|
||||||
stmt: ^rawptr
|
stmt: ^rawptr
|
||||||
rc := sqlite.prepare_v2(
|
rc := sqlite.prepare_v2(
|
||||||
@@ -216,6 +181,110 @@ db_list :: proc(d: ^Db, allocator := context.allocator) -> (results: [dynamic]En
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
db_restore_from_encrypted :: proc(db: ^rawptr, cfg: Config) -> bool {
|
||||||
|
encrypted_data, read_err := os.read_entire_file_from_path(
|
||||||
|
data_path(cfg.config_path),
|
||||||
|
context.allocator,
|
||||||
|
)
|
||||||
|
defer delete(encrypted_data)
|
||||||
|
if read_err != nil {
|
||||||
|
fmt.printf("Error reading encrypted database: %v\n", read_err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
plaintext, dec_ok := decrypt(encrypted_data, cfg.Keys[:])
|
||||||
|
if !dec_ok {
|
||||||
|
fmt.println("Error: decryption failed")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
defer delete(plaintext)
|
||||||
|
|
||||||
|
n := i64(len(plaintext))
|
||||||
|
buf := sqlite.malloc64(n)
|
||||||
|
if buf == nil {
|
||||||
|
fmt.println("Error: failed to allocate buffer for deserialization")
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
copy(buf[:len(plaintext)], plaintext)
|
||||||
|
|
||||||
|
rc := sqlite.deserialize(
|
||||||
|
db,
|
||||||
|
"main",
|
||||||
|
buf,
|
||||||
|
n,
|
||||||
|
n,
|
||||||
|
sqlite.DESERIALIZE_FREEONCLOSE | sqlite.DESERIALIZE_RESIZEABLE,
|
||||||
|
)
|
||||||
|
if rc != sqlite.OK {
|
||||||
|
sqlite.free(buf)
|
||||||
|
fmt.printf("Error deserializing database: %s\n", sqlite.db_errmsg(db))
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
get_git_remotes :: proc(dir: string) -> [dynamic]string {
|
||||||
|
remotes: [dynamic]string
|
||||||
|
remote_set: map[string]bool
|
||||||
|
defer delete(remote_set)
|
||||||
|
|
||||||
|
config_path, _ := filepath.join({dir, ".git", "config"}, context.temp_allocator)
|
||||||
|
m, _, ok := ini.load_map_from_path(config_path, context.allocator)
|
||||||
|
if !ok {
|
||||||
|
return remotes
|
||||||
|
}
|
||||||
|
defer ini.delete_map(m)
|
||||||
|
|
||||||
|
for section_name, section in m {
|
||||||
|
if strings.has_prefix(section_name, "remote ") {
|
||||||
|
if url, ok := section["url"]; ok {
|
||||||
|
remote_set[url] = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for remote in remote_set {
|
||||||
|
cloned, _ := strings.clone(remote)
|
||||||
|
append(&remotes, cloned)
|
||||||
|
}
|
||||||
|
|
||||||
|
return remotes
|
||||||
|
}
|
||||||
|
|
||||||
|
new_env_file :: proc(path: string) -> (EnvFile, bool) {
|
||||||
|
abs_path, abs_err := filepath.abs(path)
|
||||||
|
if abs_err != nil {
|
||||||
|
fmt.printf("Error getting absolute path: %v\n", abs_err)
|
||||||
|
return EnvFile{}, false
|
||||||
|
}
|
||||||
|
|
||||||
|
dir := filepath.dir(abs_path)
|
||||||
|
|
||||||
|
remotes := get_git_remotes(dir)
|
||||||
|
|
||||||
|
data, read_err := os.read_entire_file_from_path(abs_path, context.allocator)
|
||||||
|
defer delete(data)
|
||||||
|
if read_err != nil {
|
||||||
|
fmt.printf("Error reading file %s: %v\n", abs_path, read_err)
|
||||||
|
return EnvFile{}, false
|
||||||
|
}
|
||||||
|
|
||||||
|
digest := hash.hash_bytes(hash.Algorithm.SHA256, data, context.temp_allocator)
|
||||||
|
// TODO: Handle error
|
||||||
|
hex_bytes, _ := hex.encode(digest)
|
||||||
|
|
||||||
|
return EnvFile {
|
||||||
|
Path = abs_path,
|
||||||
|
Dir = dir,
|
||||||
|
Remotes = remotes,
|
||||||
|
Sha256 = string(hex_bytes),
|
||||||
|
contents = string(data),
|
||||||
|
},
|
||||||
|
true
|
||||||
|
}
|
||||||
|
|
||||||
db_insert :: proc(d: ^Db, file: EnvFile) -> bool {
|
db_insert :: proc(d: ^Db, file: EnvFile) -> bool {
|
||||||
remotes_json, marshal_err := json.marshal(file.Remotes)
|
remotes_json, marshal_err := json.marshal(file.Remotes)
|
||||||
if marshal_err != nil {
|
if marshal_err != nil {
|
||||||
@@ -355,36 +424,69 @@ db_delete :: proc(d: ^Db, path: string) -> bool {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
new_env_file :: proc(path: string) -> (EnvFile, bool) {
|
to_cstring :: proc {
|
||||||
abs_path, abs_err := filepath.abs(path)
|
string_to_cstring,
|
||||||
if abs_err != nil {
|
strings.to_cstring,
|
||||||
fmt.printf("Error getting absolute path: %v\n", abs_err)
|
}
|
||||||
return EnvFile{}, false
|
|
||||||
|
string_to_cstring :: proc(s: string, allocator := context.allocator) -> cstring {
|
||||||
|
cs, err := strings.clone_to_cstring(s, allocator)
|
||||||
|
if err != nil {
|
||||||
|
fmt.printf("Failed to convert string to cstring: %v\n", err)
|
||||||
|
panic("Allocation Exception")
|
||||||
|
}
|
||||||
|
return cs
|
||||||
|
}
|
||||||
|
|
||||||
|
clone_cstring :: proc(c: cstring, allocator := context.allocator) -> string {
|
||||||
|
str, err := strings.clone_from_cstring(c, allocator)
|
||||||
|
if err != nil {
|
||||||
|
fmt.printf("Failed to convert string to cstring: %v\n", err)
|
||||||
|
delete(str)
|
||||||
|
panic("Allocation Exception")
|
||||||
}
|
}
|
||||||
|
|
||||||
dir := filepath.dir(abs_path)
|
return str
|
||||||
|
}
|
||||||
|
|
||||||
remotes := get_git_remotes(dir)
|
db_update_required :: proc(status: SyncFlag) -> bool {
|
||||||
|
return .BackedUp in status || .DirUpdated in status
|
||||||
|
}
|
||||||
|
|
||||||
data, read_err := os.read_entire_file_from_path(abs_path, context.allocator)
|
shares_remote :: proc(f: ^EnvFile, remotes: []string) -> bool {
|
||||||
defer delete(data)
|
for r1 in f.Remotes {
|
||||||
if read_err != nil {
|
for r2 in remotes {
|
||||||
fmt.printf("Error reading file %s: %v\n", abs_path, read_err)
|
if r1 == r2 {
|
||||||
return EnvFile{}, false
|
return true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
update_dir :: proc(f: ^EnvFile, new_dir: string) {
|
||||||
|
f.Dir = new_dir
|
||||||
|
base := filepath.base(f.Path)
|
||||||
|
new_path, _ := strings.concatenate({new_dir, "/", base})
|
||||||
|
f.Path = new_path
|
||||||
|
f.Remotes = get_git_remotes(new_dir)
|
||||||
|
}
|
||||||
|
|
||||||
|
find_moved_dirs :: proc(d: ^Db, f: ^EnvFile) -> ([dynamic]string, bool) {
|
||||||
|
roots, roots_ok := find_git_roots(d.cfg)
|
||||||
|
if !roots_ok {
|
||||||
|
return {}, false
|
||||||
}
|
}
|
||||||
|
|
||||||
digest := hash.hash_bytes(hash.Algorithm.SHA256, data, context.temp_allocator)
|
moved: [dynamic]string
|
||||||
// TODO: Handle error
|
for root in roots {
|
||||||
hex_bytes, _ := hex.encode(digest)
|
remotes := get_git_remotes(root)
|
||||||
|
if shares_remote(f, remotes[:]) {
|
||||||
return EnvFile {
|
cloned, _ := strings.clone(root)
|
||||||
Path = abs_path,
|
append(&moved, cloned)
|
||||||
Dir = dir,
|
}
|
||||||
Remotes = remotes,
|
}
|
||||||
Sha256 = string(hex_bytes),
|
return moved, true
|
||||||
contents = string(data),
|
|
||||||
},
|
|
||||||
true
|
|
||||||
}
|
}
|
||||||
|
|
||||||
db_sync :: proc(d: ^Db, f: ^EnvFile) -> (SyncFlag, string) {
|
db_sync :: proc(d: ^Db, f: ^EnvFile) -> (SyncFlag, string) {
|
||||||
@@ -463,31 +565,6 @@ env_file_sync :: proc(f: ^EnvFile, dir: SyncDirection, d: ^Db) -> (SyncFlag, str
|
|||||||
return result, ""
|
return result, ""
|
||||||
}
|
}
|
||||||
|
|
||||||
find_moved_dirs :: proc(d: ^Db, f: ^EnvFile) -> ([dynamic]string, bool) {
|
|
||||||
roots, roots_ok := find_git_roots(d.cfg)
|
|
||||||
if !roots_ok {
|
|
||||||
return {}, false
|
|
||||||
}
|
|
||||||
|
|
||||||
moved: [dynamic]string
|
|
||||||
for root in roots {
|
|
||||||
remotes := get_git_remotes(root)
|
|
||||||
if shares_remote(f, remotes[:]) {
|
|
||||||
cloned, _ := strings.clone(root)
|
|
||||||
append(&moved, cloned)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return moved, true
|
|
||||||
}
|
|
||||||
|
|
||||||
update_dir :: proc(f: ^EnvFile, new_dir: string) {
|
|
||||||
f.Dir = new_dir
|
|
||||||
base := filepath.base(f.Path)
|
|
||||||
new_path, _ := strings.concatenate({new_dir, "/", base})
|
|
||||||
f.Path = new_path
|
|
||||||
f.Remotes = get_git_remotes(new_dir)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Loads the contents of the the file at f.Path into f.contents
|
// Loads the contents of the the file at f.Path into f.contents
|
||||||
//
|
//
|
||||||
// Caller is responsible for calling delete on f.contents and f.Sha256
|
// Caller is responsible for calling delete on f.contents and f.Sha256
|
||||||
@@ -509,72 +586,3 @@ env_file_backup :: proc(f: ^EnvFile) -> bool {
|
|||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
shares_remote :: proc(f: ^EnvFile, remotes: []string) -> bool {
|
|
||||||
for r1 in f.Remotes {
|
|
||||||
for r2 in remotes {
|
|
||||||
if r1 == r2 {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
|
|
||||||
get_git_remotes :: proc(dir: string) -> [dynamic]string {
|
|
||||||
remotes: [dynamic]string
|
|
||||||
remote_set: map[string]bool
|
|
||||||
defer delete(remote_set)
|
|
||||||
|
|
||||||
config_path, _ := filepath.join({dir, ".git", "config"}, context.temp_allocator)
|
|
||||||
m, _, ok := ini.load_map_from_path(config_path, context.allocator)
|
|
||||||
if !ok {
|
|
||||||
return remotes
|
|
||||||
}
|
|
||||||
defer ini.delete_map(m)
|
|
||||||
|
|
||||||
for section_name, section in m {
|
|
||||||
if strings.has_prefix(section_name, "remote ") {
|
|
||||||
if url, ok := section["url"]; ok {
|
|
||||||
remote_set[url] = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for remote in remote_set {
|
|
||||||
cloned, _ := strings.clone(remote)
|
|
||||||
append(&remotes, cloned)
|
|
||||||
}
|
|
||||||
|
|
||||||
return remotes
|
|
||||||
}
|
|
||||||
|
|
||||||
db_update_required :: proc(status: SyncFlag) -> bool {
|
|
||||||
return .BackedUp in status || .DirUpdated in status
|
|
||||||
}
|
|
||||||
|
|
||||||
to_cstring :: proc {
|
|
||||||
string_to_cstring,
|
|
||||||
strings.to_cstring,
|
|
||||||
}
|
|
||||||
|
|
||||||
string_to_cstring :: proc(s: string, allocator := context.allocator) -> cstring {
|
|
||||||
cs, err := strings.clone_to_cstring(s, allocator)
|
|
||||||
if err != nil {
|
|
||||||
fmt.printf("Failed to convert string to cstring: %v\n", err)
|
|
||||||
panic("Allocation Exception")
|
|
||||||
}
|
|
||||||
return cs
|
|
||||||
}
|
|
||||||
|
|
||||||
// Caller is responsible for freeing the result
|
|
||||||
clone_cstring :: proc(c: cstring, allocator := context.allocator) -> string {
|
|
||||||
str, err := strings.clone_from_cstring(c, allocator)
|
|
||||||
if err != nil {
|
|
||||||
fmt.printf("Failed to convert string to cstring: %v\n", err)
|
|
||||||
delete(str)
|
|
||||||
panic("Allocation Exception")
|
|
||||||
}
|
|
||||||
|
|
||||||
return str
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|||||||
48
db_test.odin
48
db_test.odin
@@ -472,51 +472,3 @@ test_update_dir :: proc(t: ^testing.T) {
|
|||||||
testing.expect_value(t, f.Path, "/new/location/.env")
|
testing.expect_value(t, f.Path, "/new/location/.env")
|
||||||
}
|
}
|
||||||
|
|
||||||
@(test)
|
|
||||||
test_closing_db_has_no_leaks :: proc(t: ^testing.T) {
|
|
||||||
base := fmt.tprintf("/tmp/envr-test-leak-%d", os.get_pid())
|
|
||||||
os.mkdir_all(base)
|
|
||||||
defer os.remove_all(base)
|
|
||||||
|
|
||||||
cfg_path, err := filepath.join([]string{base, "config.json"}, context.temp_allocator)
|
|
||||||
testing.expect(t, err == nil, "cfgPath should build successfully")
|
|
||||||
|
|
||||||
cfg := new_config([]string{"fixtures/keys/insecure-test-key"}, cfg_path)
|
|
||||||
defer delete_config(&cfg)
|
|
||||||
testing.expect(t, save_config(cfg, force = true), "save should succeed")
|
|
||||||
|
|
||||||
db, ok := db_open(cfg_path)
|
|
||||||
testing.expect(t, ok, "db should open")
|
|
||||||
if !ok do return
|
|
||||||
db_close(&db)
|
|
||||||
}
|
|
||||||
|
|
||||||
@(test)
|
|
||||||
test_open_existing_db_has_no_leaks :: proc(t: ^testing.T) {
|
|
||||||
base := fmt.tprintf("/tmp/envr-test-leak-existing-%d", os.get_pid())
|
|
||||||
os.mkdir_all(base)
|
|
||||||
defer os.remove_all(base)
|
|
||||||
|
|
||||||
cfg_path, err := filepath.join([]string{base, "config.json"}, context.temp_allocator)
|
|
||||||
testing.expect(t, err == nil, "cfgPath should build successfully")
|
|
||||||
|
|
||||||
cfg := new_config([]string{"fixtures/keys/insecure-test-key"}, cfg_path)
|
|
||||||
defer delete_config(&cfg)
|
|
||||||
testing.expect(t, save_config(cfg, force = true), "save should succeed")
|
|
||||||
|
|
||||||
// First open/close creates data.envr on disk
|
|
||||||
db, ok := db_open(cfg_path)
|
|
||||||
testing.expect(t, ok, "db should open")
|
|
||||||
if !ok do return
|
|
||||||
f := make_test_env_file("/project/.env", "abc123", "SECRET=value", []string{"git@github.com:user/repo.git"})
|
|
||||||
defer delete(f.Remotes)
|
|
||||||
testing.expect(t, db_insert(&db, f), "insert should succeed")
|
|
||||||
db_close(&db)
|
|
||||||
|
|
||||||
// Second open exercises db_restore_from_encrypted
|
|
||||||
db2, ok2 := db_open(cfg_path)
|
|
||||||
testing.expect(t, ok2, "db should open existing")
|
|
||||||
if !ok2 do return
|
|
||||||
db_close(&db2)
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,5 @@
|
|||||||
package findr
|
package findr
|
||||||
|
|
||||||
import "core:bytes"
|
|
||||||
import "core:fmt"
|
import "core:fmt"
|
||||||
import "core:os"
|
import "core:os"
|
||||||
import "core:strings"
|
import "core:strings"
|
||||||
@@ -55,26 +54,19 @@ Collector_Data :: struct {
|
|||||||
collect_worker :: proc(t: ^thread.Thread) {
|
collect_worker :: proc(t: ^thread.Thread) {
|
||||||
data := cast(^Collector_Data)t.data
|
data := cast(^Collector_Data)t.data
|
||||||
for {
|
for {
|
||||||
batch := chan.recv(data.ch) or_break
|
batch, ok := chan.recv(data.ch)
|
||||||
defer delete(batch)
|
if !ok do break
|
||||||
|
|
||||||
start := 0
|
start := 0
|
||||||
for {
|
for i in 0 ..< len(batch) {
|
||||||
remaining: []u8
|
if batch[i] == '\n' {
|
||||||
#no_bounds_check {remaining = batch[start:]}
|
if i > start {
|
||||||
|
s, _ := strings.clone(string(batch[start:i]))
|
||||||
idx := bytes.index_byte(remaining, '\n')
|
append(data.results, s)
|
||||||
if idx < 0 do break
|
}
|
||||||
|
start = i + 1
|
||||||
i := start + idx
|
|
||||||
if i > start {
|
|
||||||
segment: []u8
|
|
||||||
#no_bounds_check {segment = batch[start:i]}
|
|
||||||
s, _ := strings.clone(string(segment))
|
|
||||||
append(data.results, s)
|
|
||||||
}
|
}
|
||||||
start = i + 1
|
|
||||||
}
|
}
|
||||||
|
delete(batch)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -455,4 +447,3 @@ join_path :: proc(parent, child: string) -> string {
|
|||||||
copy(buf[pos:], child)
|
copy(buf[pos:], child)
|
||||||
return string(buf)
|
return string(buf)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
21
main.odin
21
main.odin
@@ -1,31 +1,14 @@
|
|||||||
package main
|
package main
|
||||||
|
|
||||||
|
import "core:bufio"
|
||||||
import "core:fmt"
|
import "core:fmt"
|
||||||
import "core:mem"
|
|
||||||
import "core:os"
|
import "core:os"
|
||||||
|
|
||||||
main :: proc() {
|
main :: proc() {
|
||||||
when ODIN_DEBUG {
|
|
||||||
heap_track: mem.Tracking_Allocator
|
|
||||||
mem.tracking_allocator_init(&heap_track, context.allocator)
|
|
||||||
defer mem.tracking_allocator_destroy(&heap_track)
|
|
||||||
defer if len(heap_track.allocation_map) > 0 {
|
|
||||||
for _, leak in heap_track.allocation_map {
|
|
||||||
fmt.eprintf("LEAK: %v leaked %m\n", leak.location, leak.size)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
context.allocator = mem.tracking_allocator(&heap_track)
|
|
||||||
|
|
||||||
temp_track: mem.Tracking_Allocator
|
|
||||||
mem.tracking_allocator_init(&temp_track, context.temp_allocator)
|
|
||||||
defer mem.tracking_allocator_destroy(&temp_track)
|
|
||||||
context.temp_allocator = mem.tracking_allocator(&temp_track)
|
|
||||||
}
|
|
||||||
|
|
||||||
defer free_all(context.temp_allocator)
|
defer free_all(context.temp_allocator)
|
||||||
|
|
||||||
cmd, ok := parse_args(os.args, os.to_writer(os.stdout), os.to_writer(os.stderr))
|
cmd, ok := parse_args(os.args, os.to_writer(os.stdout), os.to_writer(os.stderr))
|
||||||
defer delete_command(&cmd) // delete flushes automatically
|
defer bufio.writer_flush(cmd.out_buf)
|
||||||
if !ok {
|
if !ok {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
36
ssh.odin
36
ssh.odin
@@ -12,6 +12,24 @@ Ed25519Keypair :: struct {
|
|||||||
Private: [32]u8,
|
Private: [32]u8,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
read_wire_string :: proc(data: []u8, offset: ^int) -> (s: string, ok: bool) {
|
||||||
|
if offset^ + 4 > len(data) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
length := u32(data[offset^]) << 24 | u32(data[offset^ + 1]) << 16 |
|
||||||
|
u32(data[offset^ + 2]) << 8 | u32(data[offset^ + 3])
|
||||||
|
offset^ += 4
|
||||||
|
|
||||||
|
if offset^ + int(length) > len(data) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
s = string(data[offset^ : offset^ + int(length)])
|
||||||
|
offset^ += int(length)
|
||||||
|
ok = true
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
parse_ssh_public_key :: proc(pub_path: string) -> (pub: [32]u8, ok: bool) {
|
parse_ssh_public_key :: proc(pub_path: string) -> (pub: [32]u8, ok: bool) {
|
||||||
data, err := os.read_entire_file_from_path(pub_path, context.temp_allocator)
|
data, err := os.read_entire_file_from_path(pub_path, context.temp_allocator)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -235,21 +253,3 @@ is_encrypted_key :: proc(priv_path: string) -> bool {
|
|||||||
|
|
||||||
return ciphername != "none"
|
return ciphername != "none"
|
||||||
}
|
}
|
||||||
|
|
||||||
read_wire_string :: proc(data: []u8, offset: ^int) -> (s: string, ok: bool) {
|
|
||||||
if offset^ + 4 > len(data) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
length := u32(data[offset^]) << 24 | u32(data[offset^ + 1]) << 16 |
|
|
||||||
u32(data[offset^ + 2]) << 8 | u32(data[offset^ + 3])
|
|
||||||
offset^ += 4
|
|
||||||
|
|
||||||
if offset^ + int(length) > len(data) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
s = string(data[offset^ : offset^ + int(length)])
|
|
||||||
offset^ += int(length)
|
|
||||||
ok = true
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|||||||
Reference in New Issue
Block a user