aeolusproject/conductor

#
#   Copyright 2011 Red Hat, Inc.
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#

class TransformFieldsOfProviderAccountToCredentials < ActiveRecord::Migration
  def self.up
    transform
    remove_column :provider_accounts, :username
    remove_column :provider_accounts, :password
    remove_column :provider_accounts, :account_number
    remove_column :provider_accounts, :x509_cert_priv
    remove_column :provider_accounts, :x509_cert_pub

  end

  def self.down
    add_column :provider_accounts, :username, :string
    add_column :provider_accounts, :password, :string
    add_column :provider_accounts, :account_number, :string
    add_column :provider_accounts, :x509_cert_priv, :text
    add_column :provider_accounts, :x509_cert_pub, :text
    transform_back
  end

  def self.transform
    unless ProviderAccount.all.empty?
      ProviderAccount.all.each do |account|
        Credential.create!(:provider_account_id => account.id, :credential_definition_id => CredentialDefinition.find_by_name('username',:conditions => {:provider_type_id => account.provider.provider_type.id}).id, :value => account.username)
        Credential.create!(:provider_account_id => account.id, :credential_definition_id => CredentialDefinition.find_by_name('password',:conditions => {:provider_type_id => account.provider.provider_type.id}).id, :value => account.password)

        # transform more attributes for ec2
        if account.provider.provider_type.codename == 'ec2'
          Credential.create!(:provider_account_id => account.id, :credential_definition_id => CredentialDefinition.find_by_name('account_id',:conditions => {:provider_type_id => account.provider.provider_type.id}).id, :value => account.account_number)
          Credential.create!(:provider_account_id => account.id, :credential_definition_id => CredentialDefinition.find_by_name('x509private',:conditions => {:provider_type_id => account.provider.provider_type.id}).id, :value => account.x509_cert_priv)
          Credential.create!(:provider_account_id => account.id, :credential_definition_id => CredentialDefinition.find_by_name('x509public',:conditions => {:provider_type_id => account.provider.provider_type.id}).id, :value => account.x509_cert_pub)
        end
      end
    end
  end

  def self.transform_back
    unless Credential.all.empty?
      Credential.all.each do |credential|
        if credential.credential_definition.name == 'username'
          credential.provider_account.update_attribute(:username, credential.value)
        elsif credential.credential_definition.name == 'password'
          credential.provider_account.update_attribute(:password, credential.value)
        end
        if credential.credential_definition.provider_type.codename == 'ec2'
          if credential.credential_definition.name == 'account_id'
            credential.provider_account.update_attribute(:account_number, credential.value)
          elsif credential.credential_definition.name == 'x509private'
            credential.provider_account.update_attribute(:x509_cert_priv, credential.value)
          elsif credential.credential_definition.name == 'x509public'
            credential.provider_account.update_attribute(:x509_cert_pub, credential.value)
          end
        end
      end
    end
  end

end