From 416d837307d7573460fd9935eabf7b2b2338e959 Mon Sep 17 00:00:00 2001
From: The Magician <themagician@engineer.com>
Date: Fri, 8 Dec 2023 18:09:21 +0000
Subject: [PATCH] Implement edit command

---
 tests/test__zk_edit.sh | 32 ++++++++++++++++++++++++++++++++
 zk                     | 10 ++++++++++
 2 files changed, 42 insertions(+)
 create mode 100644 tests/test__zk_edit.sh

diff --git a/tests/test__zk_edit.sh b/tests/test__zk_edit.sh
new file mode 100644
index 0000000..c82e3e2
--- /dev/null
+++ b/tests/test__zk_edit.sh
@@ -0,0 +1,32 @@
+#!/bin/bash
+
+setup() {
+    source ../zk
+}
+
+test__zk_edit_uses_filename_when_passed_as_parameter() {
+    # shellcheck disable=SC2016
+    fake vi 'echo vi "${FAKE_PARAMS[@]}"'
+
+    export EDITOR=""
+    zettel_name="11111111111111"
+    expected="vi $ZKDIR/$zettel_name"
+
+    result="$(_zk_edit $zettel_name)"
+
+    assert_equals "$expected" "$result"
+}
+
+test__zk_edit_uses__zk_search_when_filename_not_passed_as_parameter() {
+    export EDITOR=""
+    zettel_name="11111111111111"
+    expected="vi $ZKDIR/$zettel_name"
+
+    # shellcheck disable=SC2016
+    fake vi 'echo vi "${FAKE_PARAMS[@]}"'
+    fake _zk_search echo "$zettel_name"
+
+    result="$(_zk_edit)"
+
+    assert_equals "$expected" "$result"
+}
diff --git a/zk b/zk
index db23b77..3e6713e 100755
--- a/zk
+++ b/zk
@@ -39,6 +39,16 @@ _zk_new() {
 	fi
 }
 
+_zk_edit() {
+	filename="$1"
+
+	if test -z "$filename"; then
+		filename="$(_zk_search)"
+	fi
+
+	vi "$ZKDIR/$filename"
+}
+
 main() {
 	subcommand="$(get_command "$ZKCOMMANDS" "$1")"
 	_zk_"$subcommand"