4 files changed, 113 insertions, 56 deletions

diff --git a/note_tools/__init__.py b/note_tools/__init__.py
index e69de29..8d6774d 100644
--- a/note_tools/__init__.py
+++ b/note_tools/__init__.py
@@ -0,0 +1,37 @@
+from datetime import datetime
+import os
+from typing import Optional, Tuple
+
+class NoteBaseCommand(object):    
+  date = None
+
+  def file_name(self):
+    return f"{self.date.strftime('%Y%m%d')}.md"
+
+  def get_formatted_dates(self) -> Tuple[str, str]:
+    return (self.date.strftime("%Y%m%d"), self.date.strftime("%Y-%m-%d"))
+
+  def get_date(self) -> datetime:
+    return datetime.now()
+
+  def get_date_parts(self) -> Tuple[str]:
+    return (self.date.strftime('%Y'), self.date.strftime('%m'))
+      
+  def write_file_heading(self, file_path):      
+    date_yyyymmdd = self.date.strftime('%Y-%m-%d') 
+    underline = '=' * len(date_yyyymmdd)
+    with open(file_path, 'w') as file:
+      file.write(f"{date_yyyymmdd}\n{underline}\n\n\n")
+
+  def parse_file_date (self, file_name: str) -> Tuple[str, str]:
+    year = file_name[:4]
+    month = file_name[4:6]
+    return year, month
+
+  def get_notes_path(self, project_data) -> Optional[str]:
+    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":
+          return folder_path
+    return None
diff --git a/note_tools/commands.py b/note_tools/commands.py
index 5d150a1..e69de29 100644
--- a/note_tools/commands.py
+++ b/note_tools/commands.py
@@ -1,56 +0,0 @@
-import sublime
-import sublime_plugin
-import datetime
-import os
-from pathlib import Path
-from typing import List, Optional, Tuple
-
-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:
-
-            # 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()
diff --git a/note_tools/today.py b/note_tools/today.py
new file mode 100644
index 0000000..20f2e53
--- /dev/null
+++ b/note_tools/today.py
@@ -0,0 +1,49 @@
+from . import NoteBaseCommand
+import os
+import sublime
+import sublime_plugin
+from pathlib import Path 
+from typing import List, Optional
+
+class NewNoteCommand(NoteBaseCommand, sublime_plugin.WindowCommand):
+    def run(self):
+        # Get the current date formatted as YYYYMMDD and YYYY-MM-DD
+        self.date = self.get_date()                
+        notes_path = self.get_notes_path(self.window.project_data())
+
+        if notes_path:
+            year, month = self.get_date_parts()
+            directory = os.path.join(notes_path, year, month)
+            print("directory: ", directory)
+            os.makedirs(directory, exist_ok=True)
+            file_path = os.path.join(directory, self.file_name())
+            print('path: ', file_path)
+
+            # Check if file already exists
+            if os.path.exists(file_path):
+              sublime.message_dialog(f"The file '{self.file_name()}' already exists.")
+            else:
+              self.write_file_heading(file_path)
+              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()
+
+    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]
diff --git a/note_tools/tomorrow.py b/note_tools/tomorrow.py
new file mode 100644
index 0000000..2198a3b
--- /dev/null
+++ b/note_tools/tomorrow.py
@@ -0,0 +1,27 @@
+from . import NoteBaseCommand
+import os
+from datetime import timedelta
+import sublime
+import sublime_plugin
+
+class NewNoteForTomorrowCommand(NoteBaseCommand, sublime_plugin.WindowCommand):
+    def run(self):
+      self.date = self.get_date() + timedelta(days=1)
+
+      notes_path = self.get_notes_path(self.window.project_data())
+
+      if notes_path:
+        year, month = self.get_date_parts()
+        directory = os.path.join(notes_path, year, month)
+        os.makedirs(directory, exist_ok=True)
+        file_path = os.path.join(directory, self.file_name())
+
+        # Check if file already exists
+        if os.path.exists(file_path):
+            sublime.message_dialog(f"The file '{self.file_name()}' already exists.")
+        else:
+            self.write_file_heading(file_path)
+
+      else:
+          sublime.message_dialog("No 'Notes' directory found in the project. Please add a 'Notes' folder.")
+