CartoDB/cartodb20

module CartoDB
  class UserOrganization

    def initialize(org_id, owner_id)
      @owner = ::User.where(:id => owner_id).first
      raise "The organization needs a owner" if @owner.nil?
      if !@owner.organization_id.nil? && @owner.organization_id != org_id
        raise "The user already has a organization and it's not #{org_id}"
      end
      @organization = Carto::Organization.find_by(id: org_id)
      raise "The user needs a organization" if @organization.nil?
      if !@organization.owner_id.nil? && @organization.owner_id != owner_id
        raise "The organization already has a owner and it's not #{owner_id}"
      end
      @users = ::User.where(:organization_id => org_id)
      @active = false
      if !@organization.owner_id.nil?
        @active = true
      end
    end

    def promote_user_to_admin
      raise "Organization is already active. You can't assign an admin" if @active
      @owner.organization_id = @organization.id
      @owner.db_service.move_to_own_schema
      @organization.owner_id = @owner.id
      @organization.admin_email = @owner.email
      @organization.save

      # WIP: CartoDB/cartodb-management#4467
      # Added after commenting it in setup_organization_user_schema to avoid configure_database to reset permissions
      @owner.db_service.reset_user_schema_permissions

      @owner.db_service.setup_organization_user_schema
      @owner.db_service.update_analyses_schema
      @owner.update organization_id: @organization.id
      @owner.db_service.monitor_user_notification
      @active = true
    end

    def owner
      @owner
    end

    def organization
      @organization
    end

    def users
      @users
    end

    def self.from_org_id(organization_id)
      organization = Carto::Organization.find_by(id: organization_id)
      raise "Organization with id #{org_id} does not exist" if organization.nil?
      return CartoDB::UserOrganization.new(organization.id, organization.owner_id)
    end

    def self.user?(name)
      return ::User.where(:username => name).count > 0 ? true : false
    end

    def self.organization?(name)
      Carto::Organization.where(username: name).count.positive?
    end

    def self.user_belongs_to_organization?(name)
      return nil unless CartoDB::UserOrganization.user?(name)

      organization_id = Carto::User.find_by(username: name)&.organization_id
      Carto::Organization.find_by(id: organization_id)&.name
    rescue StandardError
      nil
    end

  end
end