Module: Main::Program::InstanceMethods

Included in:
Main::Program
Defined in:
lib/main/program/instance_methods.rb

Instance Method Summary collapse

Instance Method Details

#abort(message = 'exit') ⇒ Object

Raises:

  • (SystemExit) 


207
208
209
 
# File 'lib/main/program/instance_methods.rb', line 207

def abort(message = 'exit')
  raise SystemExit.new(message)
end

#after_initialize ⇒ Object 



50
 
# File 'lib/main/program/instance_methods.rb', line 50

def after_initialize() :hook end

#after_parse_parameters ⇒ Object 



175
 
# File 'lib/main/program/instance_methods.rb', line 175

def after_parse_parameters() :hook end

#after_run ⇒ Object 



183
 
# File 'lib/main/program/instance_methods.rb', line 183

def after_run() :hook end

#before_initialize ⇒ Object 



43
 
# File 'lib/main/program/instance_methods.rb', line 43

def before_initialize() :hook end

#before_parse_parameters ⇒ Object 



164
 
# File 'lib/main/program/instance_methods.rb', line 164

def before_parse_parameters() :hook end

#before_run ⇒ Object 



179
 
# File 'lib/main/program/instance_methods.rb', line 179

def before_run() :hook end

#config(&block) ⇒ Object 



286
287
288
 
# File 'lib/main/program/instance_methods.rb', line 286

def config(&block)
  self.class.config(&block)
end

#daemon ⇒ Object 



298
299
300
 
# File 'lib/main/program/instance_methods.rb', line 298

def daemon
  @daemon ||= Main::Daemon.new(self)
end

#db(&block) ⇒ Object 



282
283
284
 
# File 'lib/main/program/instance_methods.rb', line 282

def db(&block)
  self.class.db(&block)
end

#fcall(object, method, *argv, &block) ⇒ Object 



244
245
246
247
248
249
250
251
252
253
254
 
# File 'lib/main/program/instance_methods.rb', line 244

def fcall(object, method, *argv, &block)
  method = object.method(method)
  arity = method.arity
  if arity >= 0
    argv = argv[0, arity]
  else
    arity = arity.abs - 1
    argv = argv[0, arity] + argv[arity .. -1]
  end
  method.call(*argv, &block)
end

#finalize ⇒ Object 



57
58
59
 
# File 'lib/main/program/instance_methods.rb', line 57

def finalize
  while((f = finalizers.pop)); f.call; end
end

#handle_exception(e) ⇒ Object 



211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
 
# File 'lib/main/program/instance_methods.rb', line 211

def handle_exception(e)
  if e.respond_to?(:error_handler_before)
    fcall(e, :error_handler_before, self)
  end

  if e.respond_to?(:error_handler_instead)
    fcall(e, :error_handler_instead, self)
  else
    if e.respond_to?(:status)
      exit_status(( e.status ))
    end

    if Softspoken === e or SystemExit === e
      quiet = ((SystemExit === e and e.message.respond_to?('abort')) or # see main/stdext.rb
              (SystemExit === e and e.message == 'exit'))
      stderr.puts e.message unless quiet
    else
      fatal{ e }
    end
  end

  if e.respond_to?(:error_handler_after)
    fcall(e, :error_handler_after, self)
  end

  exit_status(( exit_failure )) if exit_status == exit_success
  exit_status(( Util.integer(exit_status) rescue(exit_status ? 0 : 1) ))
end

#handle_exit(status) ⇒ Object 



240
241
242
 
# File 'lib/main/program/instance_methods.rb', line 240

def handle_exit(status)
  exit(( Integer(status) rescue 1 ))
end

#help!(status = 0) ⇒ Object 



191
192
193
194
 
# File 'lib/main/program/instance_methods.rb', line 191

def help!(status = 0)
  print usage.to_s
  exit(status)
end

#help? ⇒ Boolean

Returns:

  • (Boolean) 


196
197
198
 
# File 'lib/main/program/instance_methods.rb', line 196

def help?
  (params['help'] and params['help'].given?) or argv.first == 'help'
end

#initialize ⇒ Object 



49
 
# File 'lib/main/program/instance_methods.rb', line 49

def initialize() :hook end

#input ⇒ Object 



290
291
292
 
# File 'lib/main/program/instance_methods.rb', line 290

def input
  @input ||= params[:input].value if params[:input]
end

#instance_eval_block(*argv, &block) ⇒ Object 



264
265
266
267
268
269
270
271
 
# File 'lib/main/program/instance_methods.rb', line 264

def instance_eval_block(*argv, &block)
  singleton_class =
    class << self
      self
    end
  singleton_class.module_eval{ define_method('__instance_eval_block', &block) }
  fcall(self, '__instance_eval_block', *argv, &block)
end

#logger=(log) ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
 
# File 'lib/main/program/instance_methods.rb', line 72

def logger= log
  unless(defined?(@logger) and @logger == log)
    case log 
      when ::Logger, Logger
        @logger = log
      else
        if log.is_a?(Array)
          @logger = Logger.new(*log)
        else
          @logger = Logger.new(log)
          @logger.level = logger_level
        end
    end
  end
  @logger
end

#main_env(*args, &block) ⇒ Object 



273
274
275
 
# File 'lib/main/program/instance_methods.rb', line 273

def main_env(*args, &block)
  self.class.main_env(*args, &block)
end

#main_initialize ⇒ Object 



44
45
46
47
48
 
# File 'lib/main/program/instance_methods.rb', line 44

def main_initialize()
  setup_finalizers
  setup_io_redirection
  setup_logging
end

#modes ⇒ Object 



187
188
189
 
# File 'lib/main/program/instance_methods.rb', line 187

def modes
  self.class.modes
end

#output ⇒ Object 



294
295
296
 
# File 'lib/main/program/instance_methods.rb', line 294

def output
  @output ||= params[:output].value if params[:output]
end

#parse_parameters ⇒ Object 



165
166
167
168
169
170
171
172
173
174
 
# File 'lib/main/program/instance_methods.rb', line 165

def parse_parameters
  pre_parse_parameters
  before_parse_parameters

  self.class.parameters.parse(self)
  self.class.parameters.each{|p| params[p.name.to_s] = p}

  after_parse_parameters
  post_parse_parameters
end

#post_initialize ⇒ Object 



51
 
# File 'lib/main/program/instance_methods.rb', line 51

def post_initialize() :hook end

#post_parse_parameters ⇒ Object 



176
 
# File 'lib/main/program/instance_methods.rb', line 176

def post_parse_parameters() :hook end

#post_run ⇒ Object 



184
 
# File 'lib/main/program/instance_methods.rb', line 184

def post_run() :hook end

#pre_initialize ⇒ Object 



42
 
# File 'lib/main/program/instance_methods.rb', line 42

def pre_initialize() :hook end

#pre_parse_parameters ⇒ Object 



163
 
# File 'lib/main/program/instance_methods.rb', line 163

def pre_parse_parameters() :hook end

#pre_run ⇒ Object 



178
 
# File 'lib/main/program/instance_methods.rb', line 178

def pre_run() :hook end

#run ⇒ Object

Raises:

  • (NotImplementedError) 


180
181
182
 
# File 'lib/main/program/instance_methods.rb', line 180

def run
  raise NotImplementedError, 'run not defined'
end

#setup_finalizers ⇒ Object 



53
54
55
 
# File 'lib/main/program/instance_methods.rb', line 53

def setup_finalizers
  self.class.setup_finalizers(self)
end

#setup_io_redirection ⇒ Object 



61
62
63
64
65
 
# File 'lib/main/program/instance_methods.rb', line 61

def setup_io_redirection
  self.stdin = opts['stdin'] || opts[:stdin] || stdin
  self.stdout = opts['stdout'] || opts[:stdout] || stdout
  self.stderr = opts['stderr'] || opts[:stderr] || stderr
end

#setup_io_restoration ⇒ Object 



89
90
91
92
93
94
95
96
97
98
 
# File 'lib/main/program/instance_methods.rb', line 89

def setup_io_restoration
  [STDIN, STDOUT, STDERR].each do |io|
    dup = io.dup
    finalizers.push(
      lambda do
        io.reopen(dup)
      end
    )
  end
end

#setup_logging ⇒ Object 



67
68
69
70
 
# File 'lib/main/program/instance_methods.rb', line 67

def setup_logging
  log = self.class.logger || stderr
  self.logger = log
end

#shell! ⇒ Object 



200
201
202
203
204
205
 
# File 'lib/main/program/instance_methods.rb', line 200

def shell!
  Pry.hooks.clear_all
  prompt = "#{ name } > "
  Pry.config.prompt = proc{|*a| prompt } 
  binding.pry
end

#state_path(&block) ⇒ Object Also known as: dotdir 



277
278
279
 
# File 'lib/main/program/instance_methods.rb', line 277

def state_path(&block)
  self.class.state_path(&block)
end

#stderr=(io) ⇒ Object 



143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
 
# File 'lib/main/program/instance_methods.rb', line 143

def stderr= io
  unless(defined?(@stderr) and (@stderr == io))
    @stderr =
      if io.respond_to?('write')
        io
      else
        fd = open(io.to_s, 'w+')
        finalizers.push(lambda{ fd.close })
        fd
      end
    begin
      STDERR.reopen(@stderr)
    rescue
      $stderr = @stderr
      ::Object.send(:remove_const, 'STDERR')
      ::Object.send(:const_set, 'STDERR', @stderr)
    end
  end
end

#stdin=(io) ⇒ Object 



101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
 
# File 'lib/main/program/instance_methods.rb', line 101

def stdin= io
  unless(defined?(@stdin) and (@stdin == io))
    @stdin =
      if io.respond_to?('read')
        io
      else
        fd = open(io.to_s, 'r+')
        finalizers.push(lambda{ fd.close })
        fd
      end
    begin
      STDIN.reopen(@stdin)
    rescue
      $stdin = @stdin
      ::Object.send(:remove_const, 'STDIN')
      ::Object.send(:const_set, 'STDIN', @stdin)
    end
  end
end

#stdout=(io) ⇒ Object 



122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
 
# File 'lib/main/program/instance_methods.rb', line 122

def stdout= io
  unless(defined?(@stdout) and (@stdout == io))
    @stdout =
      if io.respond_to?('write')
        io
      else
        fd = open(io.to_s, 'w+')
        finalizers.push(lambda{ fd.close })
        fd
      end
    begin
      STDOUT.reopen(@stdout)
    rescue
      $stdout = @stdout
      ::Object.send(:remove_const, 'STDOUT')
      ::Object.send(:const_set, 'STDOUT', @stdout)
    end
  end
end