BigW Consortium Gitlab

auth.rb 1.17 KB
Newer Older
1
module GoogleApi
2
  class Auth
3 4
    attr_reader :access_token, :redirect_uri, :state

5 6
    ConfigMissingError = Class.new(StandardError)

7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
    def initialize(access_token, redirect_uri, state: nil)
      @access_token = access_token
      @redirect_uri = redirect_uri
      @state = state
    end

    def authorize_url
      client.auth_code.authorize_url(
        redirect_uri: redirect_uri,
        scope: scope,
        state: state # This is used for arbitary redirection
      )
    end

    def get_token(code)
22 23
      ret = client.auth_code.get_token(code, redirect_uri: redirect_uri)
      return ret.token, ret.expires_at
24 25 26 27 28 29 30 31 32 33 34 35 36
    end

    protected

    def scope
      raise NotImplementedError
    end

    private

    def config
      Gitlab.config.omniauth.providers.find { |provider| provider.name == "google_oauth2" }
    end
37 38 39 40 41 42 43 44 45 46 47 48

    def client
      return @client if defined?(@client)

      unless config
        raise ConfigMissingError
      end

      @client = ::OAuth2::Client.new(
        config.app_id,
        config.app_secret,
        site: 'https://accounts.google.com',
49
        token_url: '/o/oauth2/token',
50 51 52
        authorize_url: '/o/oauth2/auth'
      )
    end
53 54
  end
end