pivotal/projectmonitor

module ConfigExport
  PROJECT_ATTRIBUTES = %w[guid
                          name
                          deprecated_feed_url
                          auth_username
                          auth_password
                          enabled
                          type
                          polling_interval
                          aggregate_project_id
                          deprecated_latest_status_id
                          code
                          tracker_project_id
                          tracker_auth_token
                          cruise_control_rss_feed_url
                          jenkins_base_url
                          jenkins_build_name
                          team_city_base_url
                          team_city_build_id
                          team_city_rest_base_url
                          team_city_rest_build_type_id
                          travis_github_account
                          travis_repository
                          webhooks_enabled
                          concourse_base_url
                          concourse_job_name
                        ]
  AGGREGATE_PROJECT_ATTRIBUTES = %w[id name enabled]

  class << self
    def export
      projects = Project.all.map do |project|
        exported_project_attributes(project)
      end

      aggregate_projects = AggregateProject.all.map do |ap|
        exported_aggregate_project_attributes(ap)
      end

      {'aggregate_projects' => aggregate_projects,
       'projects' => projects}.to_yaml
    end

    def import(config)
      config = YAML.load(config)

      Project.transaction do
        cached_agg = {}
        config['aggregate_projects'].each do |aggregate_project|
          cached_agg[aggregate_project['id']] = AggregateProject.create!(aggregate_project.except('id'))
        end

        config['projects'].each do |project_attributes|
          guid = project_attributes['guid']

          model_class = project_attributes['type'].constantize
          project = if guid.present?
                      model_class.where(guid: guid).first_or_initialize
                    else
                      model_class.new
                    end

          project_attributes.each do |key, value|
            setter_method = "#{key}="
            project.send(setter_method, value) if project.respond_to?(setter_method)
          end
          project.aggregate_project_id = cached_agg[project_attributes['aggregate_project_id']].try(:id)
          project.save(validate: false)
        end
      end
    end

    private

    def exported_project_attributes(project)
      attrs = project.attributes.slice(*PROJECT_ATTRIBUTES)
      attrs['tag_list'] = project.tag_list.to_a
      attrs
    end

    def exported_aggregate_project_attributes(ap)
      attrs = ap.attributes.slice(*AGGREGATE_PROJECT_ATTRIBUTES)
      attrs['tag_list'] = ap.tag_list.to_a
      attrs
    end
  end

end