farhatahmad / greenlight

# BigBlueButton open source conferencing system - http://www.bigbluebutton.org/.

#

# Copyright (c) 2018 BigBlueButton Inc. and by respective authors (see below).

#

# This program is free software; you can redistribute it and/or modify it under the

# terms of the GNU Lesser General Public License as published by the Free Software

# Foundation; either version 3.0 of the License, or (at your option) any later

# version.

#

# BigBlueButton is distributed in the hope that it will be useful, but WITHOUT ANY

# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A

# PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.

#

# You should have received a copy of the GNU Lesser General Public License along

# with BigBlueButton; if not, see <http://www.gnu.org/licenses/>.

class ApplicationController < ActionController::Base

  include BbbServer

  before_action :redirect_to_https, :set_user_domain, :set_user_settings, :maintenance_mode?, :migration_error?,

    :user_locale, :check_admin_password, :check_user_role

  # Manually handle BigBlueButton errors

  rescue_from BigBlueButton::BigBlueButtonException, with: :handle_bigbluebutton_error

  protect_from_forgery with: :exceptions

  # Retrieves the current user.

  def current_user

    @current_user ||= User.includes(:roles, :main_room).find_by(id: session[:user_id])

    if Rails.configuration.loadbalanced_configuration

      if @current_user && !@current_user.has_role?(:super_admin) &&

         @current_user.provider != @user_domain

        @current_user = nil

        session.clear

      end

    end

    @current_user

  end

  helper_method :current_user

  def bbb_server

    @bbb_server ||= Rails.configuration.loadbalanced_configuration ? bbb(@user_domain) : bbb("greenlight")

  end

  # Force SSL

  def redirect_to_https

    if Rails.configuration.loadbalanced_configuration && request.headers["X-Forwarded-Proto"] == "http"

      redirect_to protocol: "https://"

    end

  end

  # Sets the user domain variable

  def set_user_domain

    if Rails.env.test? || !Rails.configuration.loadbalanced_configuration

      @user_domain = "greenlight"

    else

      @user_domain = parse_user_domain(request.host)

      check_provider_exists

    end

  end

  # Sets the settinfs variable

  def set_user_settings

    @settings = Setting.includes(:features).find_or_create_by(provider: @user_domain)

  end

  # Redirects the user to a Maintenance page if turned on

  def maintenance_mode?

    if ENV["MAINTENANCE_MODE"] == "true"

      render "errors/greenlight_error", status: 503, formats: :html,

        locals: {

          status_code: 503,

          message: I18n.t("errors.maintenance.message"),

          help: I18n.t("errors.maintenance.help"),

        }

    end

    if Rails.configuration.maintenance_window.present?

      unless cookies[:maintenance_window] == Rails.configuration.maintenance_window

        flash.now[:maintenance] = I18n.t("maintenance.window_alert", date: Rails.configuration.maintenance_window)

      end

    end

  end

  # Show an information page when migration fails and there is a version error.

  def migration_error?

    render :migration_error, status: 500 unless ENV["DB_MIGRATE_FAILED"].blank?

  end

  # Sets the appropriate locale.

  def user_locale(user = current_user)

    locale = if user && user.language != 'default'

      user.language

    else

      http_accept_language.language_region_compatible_from(I18n.available_locales)

    end

    begin

      I18n.locale = locale.tr('-', '_') unless locale.nil?

    rescue

      # Default to English if there are any issues in language

      logger.error("Support: User locale is not supported (#{locale}")

      I18n.locale = "en"

    end

  end

  # Checks to make sure that the admin has changed his password from the default

  # Checks if the user is banned and logs him out if he is

  def check_user_role

    if current_user&.has_role? :denied

      session.delete(:user_id)

      redirect_to root_path, flash: { alert: I18n.t("registration.banned.fail") }

    elsif current_user&.has_role? :pending

      session.delete(:user_id)

      redirect_to root_path, flash: { alert: I18n.t("registration.approval.fail") }

    end

  end

  # Relative root helper (when deploying to subdirectory).

  def relative_root

    Rails.configuration.relative_url_root || ""

  end

  helper_method :relative_root

  # Determines if the BigBlueButton endpoint is configured (or set to default).

  def bigbluebutton_endpoint_default?

    return false if Rails.configuration.loadbalanced_configuration

    Rails.configuration.bigbluebutton_endpoint_default == Rails.configuration.bigbluebutton_endpoint

  end

  helper_method :bigbluebutton_endpoint_default?

  def allow_greenlight_accounts?

    return Rails.configuration.allow_user_signup unless Rails.configuration.loadbalanced_configuration

    return false unless @user_domain && !@user_domain.empty? && Rails.configuration.allow_user_signup

    return false if @user_domain == "greenlight"

    # Proceed with retrieving the provider info

    begin

      provider_info = retrieve_provider_info(@user_domain, 'api2', 'getUserGreenlightCredentials')

      provider_info['provider'] == 'greenlight'

    rescue => e

      logger.error "Error in checking if greenlight accounts are allowed: #{e}"

      false

    end

  end

  helper_method :allow_greenlight_accounts?

  # Determine if Greenlight is configured to allow user signups.

  def allow_user_signup?

    Rails.configuration.allow_user_signup

  end

  helper_method :allow_user_signup?

  # Gets all configured omniauth providers.

  def configured_providers

    Rails.configuration.providers.select do |provider|

      Rails.configuration.send("omniauth_#{provider}")

    end

  end

  helper_method :configured_providers

  # Indicates whether users are allowed to share rooms

  def shared_access_allowed

    @settings.get_value("Shared Access") == "true"

  end

  helper_method :shared_access_allowed

  # Parses the url for the user domain

    return hostname.split('.').first if Rails.configuration.url_host.empty?

    Rails.configuration.url_host.split(',').each do |url_host|

      return hostname.chomp(url_host).chomp('.') if hostname.include?(url_host)

    end

    ''

  end

  # Include user domain in lograge logs

  def append_info_to_payload(payload)

    super

    payload[:host] = @user_domain

  end

  # Manually Handle BigBlueButton errors

  def handle_bigbluebutton_error

    render "errors/bigbluebutton_error"

  end

  # Manually deal with 401 errors

  rescue_from CanCan::AccessDenied do |_exception|

    if current_user

      render "errors/greenlight_error"

    else

      # Store the current url as a cookie to redirect to after sigining in

      cookies[:return_to] = request.url

      # Get the correct signin path

      path = if allow_greenlight_accounts?

        signin_path

      elsif Rails.configuration.loadbalanced_configuration

        omniauth_login_url(:bn_launcher)

      else

        signin_path

      end

      redirect_to path

    end

  end

  private

  def check_provider_exists

    # Checks to see if the user exists

    begin

      # Check if the session has already checked that the user exists

      # and return true if they did for this domain

      return if session[:provider_exists] == @user_domain

      retrieve_provider_info(@user_domain, 'api2', 'getUserGreenlightCredentials')

      # Add a session variable if the provider exists

      session[:provider_exists] = @user_domain

    rescue => e

      logger.error "Error in retrieve provider info: #{e}"

      # Use the default site settings

      @user_domain = "greenlight"

      @settings = Setting.find_or_create_by(provider: @user_domain)

      if e.message.eql? "No user with that id exists"

        render "errors/greenlight_error", locals: { message: I18n.t("errors.not_found.user_not_found.message"),

          help: I18n.t("errors.not_found.user_not_found.help") }

      elsif e.message.eql? "Provider not included."

        render "errors/greenlight_error", locals: { message: I18n.t("errors.not_found.user_missing.message"),

          help: I18n.t("errors.not_found.user_missing.help") }

      elsif e.message.eql? "That user has no configured provider."

        render "errors/greenlight_error", locals: { status_code: 501,

          message: I18n.t("errors.no_provider.message"),

          help: I18n.t("errors.no_provider.help") }

      else

        render "errors/greenlight_error", locals: { status_code: 500, message: I18n.t("errors.internal.message"),

          help: I18n.t("errors.internal.help"), display_back: true }

      end

    end

  end

end