BigW Consortium Gitlab

visibility_level.rb 1.93 KB
Newer Older
1 2 3 4 5 6 7
# Gitlab::VisibilityLevel module
#
# Define allowed public modes that can be used for
# GitLab projects to determine project public mode
#
module Gitlab
  module VisibilityLevel
8
    extend CurrentSettings
Felipe Artur committed
9 10 11
    extend ActiveSupport::Concern

    included do
12 13
      scope :public_only,               -> { where(visibility_level: PUBLIC) }
      scope :public_and_internal_only,  -> { where(visibility_level: [PUBLIC, INTERNAL] ) }
14 15

      scope :public_to_user, -> (user) { user && !user.external ? public_and_internal_only : public_only }
Felipe Artur committed
16
    end
17

18 19 20
    PRIVATE  = 0 unless const_defined?(:PRIVATE)
    INTERNAL = 10 unless const_defined?(:INTERNAL)
    PUBLIC   = 20 unless const_defined?(:PUBLIC)
21 22 23 24 25 26 27 28 29 30 31 32 33

    class << self
      def values
        options.values
      end

      def options
        {
          'Private'  => PRIVATE,
          'Internal' => INTERNAL,
          'Public'   => PUBLIC
        }
      end
34

35
      def allowed_for?(user, level)
36
        user.is_admin? || allowed_level?(level.to_i)
37 38
      end

39 40
      # Return true if the specified level is allowed for the current user.
      # Level should be a numeric value, e.g. `20`.
41
      def allowed_level?(level)
42
        valid_level?(level) && non_restricted_level?(level)
43 44 45
      end

      def non_restricted_level?(level)
46 47 48
        restricted_levels = current_application_settings.restricted_visibility_levels

        if restricted_levels.nil?
49 50
          true
        else
51
          !restricted_levels.include?(level)
52
        end
53 54 55 56
      end

      def valid_level?(level)
        options.has_value?(level)
57
      end
58

59 60 61
      def level_name(level)
        level_name = 'Unknown'
        options.each do |name, lvl|
62
          level_name = name if lvl == level.to_i
63 64 65 66
        end

        level_name
      end
67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
    end

    def private?
      visibility_level_field == PRIVATE
    end

    def internal?
      visibility_level_field == INTERNAL
    end

    def public?
      visibility_level_field == PUBLIC
    end
  end
end