dsu 1.2.1 → 2.0.0.alpha.1

data/lib/dsu/subcommands/base_subcommand.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+require_relative '../base_cli'
+require_relative '../support/ask'
+require_relative '../support/subcommand_help_colorizeable'
+
+module Dsu
+  module Subcommands
+    class BaseSubcommand < Dsu::BaseCLI
+      include Support::Ask
+      include Support::SubcommandHelpColorizable
+    end
+  end
+end

data/lib/dsu/subcommands/config.rb

@@ -1,13 +1,15 @@
 # frozen_string_literal: true
 
-require 'thor'
-require_relative '../support/configuration'
+require_relative '../models/configuration'
+require_relative '../views/configuration/show'
+require_relative '../views/shared/model_errors'
+require_relative '../views/shared/success'
+require_relative '../views/shared/warning'
+require_relative 'base_subcommand'
 
 module Dsu
   module Subcommands
-    class Config < ::Thor
-      include Dsu::Support::Configuration
-
+    class Config < BaseSubcommand
       default_command :help
 
       class << self
@@ -19,59 +21,117 @@ module Dsu
       desc 'info', 'Displays information about this gem configuration'
       long_desc <<-LONG_DESC
         NAME
-        \x5
+
         `dsu config info` -- Displays information about this gem configuration.
 
         SYNOPSIS
-        \x5
+
         dsu config info
       LONG_DESC
       def info
-        print_config_file
+        configuration = Models::Configuration.new
+        Views::Configuration::Show.new(config: configuration).call
       end
 
-      desc 'init', 'Creates and initializes a .dsu file in your home folder'
-      long_desc <<-LONG_DESC
-        NAME
+      if ENV['DEV_ENV']
+        desc 'init', 'Creates and initializes a .dsu file in your home folder'
+        long_desc <<-LONG_DESC
+          NAME
 
-        `dsu config init` -- will create and initialize a .dsu file ("#{Dsu::Support::FolderLocations.root_folder}/.dsu") that you may edit. Otherwise, the default configuration will be used.
+          `dsu config init` -- will create and initialize a .dsu file ("#{Dsu::Support::Fileable.root_folder}/.dsu") that you may edit. Otherwise, the default configuration will be used.
 
-        SYNOPSIS
+          SYNOPSIS
 
-        dsu config init
+          dsu config init
 
-        CONFIGURATION FILE OPTIONS
+          CONFIGURATION FILE OPTIONS
 
-        The following configuration file options are available:
+          The following configuration file options are available:
 
-        editor:
+          version:
 
-        The default editor to use when editing entry groups if the EDITOR environment variable on your system is not set. The default is 'nano'. You'll need to change the default editor on Windows systems.
+          The configuration version - DO NOT ALTER THIS VALUE!
 
-        entries_display_order:
+          editor:
 
-        The order by which entries will be displayed, 'asc' or 'desc' (ascending or descending, respectively).
+          The default editor to use when editing entry groups if the EDITOR environment variable on your system is not set. The default is 'nano'. You'll need to change the default editor on Windows systems.
 
-        Default: 'desc'
+          entries_display_order:
 
-        entries_file_name:
+          The order by which entries will be displayed, :asc or :desc (ascending or descending, respectively).
 
-        The entries file name format. It is recommended that you do not change this. The file name must include `%Y`, `%m` and `%d` `Time` formatting specifiers to make sure the file name is unique and able to be located by `dsu` functions. For example, the default file name is `%Y-%m-%d.json`; however, something like `%m-%d-%Y.json` or `entry-group-%m-%d-%Y.json` would work as well.
+          Default: :desc
 
-        ATTENTION: Please keep in mind that if you change this value `dsu` will not recognize entry files using a different format. You would (at this time), have to manually rename any existing entry file names to the new format.
+          entries_file_name:
 
-        Default: '%Y-%m-%d.json'
+          The entries file name format. It is recommended that you do not change this. The file name must include `%Y`, `%m` and `%d` `Time` formatting specifiers to make sure the file name is unique and able to be located by `dsu` functions. For example, the default file name is `%Y-%m-%d.json`; however, something like `%m-%d-%Y.json` or `entry-group-%m-%d-%Y.json` would work as well.
 
-        entries_folder:
+          ATTENTION: Please keep in mind that if you change this value `dsu` will not recognize entry files using a different format. You would (at this time), have to manually rename any existing entry file names to the new format.
 
-        This is the folder where `dsu` stores entry files. You may change this to anything you want. `dsu` will create this folder for you, as long as your system's write permissions allow this.
+          Default: '%Y-%m-%d.json'
 
-        ATTENTION: Please keep in mind that if you change this value `dsu` will not be able to find entry files in any previous folder. You would (at this time), have to manually mode any existing entry files to this new folder.
+          entries_folder:
 
-        Default: "'#{Dsu::Support::FolderLocations.root_folder}/dsu/entries'"
-      LONG_DESC
-      def init
-        create_config_file!
+          This is the folder where `dsu` stores entry files. You may change this to anything you want. `dsu` will create this folder for you, as long as your system's write permissions allow this.
+
+          ATTENTION: Please keep in mind that if you change this value `dsu` will not be able to find entry files in any previous folder. You would (at this time), have to manually mode any existing entry files to this new folder.
+
+          Default: "'#{Dsu::Support::Fileable.root_folder}/dsu/entries'"
+        LONG_DESC
+        def init
+          exit 1 if configuration_errors_or_wanings?
+
+          Models::Configuration.default.tap do |configuration|
+            configuration.save!
+            messages = ["Configuration file (#{Models::Configuration.config_file}) created."]
+            Views::Shared::Success.new(messages: messages).render
+            Views::Configuration::Show.new(config: configuration).render
+          end
+        end
+
+        desc 'delete', 'Deletes the configuration file'
+        long_desc <<-LONG_DESC
+          NAME
+
+          `dsu config delete` -- Deletes the configuration.
+
+          SYNOPSIS
+
+          dsu config delete
+
+          NOTES
+
+          Deleting the dsu configuration file will simply cause dsu to use the default configuration options (`Dsu::Models::Configuration::DEFAULT_CONFIGURATION`).
+        LONG_DESC
+        def delete
+          unless Models::Configuration.exist?
+            messages = ["Configuration file (#{Models::Configuration.config_file}) does not exist."]
+            Views::Shared::Warning.new(messages: messages).render
+            exit 1
+          end
+          Models::Configuration.delete!
+          messages = ["Configuration file (#{Models::Configuration.config_file}) deleted."]
+          Views::Shared::Success.new(messages: messages).render
+        end
+      end
+
+      private
+
+      def configuration_errors_or_wanings?
+        if Models::Configuration.exist?
+          messages = ["Configuration file (#{Models::Configuration.config_file}) already exists"]
+          Views::Shared::Warning.new(messages: messages).render
+        elsif !Dir.exist?(Models::Configuration.config_folder)
+          messages = ["Destination folder for configuration file (#{Models::Configuration.config_folder}) does not exist"] # rubocop:disable Layout/LineLength
+          Views::Shared::Error.new(messages: messages).render
+        else
+          configuration = Models::Configuration.default
+          return false if configuration.valid?
+
+          Views::Shared::ModelErrors.new(model: configuration).render
+        end
+
+        true
       end
     end
   end

data/lib/dsu/subcommands/edit.rb

@@ -1,12 +1,12 @@
 # frozen_string_literal: true
 
-require_relative '../base_cli'
+require_relative 'base_subcommand'
 require_relative '../models/entry_group'
 require_relative '../views/entry_group/show'
 
 module Dsu
   module Subcommands
-    class Edit < Dsu::BaseCLI
+    class Edit < BaseSubcommand
       map %w[d] => :date
       map %w[n] => :today
       map %w[t] => :tomorrow
@@ -53,7 +53,7 @@ module Dsu
         entry_group = Models::EntryGroup.edit(time: Time.parse(date))
         Views::EntryGroup::Show.new(entry_group: entry_group).render
       rescue ArgumentError => e
-        say "Error: #{e.message}", ERROR
+        puts apply_theme("Error: #{e.message}", theme_color: color_theme.error)
         exit 1
       end
     end

data/lib/dsu/subcommands/list.rb

@@ -1,13 +1,16 @@
 # frozen_string_literal: true
 
-require_relative '../base_cli'
 require_relative '../support/command_options/dsu_times'
+require_relative '../support/command_options/time_mneumonic'
 require_relative '../support/time_formatable'
+require_relative '../views/entry_group/shared/no_entries_to_display'
+require_relative '../views/shared/error'
+require_relative 'base_subcommand'
 
 module Dsu
   module Subcommands
-    class List < Dsu::BaseCLI
-      include Support::CommandOptions::DsuTimes
+    class List < BaseSubcommand
+      include Support::CommandOptions::TimeMneumonic
       include Support::TimeFormatable
 
       map %w[d] => :date
@@ -21,10 +24,11 @@ module Dsu
       long_desc <<-LONG_DESC
         Displays the DSU entries for today. This command has no options.
       LONG_DESC
+      option :include_all, type: :boolean, aliases: '-a', desc: 'Include dates that have no DSU entries'
       def today
         time = Time.now
-        times = sorted_dsu_times_for(times: [time, time.yesterday])
-        view_list_for(times: times)
+        times = sorted_dsu_times_for(times: [time.yesterday, time])
+        view_list_for(times: times, options: options)
       end
 
       desc 'tomorrow, t',
@@ -32,10 +36,11 @@ module Dsu
       long_desc <<-LONG_DESC
         Displays the DSU entries for tomorrow. This command has no options.
       LONG_DESC
+      option :include_all, type: :boolean, aliases: '-a', desc: 'Include dates that have no DSU entries'
       def tomorrow
         time = Time.now
-        times = sorted_dsu_times_for(times: [time.tomorrow, time])
-        view_list_for(times: times)
+        times = sorted_dsu_times_for(times: [time, time.tomorrow])
+        view_list_for(times: times, options: options)
       end
 
       desc 'yesterday, y',
@@ -43,10 +48,11 @@ module Dsu
       long_desc <<-LONG_DESC
         Displays the DSU entries for yesterday. This command has no options.
       LONG_DESC
+      option :include_all, type: :boolean, aliases: '-a', desc: 'Include dates that have no DSU entries'
       def yesterday
         time = Time.now
         times = sorted_dsu_times_for(times: [time.yesterday, time.yesterday.yesterday])
-        view_list_for(times: times)
+        view_list_for(times: times, options: options)
       end
 
       desc 'date, d DATE|MNEUMONIC',
@@ -54,130 +60,101 @@ module Dsu
       long_desc <<-LONG_DESC
         Displays the DSU entries for the DATE or MNEUMONIC provided.
 
-        \x5
         #{date_option_description}
 
-        \x5
         #{mneumonic_option_description}
       LONG_DESC
-      def date(param)
-        time = if time_mneumonic?(param)
-          time_from_mneumonic(command_option: param)
+      option :include_all, type: :boolean, aliases: '-a', desc: 'Include dates that have no DSU entries'
+      def date(date_or_mneumonic)
+        time = if time_mneumonic?(date_or_mneumonic)
+          time_from_mneumonic(command_option: date_or_mneumonic)
         else
-          Time.parse(param)
+          Time.parse(date_or_mneumonic)
         end
         times = sorted_dsu_times_for(times: [time, time.yesterday])
-        view_list_for(times: times)
+        view_list_for(times: times, options: options)
       rescue ArgumentError => e
-        say "Error: #{e.message}", ERROR
-        exit 1
+        Views::Shared::Error.new(messages: e.message).render
       end
 
-      desc 'dates, dd OPTIONS',
+      desc 'dates|dd OPTIONS',
         'Displays the DSU entries for the OPTIONS provided'
       long_desc <<~LONG_DESC
         NAME
-        \x5
-        `dsu dates|dd OPTIONS` -- will display the DSU entries for the OPTIONS provided.
+
+        $ dsu dates|dd OPTIONS -- will display the DSU entries for the OPTIONS provided.
 
         SYNOPSIS
-        \x5
-        `dsu dates|dd OPTIONS`
+
+        $ dsu dates|dd OPTIONS
 
         OPTIONS
-        \x5
+
         -a|--include-all true|false: If true, all DSU dates within the specified range will be displayed. If false, DSU dates between the first and last DSU dates that have NO entries will NOT be displayed.. The default is taken from the dsu configuration setting :include_all, see `dsu config info`.
 
-        \x5
         -f|--from DATE|MNEMONIC: The DATE or MNEUMONIC that represents the start of the range of DSU dates to display. If a relative mneumonic is used (+/-n, e.g +1, -1, etc.), the date calculated will be relative to the current date (e.g. `<MNEUMONIC>.to_i.days.from_now(Time.now)`).
 
-        \x5
         -t|--to DATE|MNEMONIC: The DATE or MNEUMONIC that represents the end of the range of DSU dates to display. If a relative mneumonic is used (+/-n, e.g +1, -1, etc.), the date calculated will be relative to the date that resulting from the `--from` option date calculation.
 
-        \x5
         #{date_option_description}
 
-        \x5
         #{mneumonic_option_description}
 
-        \x5
         EXAMPLES
 
-        \x5
         NOTE: All examples are subject to the `--include-all` option.
 
-        \x5
-        # The below will display the DSU entries for the range of dates from 1/1 to 1/4.
-        \x5`dsu list dates --from 1/1 --to +3`
-
-        \x5
-        # will display the DSU entries for the range of dates from 1/2 to 1/5.
-        \x5`dsu list dates --from 1/5 --to -3`
-
-        \x5
-        # (assuming "today" is 1/10) will display the DSU entries for the last week 1/10 to 1/3.
-        \x5`dsu list dates --from today --to -7`
-
-        \x5
-        # (assuming "today" is 5/23) will display the DSU entries for the last week 5/16 to 5/22.
-        #
-        # This example simply illustrates the fact that you can use relative mneumonics for
-        # both `--from` and `--to` options; this doesn't mean you should do so...
-        #
-        # While you can use relative mneumonics for both `--from` and `--to` options,
-        # there is always a more intuitive way. The below example basically lists one week
-        # of DSU entries back 1 week from yesterday's date:
-        \x5`dsu list dates --from -7 --to +6`
-
-        \x5
-        # This can MUCH easily be accomplished by simply using the `yesterday` mneumonic:
-        # (assuming "today" is 5/23) will display the DSU entries back 1 week from yesterday's date 5/16 to 5/22.
-        \x5`dsu list dates --from yesterday --to -6`
+        The below will display the DSU entries for the range of dates from 1/1 to 1/4:
+
+        $ dsu list dates --from 1/1 --to +3
+
+        This will display the DSU entries for the range of dates from 1/2 to 1/5:
+
+        $ dsu list dates --from 1/5 --to -3
+
+        This (assuming "today" is 1/10) will display the DSU entries for the last week 1/10 to 1/3:
+
+        $ dsu list dates --from today --to -7
+
+        This (assuming "today" is 5/23) will display the DSU entries for the last week 5/16 to 5/22.
+        This example simply illustrates the fact that you can use relative mneumonics for
+        both `--from` and `--to` options; this doesn't mean you should do so...
+
+        While you can use relative mneumonics for both `--from` and `--to` options,
+        there is always a more intuitive way. The below example basically lists one week
+        of DSU entries back 1 week from yesterday's date:
+
+        $ dsu list dates --from -7 --to +6
+
+        The above can be accomplished MUCH easier by simply using the `yesterday` mneumonic...
+
+        This (assuming "today" is 5/23) will display the DSU entries back 1 week from yesterday's date 5/16 to 5/22:
+
+        $ dsu list dates --from yesterday --to -6
       LONG_DESC
       # -f, --from FROM [DATE|MNEMONIC] (e.g. -f, --from 1/1[/yyy]|n|t|y|today|tomorrow|yesterday)
-      option :from, type: :string, aliases: '-f', banner: 'DATE|MNEMONIC'
+      option :from, type: :string, required: true, aliases: '-f', banner: 'DATE|MNEMONIC'
       # -t, --to TO [DATE|MNEMONIC] (e.g. -t, --to 1/1[/yyy]|n|t|y|today|tomorrow|yesterday)
-      option :to, type: :string, aliases: '-t', banner: 'DATE|MNEMONIC'
-
+      option :to, type: :string, required: true, aliases: '-t', banner: 'DATE|MNEMONIC'
       # Include dates that have no DSU entries.
-      option :include_all, type: :boolean, aliases: '-a'
+      option :include_all, type: :boolean, aliases: '-a', desc: 'Include dates that have no DSU entries'
       def dates
-        options = configuration.merge(self.options)
-        times = dsu_times_from!(from_command_option: options[:from], to_command_option: options[:to])
-        # Note special sort here, unlike the other commands where rules for
-        # displaying DSU entries are applied; this is more of a list command.
-        times = times_sort(times: times, entries_display_order: entries_display_order)
-        view_entry_groups(times: times, options: options) do |total_entry_groups|
-          nothing_to_display_banner_for(times) if total_entry_groups.zero?
+        options = configuration.to_h.merge(self.options).with_indifferent_access
+        times, errors = Support::CommandOptions::DsuTimes.dsu_times_for(from_option: options[:from], to_option: options[:to]) # rubocop:disable Layout/LineLength
+        if errors.any?
+          Views::Shared::Error.new(messages: errors).render
+          return
         end
-      rescue ArgumentError => e
-        say "Error: #{e.message}", ERROR
-        exit 1
-      end
-
-      private
-
-      def nothing_to_display_banner_for(entry_group_times)
-        entry_group_times.sort!
-        time_range = "#{formatted_time(time: entry_group_times.first)} " \
-                     "through #{formatted_time(time: entry_group_times.last)}"
-        say "(nothing to display for #{time_range})", INFO
-      end
 
-      # This method will unconditionally display the FIRST and LAST entry groups
-      # associated with the times provided by the <times> argument. All other
-      # entry groups will be conditionally displayed based on the :include_all
-      # value in the <options> argument.
-      def view_list_for(times:)
-        options = configuration.merge(self.options)
-        times_first_and_last = [times.first, times.last]
-        times.each do |time|
-          view_options = options.dup
-          view_options[:include_all] = true if times_first_and_last.include?(time)
-          view_entry_group(time: time, options: view_options) do
-            puts
-          end
+        # NOTE: special sort here, unlike the other commands where rules for
+        # displaying DSU entries are applied; this is more of a list command.
+        times = times_sort(times: times, entries_display_order: options[:entries_display_order])
+        view_entry_groups(times: times, options: options) do |total_entry_groups, _total_entry_groups_not_shown|
+          # nothing_to_display_banner_for(times) if total_entry_groups.zero?
+          Views::EntryGroup::Shared::NoEntriesToDisplay.new(times: times, options: options) if total_entry_groups.zero?
         end
+      rescue ArgumentError => e
+        Views::Shared::Error.new(messages: e.message).render
       end
     end
   end

data/lib/dsu/subcommands/theme.rb

@@ -0,0 +1,159 @@
+# frozen_string_literal: true
+
+require_relative '../env'
+require_relative '../support/fileable'
+require_relative '../views/color_theme/index'
+require_relative '../views/color_theme/show'
+require_relative '../views/shared/error'
+require_relative '../views/shared/info'
+require_relative 'base_subcommand'
+
+module Dsu
+  module Subcommands
+    class Theme < BaseSubcommand
+      map %w[c] => :create if Dsu.env.test?
+      map %w[d] => :delete if Dsu.env.test?
+      map %w[l] => :list
+      map %w[s] => :show
+      map %w[u] => :use
+
+      if Dsu.env.test?
+        desc 'create THEME_NAME [OPTIONS]',
+          'Creates a dsu color theme named THEME_NAME.'
+        long_desc <<-LONG_DESC
+        Create a dsu color theme named THEME_NAME in the #{Support::Fileable.themes_folder} folder.
+
+        SYNOPSIS
+
+        `dsu create THEME_NAME [-d|--description DESCRIPTION]`
+
+        OPTIONS:
+
+        -d|--description DESCRIPTION: Creates the dsu color theme with having DESCRIPTION as the color theme description.
+
+        DESCRIPTION:
+
+        Must be be between 2 and 256 characters (inclusive) in length.
+        LONG_DESC
+        option :description, type: :string, aliases: '-d', banner: 'DESCRIPTION'
+        option :prompts, type: :hash, default: {}, hide: true, aliases: '-p'
+        def create(theme_name)
+          if Models::ColorTheme.exist?(theme_name: theme_name)
+            Views::Shared::Error.new(messages: "Color theme \"#{theme_name}\" already exists.").render
+            return false
+          end
+          prompt = color_theme.prompt_with_options(prompt: "Create color theme \"#{theme_name}\"?", options: %w[y N])
+          if yes?(prompt, options: options)
+            theme_hash = Models::ColorTheme::DEFAULT_THEME.dup
+            theme_hash[:description] = options[:description] || "#{theme_name.capitalize} color theme"
+            Models::ColorTheme.new(theme_name: theme_name, theme_hash: theme_hash).save!
+            Views::Shared::Info.new(messages: "\nCreated color theme \"#{theme_name}\".").render
+            true
+          else
+            Views::Shared::Info.new(messages: "\nCanceled.").render
+            false
+          end
+        end
+
+        desc 'delete THEME_NAME',
+          'Deletes the existing dsu color theme THEME_NAME.'
+        long_desc <<-LONG_DESC
+        NAME
+
+        `dsu delete [THEME_NAME]` -- will delete the dsu color theme named THEME_NAME located in the #{Support::Fileable.themes_folder} folder.
+        LONG_DESC
+        option :prompts, type: :hash, default: {}, hide: true, aliases: '-p'
+        def delete(theme_name)
+          display_dsu_header
+
+          if theme_name == Models::ColorTheme::DEFAULT_THEME_NAME
+            display_dsu_header
+            Views::Shared::Error.new(messages: "Color theme \"#{theme_name}\" cannot be deleted.").render
+            return
+          end
+
+          unless Models::ColorTheme.exist?(theme_name: theme_name)
+            Views::Shared::Error.new(messages: "Color theme \"#{theme_name}\" does not exist.").render
+            return
+          end
+
+          prompt = color_theme.prompt_with_options(prompt: "Delete color theme \"#{theme_name}\"?",
+            options: %w[y N])
+          if yes?(prompt, options: options)
+            Models::ColorTheme.delete!(theme_name: theme_name)
+            Views::Shared::Info.new(messages: "\nDeleted color theme \"#{theme_name}\".").render
+          else
+            Views::Shared::Info.new(messages: "\nCanceled.").render
+          end
+        end
+      end
+
+      desc 'list',
+        'Lists the available dsu color themes.'
+      long_desc <<-LONG_DESC
+      NAME
+
+      `dsu list` -- lists the available dsu color themes located in the #{Support::Fileable.themes_folder} folder.
+      LONG_DESC
+      def list
+        Views::ColorTheme::Index.new.render
+      end
+
+      desc 'use THEME_NAME',
+        'Sets THEME_NAME as the current DSU color theme.'
+      long_desc <<-LONG_DESC
+      NAME
+
+      `dsu theme use [THEME_NAME]` -- sets the dsu color theme to THEME_NAME.
+
+      SYNOPSIS
+
+      If THEME_NAME is not provided, the default theme will be used.
+      If THEME_NAME does not exist, you will be given the option to create a new theme.
+
+      LONG_DESC
+      option :prompts, type: :hash, default: {}, hide: true, aliases: '-p'
+      def use(theme_name = Models::ColorTheme::DEFAULT_THEME_NAME)
+        if Dsu.env.test? && !Models::ColorTheme.exist?(theme_name: theme_name)
+          display_dsu_header
+          return unless create(theme_name)
+        end
+
+        unless Models::ColorTheme.exist?(theme_name: theme_name)
+          display_dsu_header
+          Views::Shared::Error.new(messages: "Color theme \"#{theme_name}\" does not exist.").render
+          return
+        end
+
+        configuration.theme_name = theme_name
+        configuration.save!
+        # We need to display the header after the theme is updated so that it is displayed in the
+        # correct theme color.
+        display_dsu_header
+        Views::Shared::Info.new(messages: "Using color theme \"#{theme_name}\".").render
+      end
+
+      desc 'show THEME_NAME',
+        'Displays the dsu color theme.'
+      long_desc <<-LONG_DESC
+      NAME
+
+      `dsu show THEME_NAME` -- displays the dsu color theme for THEME_NAME.
+      LONG_DESC
+      def show(theme_name = configuration.theme_name)
+        if Dsu::Models::ColorTheme.exist?(theme_name: theme_name)
+          Views::ColorTheme::Show.new(theme_name: theme_name).render
+          return
+        end
+
+        Views::Shared::Error.new(messages: "Color theme \"#{theme_name}\" does not exist.").render
+      end
+
+      private
+
+      def display_dsu_header
+        self.class.display_dsu_header
+      end
+    end
+  end
+end

data/lib/dsu/support/ask.rb

@@ -5,34 +5,29 @@ require 'thor'
 module Dsu
   module Support
     module Ask
-      ASK_YES = %w[y yes].freeze
-      # ASK_NO = %w[n no].freeze
-      # ASK_CANCEL = %w[c cancel].freeze
-      # ASK_YES_NO_CANCEL = ASK_YES.concat(ASK_NO).concat(ASK_CANCEL).freeze
-
       def ask(prompt)
         options = {}
         Thor::LineEditor.readline(prompt, options)
       end
 
-      def yes?(prompt, color = nil)
-        Thor::Base.shell.new.yes?(prompt, color)
-      end
+      def yes?(prompt, options: {})
+        auto_prompt = auto_prompt(prompt, options)
 
-      # def no?(prompt)
-      #   ask_with(prompt: prompt, values: ASK_NO)
-      # end
+        return auto_prompt unless auto_prompt.nil?
 
-      # def yes_no_cancel(prompt)
-      #   ask_with(prompt: prompt, values: ASK_YES_NO_CANCEL)
-      # end
+        Thor::Base.shell.new.yes?(prompt)
+      end
 
-      # private
+      private
 
-      # def ask_with(prompt:, values:)
-      #   p "#{prompt}"
-      #   values.include? STDIN.gets.chomp
-      # end
+      def auto_prompt(prompt, options)
+        prompt = Utils.strip_escapes(prompt)
+        @auto_prompt ||= begin
+          value = options.dig('prompts', prompt)
+          value = (value == 'true' unless value.nil?)
+          value
+        end
+      end
     end
   end
 end