1 files changed, 96 insertions, 0 deletions

diff --git a/notes/commands.py b/notes/commands.py
new file mode 100644
index 0000000..2a89a32
--- /dev/null
+++ b/notes/commands.py
@@ -0,0 +1,96 @@
+import sublime
+import sublime_plugin
+import datetime
+import os
+from pathlib import Path
+from typing import List, Optional, Tuple
+
+
+class OrganizeNotesCommand(sublime_plugin.WindowCommand):
+    def run(self):
+
+        now = datetime.datetime.now()
+
+        project_data = self.window.project_data()
+
+        if project_data and "folders" in project_data:
+            for folder in project_data['folders']:
+                folder_path = folder['path']
+                print(folder_path)
+                if os.path.basename(folder_path).lower() == "notes":
+                    notes_path = folder_path
+                    break
+
+        if notes_path:
+            notes = self.list_files(notes_path, ['.sublime-project', '.sublime-workspace'])
+            for note in notes:
+                year, month = self.parse_file_date(note.name)
+                dirs = os.path.join(notes_path, year, month)
+                os.makedirs(dirs, exist_ok=True)
+                new_note = os.path.join(notes_path, year, month, note.name)
+                os.rename(note.resolve(), new_note)
+        else:
+            sublime.message_dialog("No 'Notes' directory found in the project. Please add a 'Notes' folder.")
+            
+    def list_files(self, directory: str, ignore_extensions: Optional[List[str]] = None) -> List[Path]:
+        if ignore_extensions is None:
+            ignore_extensions = []
+        return [file for file in Path(directory).glob("*") if file.is_file() and file.suffix not in ignore_extensions]
+
+    def parse_file_date (self, file_name: str) -> Tuple[str, str]:
+        year = file_name[:4]
+        month = file_name[4:6]
+        return year, month
+
+
+class NewNoteCommand(sublime_plugin.WindowCommand):
+    def run(self):
+        # Get the current date formatted as YYYYMMDD and YYYY-MM-DD
+        date_str_yyyymmdd = datetime.datetime.now().strftime("%Y%m%d")
+        date_str_display = datetime.datetime.now().strftime("%Y-%m-%d")
+        file_name = f"{date_str_yyyymmdd}.md"
+        
+        # Create the underline based on the length of date_str_display
+        underline = '=' * len(date_str_display)
+        
+        # Get the project directory and look for a "Notes" or "notes" folder
+        project_data = self.window.project_data()
+        notes_path = None
+
+        if project_data and 'folders' in project_data:
+            for folder in project_data['folders']:
+                folder_path = folder['path']
+                if os.path.basename(folder_path).lower() == "notes":
+                    notes_path = folder_path
+                    break
+
+        if notes_path:
+            file_path = os.path.join(notes_path, file_name)
+
+            # Check if file already exists
+            if os.path.exists(file_path):
+                sublime.message_dialog(f"The file '{file_name}' already exists.")
+            else:
+                # Create the new file and write the date string with a line underneath
+                with open(file_path, 'w') as file:
+                    file.write(f"{date_str_display}\n{underline}\n\n\n")
+
+                # Open the file and set the cursor position
+                new_view = self.window.open_file(file_path)
+                self.set_cursor_position(new_view)
+
+        else:
+            sublime.message_dialog("No 'Notes' directory found in the project. Please add a 'Notes' folder.")
+
+    def set_cursor_position(self, view):
+        # Set the cursor position two lines beneath the underline
+        def on_load():
+            view.run_command("move_to", {"to": "bof"})
+            for _ in range(3):
+                view.run_command("move", {"by": "lines", "forward": True})
+        
+        # Add an event listener for when the file is loaded
+        if view.is_loading():
+            sublime.set_timeout_async(lambda: self.set_cursor_position(view), 100)
+        else:
+            on_load()