Skip to content

Commit

Permalink
more refactoriung
Browse files Browse the repository at this point in the history
  • Loading branch information
mose committed Feb 23, 2016
1 parent f7168aa commit 13f616e
Show file tree
Hide file tree
Showing 3 changed files with 22 additions and 18 deletions.
10 changes: 7 additions & 3 deletions app/web.rb
Original file line number Diff line number Diff line change
Expand Up @@ -75,18 +75,22 @@ def check_authorization
def check_authorization
redirect settings.oauth.login_url(request) unless session['access_token']
return init_session if settings.oauth.authorized?(session['access_token'])
flash[:fatal] = 'Sorry you are not authorized to read puppet repo on gitlab.'
redirect '/'
sorry
end

def session_info
Hieraviz::Store.get(session['access_token'], settings.configdata['session_renew'])
Hieraviz::Store.get session['access_token'], settings.configdata['session_renew']
end

def init_session
Hieraviz::Store.set session['access_token'], settings.oauth.user_info(session['access_token'])
settings.oauth.user_info(session['access_token'])['username']
end

def sorry
flash[:fatal] = 'Sorry you are not authorized to read puppet repo on gitlab.'
redirect '/'
end
end

get '/login' do
Expand Down
12 changes: 6 additions & 6 deletions lib/hieraviz.rb
Original file line number Diff line number Diff line change
@@ -1,10 +1,10 @@
require 'hieracles'
require "hieraviz/version"
require "hieraviz/config"
require "hieraviz/store"
require "hieraviz/facts"
require "hieraviz/auth_gitlab"
require "hieraviz/puppetdb"
require 'hieraviz/version'
require 'hieraviz/config'
require 'hieraviz/store'
require 'hieraviz/facts'
require 'hieraviz/auth_gitlab'
require 'hieraviz/puppetdb'

module Hieraviz
# Your code goes here...
Expand Down
18 changes: 9 additions & 9 deletions lib/hieraviz/auth_gitlab.rb
Original file line number Diff line number Diff line change
Expand Up @@ -4,23 +4,23 @@ module Hieraviz
class AuthGitlab

def initialize(settings)
@@client ||= OAuth2::Client.new(
settings['application_id'],
settings['secret'],
:site => settings['host']
)
@client = OAuth2::Client.new(
settings['application_id'],
settings['secret'],
site: settings['host']
)
@settings = settings
end

def access_token(request, code)
@@client.auth_code.get_token(code, :redirect_uri => redirect_uri(request.url))
@client.auth_code.get_token(code, redirect_uri: redirect_uri(request.url))
end

def get_response(url, token)
a_token = OAuth2::AccessToken.new(@@client, token)
a_token = OAuth2::AccessToken.new(@client, token)
begin
JSON.parse(a_token.get(url).body)
rescue Exception => e
rescue StandardError => e
{ 'error' => JSON.parse(e.message.split(/\n/)[1])['message'] }
end
end
Expand All @@ -34,7 +34,7 @@ def redirect_uri(url)
end

def login_url(request)
@@client.auth_code.authorize_url(:redirect_uri => redirect_uri(request.url))
@client.auth_code.authorize_url(redirect_uri: redirect_uri(request.url))
end

def authorized?(token)
Expand Down

0 comments on commit 13f616e

Please sign in to comment.