Class: RockBooks::CommandLineInterface

Inherits:
Object
  • Object
show all
Extended by:
Forwardable
Includes:
JournalEntryFilters
Defined in:
lib/rock_books/cmd_line/command_line_interface.rb

Defined Under Namespace

Classes: BadCommandError, Command

Constant Summary collapse

PROJECT_URL =

For conveniently finding the project on Github from the shell

'https://github.com/keithrbennett/rock_books'
HELP_TEXT =

Help text to be used when requested by ‘h’ command, in case of unrecognized or nonexistent command, etc.

"
Command Line Switches:                    [rock-books version #{RockBooks::VERSION} at https://github.com/keithrbennett/rock_books]

-i   input directory specification, default: '#{DEFAULT_INPUT_DIR}'
-o   output (reports) directory specification, default: '#{DEFAULT_OUTPUT_DIR}'
-r   receipts directory, default: '#{DEFAULT_RECEIPT_DIR}'
-s   run in shell mode

Commands:

rec[eipts]                - receipts: a/:a all, m/:m missing, e/:e existing
rep[orts]                 - return an OpenStruct containing all reports (interactive shell mode only)
d[isplay_reports]         - display all reports on stdout
w[rite_reports]           - write all reports to the output directory (see -o option)
c[hart_of_accounts]       - chart of accounts
h[elp]                    - prints this help
jo[urnals]                - list of the journals' short names
proj[ect_page]            - open the RockBooks Github project page in a browser
rel[oad_data]             - reload data from input files
q[uit]                    - exits this program (interactive shell mode only) (see also 'x')
x[it]                     - exits this program (interactive shell mode only) (see also 'q')

When in interactive shell mode:
* use quotes for string parameters such as method names.
* for pry commands, use prefix `%`.
* you can use the global variable $filter to filter reports

"

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from JournalEntryFilters

account_code, all, any, date_in_range, date_on_or_after, date_on_or_before, day, filter, month, none, null_filter, to_date, year

Constructor Details

#initialize(run_options) ⇒ CommandLineInterface 



71
72
73
74
75
76
77
78
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 71

def initialize(run_options)
  @run_options = run_options
  @interactive_mode = !!(run_options.interactive_mode)
  @verbose_mode = run_options.verbose

  validate_run_options(run_options)
  # book_set is set with a lazy initializer
end

Dynamic Method Handling

This class handles dynamic methods through the method_missing method

#method_missing(method_name, *method_args) ⇒ Object

For use by the shell when the user types the DSL commands



208
209
210
211
212
213
214
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 208

def method_missing(method_name, *method_args)
  attempt_command_action(method_name.to_s, *method_args) do
    puts(%Q{"#{method_name}" is not a valid command or option. } \
      << 'If you intend for this to be a string literal, ' \
      << 'use quotes or %q{}/%Q{}.')
  end
end

Instance Attribute Details

#book_set ⇒ Object (readonly)

Returns the value of attribute book_set.



18
19
20
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 18

def book_set
  @book_set
end

#interactive_mode ⇒ Object (readonly)

Returns the value of attribute interactive_mode.



18
19
20
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 18

def interactive_mode
  @interactive_mode
end

#run_options ⇒ Object (readonly)

Returns the value of attribute run_options.



18
19
20
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 18

def run_options
  @run_options
end

#verbose_mode ⇒ Object (readonly)

Returns the value of attribute verbose_mode.



18
19
20
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 18

def verbose_mode
  @verbose_mode
end

Instance Method Details

#attempt_command_action(command, *args, &error_handler_block) ⇒ Object

Look up the command name and, if found, run it. If not, execute the passed block.



186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 186

def attempt_command_action(command, *args, &error_handler_block)
  no_command_specified = command.nil?
  command = 'help' if no_command_specified

  action = find_command_action(command)
  result = nil

  if action
    result = action.(*args)
  else
    error_handler_block.call
    nil
  end

  if no_command_specified
    puts enclose_in_hyphen_lines('! No operations specified !')
  end
  result
end

#call ⇒ Object 



399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 399

def call
  begin
    # By this time, the Main class has removed the command line options, and all that is left
    # in ARGV is the commands and their options.
    if @interactive_mode
      run_shell
    else
      process_command_line
    end

  rescue BadCommandError => error
    separator_line = "! #{'-' * 75} !\n"
    puts '' << separator_line << error.to_s << "\n" << separator_line
    exit(-1)
  end
end

#cmd_c ⇒ Object 



241
242
243
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 241

def cmd_c
  puts chart_of_accounts.report_string
end

#cmd_d ⇒ Object 



293
294
295
296
297
298
299
300
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 293

def cmd_d
  book_set.all_reports($filter).each do |short_name, report_text|
    puts "#{short_name}:\n\n"
    puts report_text
    puts "\n\n\n"
  end
  nil
end

#cmd_h ⇒ Object 



246
247
248
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 246

def cmd_h
  print_help
end

#cmd_j ⇒ Object 



251
252
253
254
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 251

def cmd_j
  journal_names = book_set.journals.map(&:short_name)
  interactive_mode ? journal_names : ap(journal_names)
end

#cmd_proj ⇒ Object 



303
304
305
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 303

def cmd_proj
  `open https://github.com/keithrbennett/rock_books`
end

#cmd_rec(options) ⇒ Object 



308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 308

def cmd_rec(options)
  unless run_options.do_receipts
    raise Error.new("Receipt processing was requested but has been disabled with --no-receipts.")
  end

  missing, existing = book_set.missing_and_existing_receipts

  print_missing  = -> { puts "Missing Receipts:"; ap missing }
  print_existing = -> { puts "Existing Receipts:"; ap existing }

  case options.first.to_s
    when 'a'  # all
      if run_options.interactive_mode
        { missing: missing, existing: existing }
      else
         print_missing.(); print_existing.()
      end

    when 'm'
      run_options.interactive_mode ? missing : print_missing.()

    when 'e'
      run_options.interactive_mode ? existing : print_existing.()

  else
    message = "Invalid option for receipts. Must be 'a' for all, 'm' for missing, or 'e' for existing."
    if run_options.interactive_mode
      puts message
    else
      raise Error.new(message)
    end
  end
end

#cmd_rel ⇒ Object 



268
269
270
271
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 268

def cmd_rel
  reload_data
  nil
end

#cmd_rep ⇒ Object

All reports as Ruby objects; only makes sense in shell mode.



275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 275

def cmd_rep
  unless run_options.interactive_mode
    raise Error.new("Option 'all_reports' is only available in shell mode. Try 'display_reports' or 'write_reports'.")
  end

  os = OpenStruct.new(book_set.all_reports($filter))

  # add hash methods for convenience
  def os.keys; to_h.keys.map(&:to_s); end
  def os.values; to_h.values; end

  # to access as array, e.g. `a.at(1)`
  def os.at(index); self.public_send(keys[index]); end

  os
end

#cmd_w ⇒ Object 



342
343
344
345
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 342

def cmd_w
  book_set.all_reports_to_files(run_options.output_dir, $filter)
  nil
end

#cmd_x ⇒ Object 



348
349
350
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 348

def cmd_x
  quit
end

#commands ⇒ Object 



353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 353

def commands
  @commands_ ||= [
      Command.new('rec', 'receipts',          -> (*options)  { cmd_rec(options)  }),
      Command.new('rep', 'reports',           -> (*_options) { cmd_rep           }),
      Command.new('d',   'display_reports',   -> (*_options) { cmd_d             }),
      Command.new('w',   'write_reports',     -> (*_options) { cmd_w             }),
      Command.new('c',   'chart_of_accounts', -> (*_options) { cmd_c             }),
      Command.new('jo',  'journals',          -> (*_options) { cmd_j             }),
      Command.new('h',   'help',              -> (*_options) { cmd_h             }),
      Command.new('proj','project_page',      -> (*_options) { cmd_proj          }),
      Command.new('q',   'quit',              -> (*_options) { cmd_x             }),
      Command.new('rel', 'reload_data',       -> (*_options) { cmd_rel           }),
      Command.new('x',   'xit',               -> (*_options) { cmd_x             })
  ]
end

#enclose_in_hyphen_lines(string) ⇒ Object 



141
142
143
144
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 141

def enclose_in_hyphen_lines(string)
  hyphen_line = "#{'-' * 80}\n"
  hyphen_line + string + "\n" + hyphen_line
end

#find_command_action(command_string) ⇒ Object 



370
371
372
373
374
375
376
377
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 370

def find_command_action(command_string)
  result = commands.detect do |cmd|
    cmd.max_string.start_with?(command_string) \
  && \
  command_string.length >= cmd.min_string.length  # e.g. 'c' by itself should not work
  end
  result ? result.action : nil
end

#load_data ⇒ Object Also known as: reload_data 



262
263
264
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 262

def load_data
  @book_set = BookSetLoader.load(run_options)
end

#post_process(object) ⇒ Object

If a post-processor has been configured (e.g. YAML or JSON), use it.



381
382
383
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 381

def post_process(object)
  post_processor ? post_processor.(object) : object
end

#post_processor ⇒ Object 



386
387
388
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 386

def post_processor
  run_options.post_processor
end

#print_help ⇒ Object 



136
137
138
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 136

def print_help
  puts HELP_TEXT
end

#process_command_line ⇒ Object

Processes the command (ARGV) and any relevant options (ARGV).

CAUTION! In interactive mode, any strings entered (e.g. a network name) MUST be in a form that the Ruby interpreter will recognize as a string, i.e. single or double quotes, %q, %Q, etc. Otherwise it will assume it’s a method name and pass it to method_missing!



223
224
225
226
227
228
229
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 223

def process_command_line
  attempt_command_action(ARGV[0], *ARGV[1..-1]) do
    print_help
    raise BadCommandError.new(
        %Q{! Unrecognized command. Command was #{ARGV.first.inspect} and options were #{ARGV[1..-1].inspect}.})
  end
end

#quit ⇒ Object 



232
233
234
235
236
237
238
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 232

def quit
  if interactive_mode
    exit(0)
  else
    puts "This command can only be run in shell mode."
  end
end

#run_pry ⇒ Object

Pry will output the content of the method from which it was called. This small method exists solely to reduce the amount of pry’s output that is not needed here.



150
151
152
153
154
155
156
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 150

def run_pry
  binding.pry

  # the seemingly useless line below is needed to avoid pry's exiting
  # (see https://github.com/deivid-rodriguez/pry-byebug/issues/45)
  _a = nil
end

#run_shell ⇒ Object

Runs a pry session in the context of this object. Commands and options specified on the command line can also be specified in the shell.



161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 161

def run_shell
  begin
    require 'pry'
  rescue LoadError
    message = "The 'pry' gem and/or one of its prerequisites, required for running the shell, was not found." +
        " Please `gem install pry` or, if necessary, `sudo gem install pry`."
    raise Error.new(message)
  end

  print_help

  # Enable the line below if you have any problems with pry configuration being loaded
  # that is messing up this runtime use of pry:
  # Pry.config.should_load_rc = false

  # Strangely, this is the only thing I have found that successfully suppresses the
  # code context output, which is not useful here. Anyway, this will differentiate
  # a pry command from a DSL command, which _is_ useful here.
  Pry.config.command_prefix = '%'

  run_pry
end

#td(date_string) ⇒ Object

Easier than remembering and typing Date.iso8601.



394
395
396
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 394

def td(date_string)
  Date.iso8601(date_string)
end

#validate_run_options(options) ⇒ Object 



81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
 
# File 'lib/rock_books/cmd_line/command_line_interface.rb', line 81

def validate_run_options(options)

  return if options.suppress_command_line_validation

  validate_input_dir = -> do
    File.directory?(options.input_dir) ? nil : "Input directory '#{options.input_dir}' does not exist. "
  end

  validate_output_dir = -> do

    # We need to create the reports directory if it does not already exist.
    # mkdir_p silently returns if the directory already exists.
    begin
      FileUtils.mkdir_p(options.output_dir)
      nil
    rescue Errno::EACCES => error
      "Output directory '#{options.output_dir}' does not exist and could not be created. "
    end
  end

  validate_receipts_dir = -> do
    File.directory?(options.receipt_dir) ? nil : \
        "Receipts directory '#{options.receipt_dir}' does not exist. "
  end

  output = []
  output << validate_input_dir.()
  output << validate_output_dir.()
  if run_options.do_receipts
    output << validate_receipts_dir.()
  end

  output.compact!

  unless output.empty?
    message = <<~HEREDOC
    #{output.compact.join("\n")}

    Running this program assumes that you you have:

    * an input directory containing documents with your accounting data. 
      The default directory for this is #{DEFAULT_INPUT_DIR} and can be overridden
      with the -i/--input_dir option.

    * Unless receipt handling is disabled with the --no-receipts option,
      a directory where receipts can or will be stored.
      The default directory for this is #{DEFAULT_RECEIPT_DIR} and can be overridden
      with the -r/--receipt_dir option.
    
    HEREDOC
    raise Error.new(message)
  end
end