class Conoha

# File lib/conoha.rb, line 15
def self.authenticate!
  uri = "https://identity.#{region}.conoha.io/v2.0/tokens"
  payload = payload_for_authentication @@username, @@password, @@tenant_id
  res = https_post uri, payload, nil
  if res.code == '401'
    raise StandardError.new 'Authentication failure'
  end
  @@authtoken = JSON.parse(res.body)["access"]["token"]["id"]
  save_config!
end

# File lib/conoha.rb, line 26
def self.authenticate_user!(user_id)
  uri = "https://identity.#{region}.conoha.io/v2.0/tokens"

  credential = @@accounts[user_id]
  if credential.nil?
    raise StandardError.new "User \"#{user_id}\" doesn't exist."
  end

  payload = payload_for_authentication credential['username'], credential['password'], credential['tenant_id']
  res = https_post uri, payload, nil
  if res.code == '401'
    raise StandardError.new 'Authentication failure'
  end

  auth_token = JSON.parse(res.body)["access"]["token"]["id"]
  set_class_variables_from_credential! credential, auth_token
  save_config!
end

# File lib/conoha.rb, line 135
def self.boot(server_id)
  server_action server_id, "os-start"
end

# File lib/conoha.rb, line 83
def self.create(os, ram, name_tag = nil, user_data: nil)
  image_ref = image_ref_from_image_tag(image_tag_dictionary(os))
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers"
  payload = payload_for_create_vps image_ref, ram, public_key, user_data: user_data
  if name_tag
    payload[:server].merge!({metadata: {instance_name_tag: name_tag}})
  end
  res = https_post uri, payload, authtoken
  JSON.parse(res.body)["server"]["id"]
end

# File lib/conoha.rb, line 167
def self.create_from_image(image_ref, ram, user_data: nil)
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers"
  payload = payload_for_create_vps image_ref, ram, public_key, user_data: user_data
  res = https_post uri, payload, authtoken
  JSON.parse(res.body)["server"]["id"]
end

# File lib/conoha.rb, line 155
def self.create_image(server_id, name)
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers/#{server_id}/action"
  res = https_post uri, {"createImage" => {"name" => name}}, authtoken
  res.code == '202' ? 'OK' : 'Error'
end

# File lib/conoha.rb, line 108
def self.delete(server_id)
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers/#{server_id}"
  res = https_delete uri, authtoken
  res.code == '204' ? 'OK' : 'Error'
end

# File lib/conoha.rb, line 161
def self.delete_image(image_ref)
  uri = "https://image-service.#{region}.conoha.io/v2/images/#{image_ref}"
  res = https_delete uri, authtoken
  res.code == '204' ? 'OK' : 'Error'
end

# File lib/conoha.rb, line 149
def self.images
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/images"
  res = https_get uri, authtoken
  JSON.parse(res.body)["images"].map { |e| [e["name"], e["id"]] }
end

# File lib/conoha.rb, line 7
def self.init!
  load_config!
end

# File lib/conoha.rb, line 114
def self.ip_address_of(server_id)
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers/#{server_id}"
  res = https_get uri, authtoken
  # NOTE: values[1] is needed if eth1 exists.
  JSON.parse(res.body)["server"]["addresses"].values[0].map{ |e| e["addr"] }
end

# File lib/conoha.rb, line 174
def self.name_tag(server_id)
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers/#{server_id}/metadata"
  res = https_get uri, authtoken
  JSON.parse(res.body)["metadata"]["instance_name_tag"]
end

# File lib/conoha.rb, line 145
def self.reboot(server_id, type = "SOFT")
  server_action(server_id, "reboot", { "type" => type })
end

# File lib/conoha.rb, line 94
def self.rebuild(server_id, os)
  image_ref = image_ref_from_image_tag(image_tag_dictionary(os))
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers/#{server_id}/action"
  payload = {
    rebuild: {
      imageRef: image_ref,
      adminPass: randstr,
      key_name: public_key
    }
  }
  res = https_post uri, payload, authtoken
  res.code == '202' ? 'OK' : 'Error'
end

# File lib/conoha.rb, line 11
def self.region
  @@region || 'tyo1'
end

# File lib/conoha.rb, line 129
def self.server_action(server_id, action, action_value = nil)
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers/#{server_id}/action"
  res = https_post uri, {action => action_value}, authtoken
  res.code == '202' ? 'OK' : 'Error'
end

# File lib/conoha.rb, line 66
def self.servers
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers"
  res = https_get uri, authtoken
  JSON.parse(res.body)["servers"]
end

# File lib/conoha.rb, line 139
def self.shutdown(server_id)
  server_action server_id, "os-stop"
end

# File lib/conoha.rb, line 121
def self.status_of(server_id)
  uri = "https://compute.#{region}.conoha.io/v2/#{tenant_id}/servers/#{server_id}"
  res = https_get uri, authtoken
  JSON.parse(res.body)["server"]["status"]
end

# File lib/conoha.rb, line 45
def self.username
  @@username
end

# File lib/conoha.rb, line 72
def self.vps_list
  servers.map { |e| e["id"] }
end

# File lib/conoha.rb, line 54
def self.whoami
  if @@accounts.nil?
    1
  else
    if result = @@accounts.find { |k, v| v['username'] == @@username }
      result.first
    else
      2
    end
  end
end

# File lib/conoha.rb, line 250
def self.authtoken
  @@authtoken
end

# File lib/conoha.rb, line 184
def self.config_file_path
  ENV['HOME'] + '/.conoha-config.json'
end

# File lib/conoha.rb, line 188
  def self.config_file_string
    unless File.exist? config_file_path
      STDERR.print <<EOS
Create "~/.conoha-config.json".
For example:

cat <<EOF > ~/.conoha-config.json
{
  "username": "gncu123456789",
  "password": "your-password",
  "tenant_id": "0123456789abcdef",
  "public_key": "your-registered-public-key-name"
}
EOF
chmod 600 ~/.conoha-config.json # For security
EOS
      exit 1
    end
    File.open(config_file_path).read
  end

# File lib/conoha.rb, line 269
  def self.flavor_ref(ram)
    dictionary = {
      'g-512mb' => 'd92b02ce-9a4f-4544-8d7f-ae8380bc08e7',
      'g-1gb'   => '7eea7469-0d85-4f82-8050-6ae742394681',
      'g-2gb'   => '294639c7-72ba-43a5-8ff2-513c8995b869',
      'g-4gb'   => '62e8fb4b-6a26-46cd-be13-e5bbf5614d15',
      'g-8gb'   => '965affd4-d9e8-4ffb-b9a9-624d63e2d83f',
      'g-16gb'  => '3aa001cd-95b6-46c9-a91e-e62d6f7f06a3',
      'g-32gb'  => 'a20905c6-3733-46c4-81cc-458c7dca1bae',
      'g-64gb'  => 'c2a97b05-1b4b-4038-bbcb-343201659279',
    }
    if dictionary.keys.include? ram
      dictionary[ram]
    else
STDERR.print <<EOS
select ram flavor name from the following list:

#{dictionary.keys.map { |e| "  #{e}" }.join("\n")}
EOS
      exit 1
    end
  end

# File lib/conoha.rb, line 261
def self.image_ref_from_image_tag(image_tag)
  if image = images.find { |e| e[0] == image_tag }
    image[1]
  else
    raise StandardError.new "Tag \"#{image_tag}\" doesn't exist in image list."
  end
end

# File lib/conoha.rb, line 209
def self.load_config!
  unless @@config_loaded
    config = JSON.parse config_file_string
    @@username = config["username"]
    @@password = config["password"]
    @@region = config["region"]
    @@tenant_id = config["tenant_id"]
    @@public_key = config["public_key"]
    @@authtoken = config["authtoken"]
    @@accounts = config["accounts"]
    @@config_loaded = true
  end
end

# File lib/conoha.rb, line 296
def self.payload_for_authentication username, password, tenant_id
  {
    auth: {
      passwordCredentials: {
        username: username,
        password: password
      },
      tenantId: tenant_id
    }
  }
end

# File lib/conoha.rb, line 312
def self.payload_for_create_vps image_ref, ram, public_key, user_data: nil
  ret =
    {
      server: {
        adminPass: randstr,
        imageRef: image_ref,
        flavorRef: flavor_ref(ram),
        key_name: public_key,
        security_groups: [{name: 'default'}, {name: 'gncs-ipv4-all'}],
      }
    }
  ret[:server].merge!({ user_data: user_data }) if user_data
  ret
end

# File lib/conoha.rb, line 254
def self.public_key
  @@public_key
end

# File lib/conoha.rb, line 292
def self.randstr
  (1..60).map{['0'..'9','a'..'z','A'..'Z'].map(&:to_a).flatten.sample}.join
end

# File lib/conoha.rb, line 231
def self.save_config!
  s = JSON.generate({
    username: @@username,
    password: @@password,
    region: @@region,
    tenant_id: @@tenant_id,
    public_key: @@public_key,
    authtoken: @@authtoken,
    accounts: @@accounts
  })
  File.open(config_file_path, 'w') do |f|
    f.write s
  end
end

# File lib/conoha.rb, line 223
def self.set_class_variables_from_credential! credential, auth_token
  @@username = credential['username']
  @@password = credential['password']
  @@tenant_id = credential['tenant_id']
  @@public_key = credential['public_key']
  @@authtoken = authtoken
end

# File lib/conoha.rb, line 246
def self.tenant_id
  @@tenant_id
end