Orbit/lib/tasks/new_ui.rake

namespace :new_ui do

  MODULE_APPS = [ %w(announcement bulletin),
                  %w(archive archive_file),
                  %w(ask ask_question ask_category),
                  %w(asset asset),
                  %w(faq qa),
                  %w(gallery album gallery_category),
                  %w(location location),
                  %w(personal_book writing_book),
                  %w(personal_conference writing_conference),
                  %w(personal_experience experience),
                  %w(personal_honor honor),
                  %w(personal_journal writing_journal),
                  %w(personal_patent writing_patent),
                  %w(personal_project project),
                  %w(web_resource web_link)
                ] # app_key, model_name, category_name

  task :migrate => :environment do
    @db = Mongoid.database
    migrate_tags
    migrate_categories
    migrate_approval_on_object
    migrate_managers
    migrate_ad_images_dates
    migrate_ad_banners
    save_pages
    menu_enabled_for_to_hash
    link_url_to_hash
  end

  # :category_name is optional, depends on the naming of the category model: if no conventional, specify it
  task :migrate_categories, [:app_key, :model_name, :category_name] => :environment do |t, args|
    migrate_categories(args)
  end

  task :migrate_tags => :environment do
    migrate_tags
  end

  task :fix_tagged_ids => :environment do
    fix_tagged_ids
  end

  task :migrate_approval_on_object => :environment do
    migrate_approval_on_object
  end

  task :migrate_managers => :environment do
    migrate_managers
  end

  task :migrate_ad_images_dates  => :environment do
    migrate_ad_images_dates
  end

  task :migrate_ad_banners  => :environment do
    migrate_ad_banners
  end

  task :save_pages => :environment do
    save_pages
  end

  task :menu_enabled_for_to_hash => :environment do
    menu_enabled_for_to_hash
  end

  task :link_url_to_hash => :environment do
    link_url_to_hash
  end

  def migrate_categories(args = nil)
    if args && args[:app_key] && args[:model_name]
      migrate_category(args[:app_key], args[:model_name], args[:category_name])
    else
      MODULE_APPS.each{|value_array| migrate_category(value_array[0], value_array[1], value_array[2])}
    end
  end

  def migrate_category(app_key, model_name, category_name = nil)
    @db ||= Mongoid.database
    category_name = "#{model_name}_category" unless category_name
    collection_category = @db[category_name.pluralize]
    categories = collection_category.find.entries if collection_category
    collection_model = @db[model_name.pluralize]
    collection_buffer_category = @db['buffer_categories']
    collection_prototype_auth = @db['prototype_auths']
    if categories.present?
      module_app = ModuleApp.where(key: app_key).first
      if module_app.try(:has_category)
        categories.each do |category|
          new_category = module_app.categories.build
          new_category.title_translations = category['title']
          new_category.disable = category['disable']
          new_category.save
          collection_model.find("#{category_name}_id" => category['_id']).each do |object|
            object['category_id'] = new_category.id
            object.delete("#{category_name}_id")
            collection_model.save object
            collection_buffer_category.save ({ '_type' => 'BufferCategory', 'category_id' => new_category.id, 'categorizable_type' => model_name.camelize, 'categorizable_id' => object['_id'] })
          end
          object_auths = collection_prototype_auth.find(_type: 'ObjectAuth', obj_authable_id: category['_id'], '$or' => [{privilege_user_ids: {'$not' => {'$size' => 0}}}, {role_ids: {'$not' => {'$size' => 0}}}, {sub_role_ids: {'$not' => {'$size' => 0}}}]).entries
          if object_auths.present?
            object_auths.each do |object_auth|
              if object_auth['title'].eql?('fact_check')
                authorization = new_category.auth_approval || new_category.create_auth_approval(module_app_id: module_app.id, title: "category_approval_#{module_app.key}")   
              else
                authorization = new_category.auth_sub_manager || new_category.create_auth_sub_manager(module_app_id: module_app.id, title: "category_authorization_#{module_app.key}")
              end
              add_users(object_auth, authorization)
              collection_prototype_auth.remove(object_auth)
            end
          end
        end
        collection_category.drop
        collection_prototype_auth.remove(obj_authable_type: category_name.classify)
      end
    else
      collection_category.drop
      collection_prototype_auth.remove(obj_authable_type: category_name.classify)
    end
  end

  def fix_tagged_ids
    Tag.all.each do |tag|
      tag.taggings.each do |tagging|
        tagged_ids = tagging.taggable.tagged_ids
        unless tagged_ids.include?(tag.id.to_s)
          tagged_ids << tag.id.to_s
          tagging.taggable.update_attribute(:tagged_ids, tagged_ids)
        end
      end
    end
  end

  def migrate_tags
    @db ||= Mongoid.database
    tags = Tag.where(_type: 'Tag').entries
    tags.each_with_index do |tag, i|
      unless tag.tag_lease
        if module_app = ModuleApp.find(tag[:module_tag_id])
          new_tag = module_app.module_tags.build
          new_tag.name_translations = tag['name']
          new_tag.is_default = tag['is_default']
          new_tag.save
          n_tag = new_tag.tag
          n_tag.cloud_view_count = tag['cloud_view_count']
          n_tag.view_count = tag['view_count']
          n_tag.save
          tag.taggings.each do |tagging|
            tagged_ids = tagging.taggable.tagged_ids
            unless tagged_ids.include?(tag.id.to_s)
              tagged_ids << tag.id.to_s
              tagging.taggable.update_attribute(:tagged_ids, tagged_ids)
            end
            tagging.taggable.taggings.create(tag_id: n_tag.id)
          end
        end
        tag.destroy
      end
    end
    old_tags = Tag.where(_type: {'$ne' => 'Tag'}).delete
    @db = Mongoid.database
    @db['proto_tags'].drop
  end

  def migrate_approval_on_object
    @db ||= Mongoid.database
    collection_prototype_auth = @db['prototype_auths']
    object_auths = collection_prototype_auth.find({_type: 'ObjectAuth', obj_authable_type: {'$in' => ["AdBanner", "PageContext"]}, '$or' => [{privilege_user_ids: {'$not' => {'$size' => 0}}}, {role_ids: {'$not' => {'$size' => 0}}}, {sub_role_ids: {'$not' => {'$size' => 0}}}]}).entries
    object_auths.each do |object_auth|
      case object_auth['obj_authable_type']
      when 'AdBanner'
        module_app = ModuleApp.where(key: 'ad_banner').first
      when 'PageContext'
        module_app = ModuleApp.where(key: 'page_content').first
      end
      if module_app
        object = object_auth['obj_authable_type'].constantize.find(object_auth['obj_authable_id']) rescue nil
        if object 
          authorization = object.create_auth_sub_manager(module_app_id: module_app.id, title: "#{object_auth['obj_authable_type'].underscore}_authorization_#{module_app.key}")
          add_users(object_auth, authorization) if authorization
        end
      end
      collection_prototype_auth.remove(object_auth)
    end
    collection_prototype_auth.remove(obj_authable_type: {'$in' => ["AdBanner", "PageContext"]})
  end

  def add_users(object_auth, authorization)    
    if authorization
      if object_auth['role_ids'].present?
        roles = Role.find(object_auth['role_ids']) rescue nil
        authorization.add_roles(roles) if roles
      end
      if object_auth['sub_role_ids'].present?
        sub_roles = SubRole.find(object_auth['sub_role_ids']) rescue nil
        authorization.add_sub_roles(sub_roles) if sub_roles
      end
      if object_auth['privilege_user_ids'].present?
        users = User.find(object_auth['privilege_user_ids']) rescue nil
        authorization.add_users(users) if users
      end
    end
  end

  def migrate_managers
    @db ||= Mongoid.database
    collection_app_manager = @db['app_managers']
    app_managers = collection_app_manager.find(managing_app_id: {'$not' => {'$size' => 0}}).entries
    app_managers.each do |app_manager|
      module_app = ModuleApp.find(app_manager['managing_app_id']) rescue nil
      user = User.find(app_manager['user_id']) rescue nil
      if module_app && user
        authorization = module_app.auth_manager || module_app.create_auth_manager(module_app_id: module_app.id, title: module_app.key)
        authorization.add_users(user)
      end
    end
    collection_app_manager.drop
  end

  task :reset_user_position => :environment do
    User.where(email: /guest/).each{|d|d.update_attribute(:position, nil)}
    User.not_guest_user.each_with_index do |user, i|
      user.update_attribute(:position, i)
    end
  end

  def migrate_ad_images_dates
    @db ||= Mongoid.database
    collection_ad_image = @db['ad_images']
    ad_images = collection_ad_image.find()
    ad_images.each do |ad_image|
      ad_image['postdate'] = ad_image['post_date']
      ad_image['deadline'] = ad_image['unpost_date']
      ad_image.delete('post_date')
      ad_image.delete('unpost_date')
      collection_ad_image.save(ad_image)
    end
  end

  def migrate_ad_banners
    @db ||= Mongoid.database
    collection_ad_banner = @db['ad_banners']
    ad_banners = collection_ad_banner.find()
    ad_banners.each do |ad_banner|
      ad_banner['timeout'] = ad_banner['transition_msec']/1000 if ad_banner['transition_msec'].present?
      ad_banner['speed'] = 500 unless ad_banner['speed'].present?
      ad_banner['width'], ad_banner['height'] = ad_banner['best_size'].gsub('px', '').delete(' ').split('x').map{|d| d.to_i} if ad_banner['best_size'].present?
      ad_banner['ad_fx'] = (AdBanner::FX_TYPES.include?(ad_banner['ad_fx']) ? ad_banner['ad_fx'] : AdBanner::FX_TYPES[0])
      ad_banner.delete('best_size')
      ad_banner.delete('transition_msec')
      collection_ad_banner.save(ad_banner)
    end
  end

  def save_pages
    Page.all.each do |page| 
      page.save
    end
  end

  def menu_enabled_for_to_hash
    @db ||= Mongoid.database
    collection_item = @db['items']
    items = collection_item.find()
    items.each do |item|
      item['menu_enabled_for'] = \
      if item['menu_enabled_for'].present?
        VALID_LOCALES.inject({}) do |enable, locale|
          enable[locale] = Array(item['menu_enabled_for']).include?(locale.to_s)
          enable
        end
      else
        VALID_LOCALES.inject({}) do |enable, locale|
          enable[locale] = false
          enable
        end
      end
      collection_item.save(item)
    end
  end

  def link_url_to_hash
    @db ||= Mongoid.database
    collection_link = @db['items']
    links = collection_link.find({_type: 'Link'})
    links.each do |link|
      link['url'] = VALID_LOCALES.inject({}) do |enable, locale|
        enable[locale] = link['url']
        enable
      end
      collection_link.save(link)
    end
  end

end