ruby_todo 0.4.1 → 1.0.1

data/lib/ruby_todo/cli.rb

@@ -7,14 +7,154 @@ require "tty-table"
 require "time"
 require "json"
 require "fileutils"
+require "csv"
 require_relative "models/notebook"
 require_relative "models/task"
 require_relative "models/template"
 require_relative "database"
+require_relative "commands/ai_assistant"
 
 module RubyTodo
+  module Export
+    private
+
+    def export_notebook(notebook)
+      {
+        "name" => notebook.name,
+        "created_at" => notebook.created_at,
+        "updated_at" => notebook.updated_at,
+        "tasks" => notebook.tasks.map { |task| task_to_hash(task) }
+      }
+    end
+
+    def export_all_notebooks(notebooks)
+      {
+        "notebooks" => notebooks.map { |notebook| export_notebook(notebook) }
+      }
+    end
+
+    def task_to_hash(task)
+      {
+        "title" => task.title,
+        "description" => task.description,
+        "status" => task.status,
+        "priority" => task.priority,
+        "tags" => task.tags,
+        "due_date" => task.due_date&.iso8601,
+        "created_at" => task.created_at&.iso8601,
+        "updated_at" => task.updated_at&.iso8601
+      }
+    end
+
+    def export_to_json(data, filename)
+      File.write(filename, JSON.pretty_generate(data))
+    end
+
+    def export_to_csv(data, filename)
+      CSV.open(filename, "wb") do |csv|
+        if data["notebooks"]
+          export_multiple_notebooks_to_csv(data, csv)
+        else
+          export_single_notebook_to_csv(data, csv)
+        end
+      end
+    end
+
+    def export_multiple_notebooks_to_csv(data, csv)
+      csv << ["Notebook", "Task ID", "Title", "Description", "Status", "Priority", "Tags", "Due Date",
+              "Created At", "Updated At"]
+
+      data["notebooks"].each do |notebook|
+        notebook_name = notebook["name"]
+        notebook["tasks"].each_with_index do |task, index|
+          csv << [
+            notebook_name,
+            index + 1,
+            task["title"],
+            task["description"],
+            task["status"],
+            task["priority"],
+            task["tags"],
+            task["due_date"],
+            task["created_at"],
+            task["updated_at"]
+          ]
+        end
+      end
+    end
+
+    def export_single_notebook_to_csv(data, csv)
+      csv << ["Task ID", "Title", "Description", "Status", "Priority", "Tags", "Due Date", "Created At",
+              "Updated At"]
+
+      data["tasks"].each_with_index do |task, index|
+        csv << [
+          index + 1,
+          task["title"],
+          task["description"],
+          task["status"],
+          task["priority"],
+          task["tags"],
+          task["due_date"],
+          task["created_at"],
+          task["updated_at"]
+        ]
+      end
+    end
+  end
+
+  module Import
+    private
+
+    def import_tasks(notebook, tasks_data)
+      count = 0
+
+      tasks_data.each do |task_data|
+        # Convert ISO8601 string to Time object
+        due_date = Time.parse(task_data["due_date"]) if task_data["due_date"]
+
+        task = Task.create(
+          notebook: notebook,
+          title: task_data["title"],
+          description: task_data["description"],
+          status: task_data["status"] || "todo",
+          priority: task_data["priority"],
+          tags: task_data["tags"],
+          due_date: due_date
+        )
+
+        count += 1 if task.persisted?
+      end
+
+      count
+    end
+
+    def import_all_notebooks(data)
+      results = { notebooks: 0, tasks: 0 }
+
+      data["notebooks"].each do |notebook_data|
+        notebook_name = notebook_data["name"]
+        notebook = Notebook.find_by(name: notebook_name)
+
+        unless notebook
+          notebook = Notebook.create(name: notebook_name)
+          results[:notebooks] += 1 if notebook.persisted?
+        end
+
+        if notebook.persisted?
+          tasks_count = import_tasks(notebook, notebook_data["tasks"])
+          results[:tasks] += tasks_count
+        end
+      end
+
+      results
+    end
+  end
+
   class CLI < Thor
     include Thor::Actions
+    include Export
+    include Import
 
     map %w[--version -v] => :version
     desc "version", "Show the Ruby Todo version"
@@ -203,19 +343,95 @@ module RubyTodo
       say "Ruby Todo has been initialized successfully!".green
     end
 
-    # Register subcommands
-    desc "notebook SUBCOMMAND", "Manage notebooks"
-    subcommand "notebook", NotebookCommand
+    # Register subcommands with colon format
+    desc "notebook:create NAME", "Create a new notebook"
+    def notebook_create(name)
+      Notebook.create(name: name)
+      puts "Created notebook: #{name}".green
+    end
 
-    desc "template SUBCOMMAND", "Manage task templates"
-    subcommand "template", TemplateCommand
+    desc "notebook:list", "List all notebooks"
+    def notebook_list
+      notebooks = Notebook.all
+      return say "No notebooks found".yellow if notebooks.empty?
+
+      table = TTY::Table.new(
+        header: ["ID", "Name", "Tasks", "Created At", "Default"],
+        rows: notebooks.map do |notebook|
+          [
+            notebook.id,
+            notebook.name,
+            notebook.tasks.count,
+            notebook.created_at,
+            notebook.is_default? ? "✓" : ""
+          ]
+        end
+      )
+      puts table.render(:ascii)
+    end
 
-    desc "task add [NOTEBOOK] [TITLE]", "Add a new task to a notebook"
+    desc "notebook:set_default NOTEBOOK", "Set a notebook as the default"
+    def notebook_set_default(name)
+      notebook = Notebook.find_by(name: name)
+      if notebook
+        notebook.make_default!
+        say "Successfully set '#{name}' as the default notebook".green
+      else
+        say "Notebook '#{name}' not found".red
+      end
+    end
+
+    desc "template:create NAME", "Create a new task template"
+    def template_create(name)
+      TemplateCommand.new.create(name)
+    end
+
+    desc "template:list", "List all templates"
+    def template_list
+      TemplateCommand.new.list
+    end
+
+    desc "template:show NAME", "Show details of a specific template"
+    def template_show(name)
+      TemplateCommand.new.show(name)
+    end
+
+    desc "template:delete NAME", "Delete a template"
+    def template_delete(name)
+      TemplateCommand.new.delete(name)
+    end
+
+    desc "template:use NAME NOTEBOOK", "Create a task from a template in the specified notebook"
+    def template_use(name, notebook)
+      TemplateCommand.new.use(name, notebook)
+    end
+
+    # Register AI Assistant subcommand with colon format
+    desc "ai:ask PROMPT", "Ask the AI assistant to perform tasks using natural language"
+    method_option :api_key, type: :string, desc: "OpenAI API key"
+    method_option :verbose, type: :boolean, default: false, desc: "Show detailed response"
+    def ai_ask(*prompt_args)
+      prompt = prompt_args.join(" ")
+      ai_command.ask(prompt)
+    end
+
+    desc "ai:configure", "Configure the AI assistant settings"
+    def ai_configure
+      ai_command.configure
+    end
+
+    # Map commands to use colon format
+    map "notebook:list" => :notebook_list
+    map "ai:ask" => :ai_ask
+    map "ai:configure" => :ai_configure
+
+    # Task commands
+    desc "task:add [NOTEBOOK] [TITLE]", "Add a new task to a notebook"
     method_option :description, type: :string, desc: "Task description"
     method_option :due_date, type: :string, desc: "Due date (YYYY-MM-DD HH:MM)"
     method_option :priority, type: :string, desc: "Priority (high, medium, low)"
     method_option :tags, type: :string, desc: "Tags (comma-separated)"
-    def add(notebook_name, title)
+    def task_add(notebook_name, title)
       notebook = Notebook.find_by(name: notebook_name)
       unless notebook
         say "Notebook '#{notebook_name}' not found".red
@@ -244,11 +460,11 @@ module RubyTodo
         say "Priority: #{priority}" if priority
         say "Tags: #{tags}" if tags
       else
-        say "Error creating task: #{task.errors.full_messages.join(", ")}".red
+        say "Error adding task: #{task.errors.full_messages.join(", ")}".red
       end
     end
 
-    desc "task list [NOTEBOOK]", "List all tasks in a notebook"
+    desc "task:list [NOTEBOOK]", "List all tasks in a notebook"
     method_option :status, type: :string, desc: "Filter by status (todo, in_progress, done, archived)"
     method_option :priority, type: :string, desc: "Filter by priority (high, medium, low)"
     method_option :due_soon, type: :boolean, desc: "Show only tasks due soon (within 24 hours)"
@@ -297,7 +513,7 @@ module RubyTodo
       puts table.render(:ascii)
     end
 
-    desc "task show [NOTEBOOK] [TASK_ID]", "Show detailed information about a task"
+    desc "task:show [NOTEBOOK] [TASK_ID]", "Show detailed information about a task"
     def task_show(notebook_name, task_id)
       notebook = Notebook.find_by(name: notebook_name)
       unless notebook
@@ -323,14 +539,14 @@ module RubyTodo
       say "Updated: #{task.updated_at}"
     end
 
-    desc "task edit [NOTEBOOK] [TASK_ID]", "Edit an existing task"
+    desc "task:edit [NOTEBOOK] [TASK_ID]", "Edit an existing task"
     method_option :title, type: :string, desc: "New title"
     method_option :description, type: :string, desc: "New description"
     method_option :due_date, type: :string, desc: "New due date (YYYY-MM-DD HH:MM)"
     method_option :priority, type: :string, desc: "New priority (high, medium, low)"
     method_option :tags, type: :string, desc: "New tags (comma-separated)"
     method_option :status, type: :string, desc: "New status (todo, in_progress, done, archived)"
-    def edit(notebook_name, task_id)
+    def task_edit(notebook_name, task_id)
       notebook = Notebook.find_by(name: notebook_name)
       unless notebook
         say "Notebook '#{notebook_name}' not found".red
@@ -366,8 +582,8 @@ module RubyTodo
       end
     end
 
-    desc "task move [NOTEBOOK] [TASK_ID] [STATUS]", "Move a task to a different status"
-    def move(notebook_name, task_id, status)
+    desc "task:move [NOTEBOOK] [TASK_ID] [STATUS]", "Move a task to a different status"
+    def task_move(notebook_name, task_id, status)
       notebook = Notebook.find_by(name: notebook_name)
       unless notebook
         say "Notebook '#{notebook_name}' not found".red
@@ -387,8 +603,8 @@ module RubyTodo
       end
     end
 
-    desc "task delete [NOTEBOOK] [TASK_ID]", "Delete a task"
-    def delete(notebook_name, task_id)
+    desc "task:delete [NOTEBOOK] [TASK_ID]", "Delete a task"
+    def task_delete(notebook_name, task_id)
       notebook = Notebook.find_by(name: notebook_name)
       unless notebook
         say "Notebook '#{notebook_name}' not found".red
@@ -405,9 +621,9 @@ module RubyTodo
       say "Deleted task #{task_id}".green
     end
 
-    desc "task search [QUERY]", "Search for tasks across all notebooks"
+    desc "task:search [QUERY]", "Search for tasks across all notebooks"
     method_option :notebook, type: :string, desc: "Limit search to a specific notebook"
-    def search(query)
+    def task_search(query)
       notebooks = if options[:notebook]
                     [Notebook.find_by(name: options[:notebook])].compact
                   else
@@ -579,140 +795,70 @@ module RubyTodo
       end
     end
 
-    # Task-related command aliases
-    map "task:list" => "task_list"
-    map "task:show" => "task_show"
-    map "task:add" => "add"
-    map "task:edit" => "edit"
-    map "task:delete" => "delete"
-    map "task:move" => "move"
-    map "task:search" => "search"
-
-    private
-
-    def export_notebook(notebook)
-      {
-        "name" => notebook.name,
-        "created_at" => notebook.created_at,
-        "updated_at" => notebook.updated_at,
-        "tasks" => notebook.tasks.map { |task| task_to_hash(task) }
-      }
-    end
-
-    def export_all_notebooks(notebooks)
-      {
-        "notebooks" => notebooks.map { |notebook| export_notebook(notebook) }
-      }
-    end
-
-    def task_to_hash(task)
-      {
-        "title" => task.title,
-        "description" => task.description,
-        "status" => task.status,
-        "priority" => task.priority,
-        "tags" => task.tags,
-        "due_date" => task.due_date&.iso8601,
-        "created_at" => task.created_at&.iso8601,
-        "updated_at" => task.updated_at&.iso8601
-      }
-    end
-
-    def export_to_json(data, filename)
-      File.write(filename, JSON.pretty_generate(data))
-    end
-
-    def export_to_csv(data, filename)
-      require "csv"
-
-      CSV.open(filename, "wb") do |csv|
-        if data["notebooks"]
-          # Multiple notebooks export
-          csv << ["Notebook", "Task ID", "Title", "Description", "Status", "Priority", "Tags", "Due Date",
-                  "Created At", "Updated At"]
-
-          data["notebooks"].each do |notebook|
-            notebook_name = notebook["name"]
-            notebook["tasks"].each_with_index do |task, index|
-              csv << [
-                notebook_name,
-                index + 1,
-                task["title"],
-                task["description"],
-                task["status"],
-                task["priority"],
-                task["tags"],
-                task["due_date"],
-                task["created_at"],
-                task["updated_at"]
-              ]
-            end
-          end
-        else
-          # Single notebook export
-          csv << ["Task ID", "Title", "Description", "Status", "Priority", "Tags", "Due Date", "Created At",
-                  "Updated At"]
-
-          data["tasks"].each_with_index do |task, index|
-            csv << [
-              index + 1,
-              task["title"],
-              task["description"],
-              task["status"],
-              task["priority"],
-              task["tags"],
-              task["due_date"],
-              task["created_at"],
-              task["updated_at"]
-            ]
-          end
-        end
+    # Map all commands to use colon format
+    map "task:add" => :task_add
+    map "task:list" => :task_list
+    map "task:show" => :task_show
+    map "task:edit" => :task_edit
+    map "task:move" => :task_move
+    map "task:delete" => :task_delete
+    map "task:search" => :task_search
+    map "notebook:create" => :notebook_create
+    map "notebook:list" => :notebook_list
+    map "notebook:set_default" => :notebook_set_default
+    map "template:create" => :template_create
+    map "template:list" => :template_list
+    map "template:show" => :template_show
+    map "template:delete" => :template_delete
+    map "template:use" => :template_use
+
+    # Remove old command mappings
+    no_commands do
+      def self.remove_old_commands
+        remove_command :create
+        remove_command :list
+        remove_command :show
+        remove_command :delete
+        remove_command :use
+        remove_command :ai
       end
     end
 
-    def import_tasks(notebook, tasks_data)
-      count = 0
-
-      tasks_data.each do |task_data|
-        # Convert ISO8601 string to Time object
-        due_date = Time.parse(task_data["due_date"]) if task_data["due_date"]
-
-        task = Task.create(
-          notebook: notebook,
-          title: task_data["title"],
-          description: task_data["description"],
-          status: task_data["status"] || "todo",
-          priority: task_data["priority"],
-          tags: task_data["tags"],
-          due_date: due_date
-        )
-
-        count += 1 if task.persisted?
+    remove_old_commands
+
+    # Override the help command to show only colon-formatted commands
+    def help(command = nil)
+      if command.nil?
+        puts "Commands:"
+        puts "  ruby_todo ai:ask PROMPT                           # Ask the AI assistant to perform tasks"
+        puts "  ruby_todo ai:configure                           # Configure the AI assistant settings"
+        puts "  ruby_todo notebook:create NAME                   # Create a new notebook"
+        puts "  ruby_todo notebook:list                         # List all notebooks"
+        puts "  ruby_todo notebook:set_default NAME             # Set a notebook as the default"
+        puts "  ruby_todo task:add [NOTEBOOK] [TITLE]           # Add a new task to a notebook"
+        puts "  ruby_todo task:list [NOTEBOOK]                  # List all tasks in a notebook"
+        puts "  ruby_todo task:show [NOTEBOOK] [TASK_ID]        # Show task details"
+        puts "  ruby_todo task:edit [NOTEBOOK] [TASK_ID]        # Edit a task"
+        puts "  ruby_todo task:move [NOTEBOOK] [TASK_ID] STATUS # Move a task to a different status"
+        puts "  ruby_todo task:delete [NOTEBOOK] [TASK_ID]      # Delete a task"
+        puts "  ruby_todo task:search [QUERY]                   # Search for tasks"
+        puts "  ruby_todo template:create NAME                  # Create a task template"
+        puts "  ruby_todo template:list                        # List all templates"
+        puts "  ruby_todo template:show NAME                   # Show template details"
+        puts "  ruby_todo template:delete NAME                 # Delete a template"
+        puts "  ruby_todo template:use NAME NOTEBOOK          # Use a template"
+        puts "  ruby_todo export [NOTEBOOK] [FILENAME]         # Export tasks"
+        puts "  ruby_todo import [FILENAME]                    # Import tasks"
+        puts "  ruby_todo init                                # Initialize todo list"
+        puts "  ruby_todo version                             # Show version"
+        puts "\nOptions:"
+        puts "  [--notebook=NOTEBOOK]  # Specify the notebook to use"
+      else
+        super
       end
-
-      count
     end
 
-    def import_all_notebooks(data)
-      results = { notebooks: 0, tasks: 0 }
-
-      data["notebooks"].each do |notebook_data|
-        notebook_name = notebook_data["name"]
-        notebook = Notebook.find_by(name: notebook_name)
-
-        unless notebook
-          notebook = Notebook.create(name: notebook_name)
-          results[:notebooks] += 1 if notebook.persisted?
-        end
-
-        if notebook.persisted?
-          tasks_count = import_tasks(notebook, notebook_data["tasks"])
-          results[:tasks] += tasks_count
-        end
-      end
-
-      results
-    end
+    private
 
     def display_notebook_stats(notebook)
       stats = notebook.statistics
@@ -856,5 +1002,9 @@ module RubyTodo
 
       text.length > length ? "#{text[0...length]}..." : text
     end
+
+    def ai_command
+      @ai_command ||= AIAssistantCommand.new
+    end
   end
 end