Class: Flydata::Command::Base

Inherits:
Object
  • Object
show all
Includes:
ExclusiveRunnable, Flydata::CommandLoggable
Defined in:
lib/flydata/command/base.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from ExclusiveRunnable

included

Methods included from Flydata::CommandLoggable

#before_logging, #log_error_stderr, #log_info_stdout, #log_warn_stderr

Constructor Details

#initialize(options = Slop.new) ⇒ Base

Returns a new instance of Base.



18
19
20
21
 
# File 'lib/flydata/command/base.rb', line 18

def initialize(options = Slop.new)
  @api_client = ApiClient.instance
  @opts = options
end

Instance Attribute Details

#opts ⇒ Object (readonly)

Returns the value of attribute opts.



22
23
24
 
# File 'lib/flydata/command/base.rb', line 22

def opts
  @opts
end

Instance Method Details

#ask_input_table_name ⇒ Object 



138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
 
# File 'lib/flydata/command/base.rb', line 138

def ask_input_table_name
  input = nil
  loop do
    log_info_stdout("Input a table name.")
    say(">> ")
    input = gets.strip
    if input =~ /^[a-zA-Z0-9_]+$/
      break
    else
      puts "Please enter a valid table name."
    end
  end
  log_info(">> #{input}")
  input
end

#ask_yes_no(message, default_yes = true) ⇒ Object 



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
 
# File 'lib/flydata/command/base.rb', line 99

def ask_yes_no(message, default_yes=true)
  suffix = default_yes ? "(Y/n)" : "(y/n)"
  prompt = "#{message} #{suffix}:  "
  if opts && opts.yes? # Yes option
    log_info_stdout("#{prompt}Yes")
    return true
  end

  loop do
    ans = ask(prompt)
    return true if default_yes and ans == ''
    if ans.size > 0
      case ans[0].downcase
      when 'y'
        log_info("#{prompt}Yes")
        return true
      when 'n'
        log_info("#{prompt}No")
        return false
      end
    end
    log_warn_stderr(" ! Please answer y[es] or n[o]")
    newline
  end
end

#choose_one(message, prompt, menu_list, value_list = []) ⇒ Object 



124
125
126
127
128
129
130
131
132
133
134
135
136
137
 
# File 'lib/flydata/command/base.rb', line 124

def choose_one(message, prompt, menu_list, value_list=[])
  choice = nil
  value_list = menu_list unless menu_list.size == value_list.size
  log_info_stdout(message) if message
  choose do |menu|
    menu.index = :number
    menu.index_suffix = ") "
    menu.select_by = :index
    menu.prompt = prompt ? prompt : ">> "
    menu.choices(*menu_list) {|item| choice = value_list[menu_list.index(item)]}
  end
  log_info_stdout("  -> #{choice}")
  choice
end

#dashboard_url ⇒ Object 



86
87
88
 
# File 'lib/flydata/command/base.rb', line 86

def dashboard_url
  "#{flydata.flydata_api_host}/dashboard"
end

#data_entry(refresh: false) ⇒ Object 



46
47
48
49
50
51
52
 
# File 'lib/flydata/command/base.rb', line 46

def data_entry(refresh: false)
  if @de.nil? || refresh
    @de = retrieve_data_entries.first
  end
  raise "No data entry exists.  Please set one up on the FlyData Console (#{dashboard_url})" unless @de
  @de
end

#data_port ⇒ Object 



54
55
56
57
 
# File 'lib/flydata/command/base.rb', line 54

def data_port
  return @data_port if @data_port
  @data_port = flydata.data_port.get
end

#flydata ⇒ Object 



23
 
# File 'lib/flydata/command/base.rb', line 23

def flydata; @api_client end

#newline ⇒ Object

print console



95
 
# File 'lib/flydata/command/base.rb', line 95

def newline; puts end

#redshift_cluster ⇒ Object 



59
60
61
62
63
64
65
66
67
 
# File 'lib/flydata/command/base.rb', line 59

def redshift_cluster
  return @redshift_cluster if @redshift_cluster
  @redshift_cluster = flydata.redshift_cluster.show_default
  @redshift_cluster['password'] = Flydata::Util::Encryptor.decrypt(
    @redshift_cluster['encrypted_password'],
    data_port['key'],
    'redshift_cluster password')
  @redshift_cluster
end

#redshift_console_url ⇒ Object 



90
91
92
 
# File 'lib/flydata/command/base.rb', line 90

def redshift_console_url
  "#{flydata.flydata_api_host}/redshift_clusters/query/new"
end

#register_crontab ⇒ Object 



77
78
79
80
81
82
83
84
 
# File 'lib/flydata/command/base.rb', line 77

def register_crontab
  data_entries = retrieve_data_entries
  if data_entries.any?{|e| e['log_deletion']}
    # require on demand to avoid mutual require issue
    require 'flydata/command/crontab'
    Flydata::Command::Crontab.new.run
  end
end

#retrieve_data_entries ⇒ Object 



25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
 
# File 'lib/flydata/command/base.rb', line 25

def retrieve_data_entries
  data_entries = flydata.get('/data_entries')
  unless flydata.response.code == 200 and data_entries
    raise "Failed to retrieve data_entries"
  end
  data_entries.collect do |de|
    source = Source.create(de)
    if Flydata::Preference::DataEntryPreference.conf_exists?(de)
      Flydata::Preference::DataEntryPreference.load_conf(de, source)
    else
      Flydata::Preference::DataEntryPreference.filter_data_entry(de, source)
      de
    end
  end
end

#separator(str = "=") ⇒ Object 



96
97
98
 
# File 'lib/flydata/command/base.rb', line 96

def separator(str="=")
  puts str * 64
end

#show_purpose_name ⇒ Object 



41
42
43
44
 
# File 'lib/flydata/command/base.rb', line 41

def show_purpose_name
  de = data_entry
  log_info_stdout("Your current application name is '#{de['purpose_name']}'")
end

#source(refresh: false) ⇒ Object 



69
70
71
72
73
74
75
 
# File 'lib/flydata/command/base.rb', line 69

def source(refresh: false)
  if @source.nil? || refresh
    @source = nil
    @source = Source.create(data_entry(refresh: refresh))
  end
  @source
end