Class: Arql::App

Inherits:
Object show all
Defined in:
lib/arql/app.rb

Class Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(options) ⇒ App 



14
15
16
17
18
19
20
21
22
23
 
# File 'lib/arql/app.rb', line 14

def initialize(options)
  require 'active_support/all'
  require 'active_record'
  require "arql/connection"
  require "arql/definition"
  @options = options
  Connection.open(connect_options)
  @definition = Definition.new(effective_config)
  load_initializer!
end

Class Attribute Details

.log_io ⇒ Object

Returns the value of attribute log_io.



7
8
9
 
# File 'lib/arql/app.rb', line 7

def log_io
  @log_io
end

Class Method Details

.config ⇒ Object 



9
10
11
 
# File 'lib/arql/app.rb', line 9

def config
  @@effective_config
end

Instance Method Details

#append_sql ⇒ Object 



116
117
118
119
120
121
 
# File 'lib/arql/app.rb', line 116

def append_sql
  write_sql_file = effective_config[:append_sql]
  App.log_io ||= MultiIO.new
  ActiveRecord::Base.logger = Logger.new(App.log_io)
  App.log_io << File.new(write_sql_file, 'a')
end

#config ⇒ Object 



57
58
59
 
# File 'lib/arql/app.rb', line 57

def config
  @config ||= YAML.load(IO.read(@options.config_file)).with_indifferent_access
end

#connect_options ⇒ Object 



25
26
27
28
29
30
31
32
33
34
 
# File 'lib/arql/app.rb', line 25

def connect_options
  connect_conf = effective_config.slice(:adapter, :host, :username,
                         :password, :database, :encoding,
                         :pool, :port, :socket)
  if effective_config[:ssh].present?
    connect_conf.merge!(start_ssh_proxy!)
  end

  connect_conf
end

#effective_config ⇒ Object 



68
69
70
 
# File 'lib/arql/app.rb', line 68

def effective_config
  @@effective_config ||= selected_config.deep_merge(@options.to_h)
end

#load_initializer! ⇒ Object 



36
37
38
39
40
41
42
43
 
# File 'lib/arql/app.rb', line 36

def load_initializer!
  return unless effective_config[:initializer]
  unless File.exists?(effective_config[:initializer])
    STDERR.puts "Specified initializer file not found, #{effective_config[:initializer]}"
    exit(1)
  end
  load(effective_config[:initializer])
end

#run! ⇒ Object 



72
73
74
75
76
77
78
79
80
81
82
83
84
85
 
# File 'lib/arql/app.rb', line 72

def run!
  show_sql if should_show_sql?
  write_sql if should_write_sql?
  append_sql if should_append_sql?
  if effective_config[:code].present?
    eval(effective_config[:code])
  elsif effective_config[:args].present?
    effective_config[:args].each { |rb| load(rb) }
  elsif STDIN.isatty
    run_repl!
  else
    eval(STDIN.read)
  end
end

#run_repl! ⇒ Object 



87
88
89
 
# File 'lib/arql/app.rb', line 87

def run_repl!
  Repl.new
end

#selected_config ⇒ Object 



61
62
63
64
65
66
 
# File 'lib/arql/app.rb', line 61

def selected_config
  if @options.env.present? && !config[@options.env].present?
    STDERR.puts "Specified ENV `#{@options.env}' not exists"
  end
  config[@options.env]
end

#should_append_sql? ⇒ Boolean 



99
100
101
 
# File 'lib/arql/app.rb', line 99

def should_append_sql?
  effective_config[:append_sql]
end

#should_show_sql? ⇒ Boolean 



91
92
93
 
# File 'lib/arql/app.rb', line 91

def should_show_sql?
  effective_config[:show_sql]
end

#should_write_sql? ⇒ Boolean 



95
96
97
 
# File 'lib/arql/app.rb', line 95

def should_write_sql?
  effective_config[:write_sql]
end

#show_sql ⇒ Object 



103
104
105
106
107
 
# File 'lib/arql/app.rb', line 103

def show_sql
  App.log_io ||= MultiIO.new
  ActiveRecord::Base.logger = Logger.new(App.log_io)
  App.log_io << STDOUT
end

#start_ssh_proxy! ⇒ Object 



45
46
47
48
49
50
51
52
53
54
55
 
# File 'lib/arql/app.rb', line 45

def start_ssh_proxy!
  ssh_config = effective_config[:ssh]
  local_ssh_proxy_port = Arql::SSHProxy.connect(ssh_config.slice(:host, :user, :port, :password).merge(
                                                                                                       forward_host: effective_config[:host],
                                                                                                       forward_port: effective_config[:port],
                                                                                                       local_port: ssh_config[:local_port]))
  {
    host: '127.0.0.1',
    port: local_ssh_proxy_port
  }
end

#write_sql ⇒ Object 



109
110
111
112
113
114
 
# File 'lib/arql/app.rb', line 109

def write_sql
  write_sql_file = effective_config[:write_sql]
  App.log_io ||= MultiIO.new
  ActiveRecord::Base.logger = Logger.new(App.log_io)
  App.log_io << File.new(write_sql_file, 'w')
end