Class: Cloudflare::DNS::Update::Command::Top

Inherits:
Samovar::Command
  • Object
show all
Defined in:
lib/cloudflare/dns/update/command.rb

Overview

The top level utopia command.

Instance Attribute Summary collapse

Instance Method Summary collapse

Instance Attribute Details

#connection ⇒ Object

Returns the value of attribute connection.



59
60
61
 
# File 'lib/cloudflare/dns/update/command.rb', line 59

def connection
  @connection
end

Instance Method Details

#call ⇒ Object 



164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
 
# File 'lib/cloudflare/dns/update/command.rb', line 164

def call
  if @options[:version]
    puts VERSION
  elsif @options[:help]
    print_usage
  else
    Sync do
      connect! do
        initialize_zone
        initialize_domains
        initialize_command
        
        update_domains
      end
    end
  end
end

#configuration_store ⇒ Object 



55
56
57
 
# File 'lib/cloudflare/dns/update/command.rb', line 55

def configuration_store
  @store ||= YAML::Store.new(@options[:configuration])
end

#connect! ⇒ Object 



61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
 
# File 'lib/cloudflare/dns/update/command.rb', line 61

def connect!
  configuration_store.transaction do |configuration|
    unless configuration[:token]
      prompt.puts "This configuration does not contain authorization token, we require some details."
      configuration[:token] = prompt.mask("Cloudflare token:")
    end
    
    @connection = Cloudflare.connect(token: configuration[:token])
  end
  
  return @connection unless block_given?
  
  begin
    yield @connection
  rescue Interrupt
    # Exit gracefully
  ensure
    @connection.close
  end
end

#initialize_command ⇒ Object 



109
110
111
112
113
114
115
 
# File 'lib/cloudflare/dns/update/command.rb', line 109

def initialize_command
  configuration_store.transaction do |configuration|
    unless configuration[:content_command]
      configuration[:content_command] = prompt.ask("What command to get content for record?", default: 'curl -s ipinfo.io/ip')
    end
  end
end

#initialize_domains ⇒ Object 



92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
 
# File 'lib/cloudflare/dns/update/command.rb', line 92

def initialize_domains
  configuration_store.transaction do |configuration|
    domains = configuration[:domains]
    
    while domains.nil? || domains.empty?
      prompt.puts "Getting list of domains for #{configuration[:zone][:name]}..."
      
      zone_id = configuration[:zone][:id]
      zone = @connection.zones.find_by_id(zone_id)
      
      dns_records = prompt.multi_select("What records do you want to update (select 1 or more)?", zone.dns_records)
      
      domains = configuration[:domains] = dns_records.map(&:value)
    end
  end
end

#initialize_zone ⇒ Object 



82
83
84
85
86
87
88
89
90
 
# File 'lib/cloudflare/dns/update/command.rb', line 82

def initialize_zone
  configuration_store.transaction do |configuration|
    unless configuration[:zone]
      zone = prompt.select("What zone do you want to update?", @connection.zones)
      
      configuration[:zone] = zone.value
    end
  end
end

#logger ⇒ Object 



47
48
49
 
# File 'lib/cloudflare/dns/update/command.rb', line 47

def logger
  @logger ||= Async.logger
end

#prompt ⇒ Object 



51
52
53
 
# File 'lib/cloudflare/dns/update/command.rb', line 51

def prompt
  @prompt ||= TTY::Prompt.new
end

#update_domain(zone, record, content) ⇒ Object 



117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
 
# File 'lib/cloudflare/dns/update/command.rb', line 117

def update_domain(zone, record, content)
  if content != record[:content] || @options[:force]
    logger.info "Content changed #{content.inspect}, updating records..."
    
    domain = zone.dns_records.find_by_id(record[:id])
      
    begin
      domain.update_content(content)
      
      logger.info "Updated domain: #{record[:name]} #{record[:type]} #{content}"
      record[:content] = content
    rescue => error
      logger.warn("Failed to update domain: #{record[:name]} #{record[:type]} #{content}") {error}
    end
  else
    logger.debug "Content hasn't changed: #{record[:name]} #{record[:type]} #{content}"
  end
end

#update_domains(content = nil) ⇒ Object 



136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
 
# File 'lib/cloudflare/dns/update/command.rb', line 136

def update_domains(content = nil)
  configuration_store.transaction do |configuration|
    unless content
      logger.debug "Executing content command: #{configuration[:content_command]}"
      content, status = Open3.capture2(configuration[:content_command])
      
      unless status.success?
        raise RuntimeError.new("Content command failed with non-zero output: #{status}")
      end
      
      # Make sure there is no trailing space:
      content.chomp!
      
      configuration[:content] = content
    end
    
    unless zone = @connection.zones.find_by_id(configuration[:zone][:id])
      raise RuntimeError.new("Couldn't load zone #{configuration[:zone].inspect} from API!")
    end
    
    configuration[:domains].each do |record|
      update_domain(zone, record, content)
    end
  end
  
  return content
end