Merge pull request #275 from NYULibraries/mcain/shibboleth-omniauth

Integrate Shibboleth SSO

Author: the-codetrane

.tool-versions

@@ -0,0 +1,2 @@
+ruby 2.7.8
+nodejs 18.10.0

Gemfile

@@ -10,7 +10,8 @@ gem 'jbuilder'
 gem 'jquery-rails'
 gem 'mimemagic', github: 'mimemagicrb/mimemagic', ref: '01f92d86d15d85cfd0f20dabd025dcbd36a8a60f'
 gem 'mysql2'
-gem 'omniauth-nyulibraries', git: 'https://github.com/NYULibraries/omniauth-nyulibraries', branch: 'v2.2.0'
+gem 'omniauth'
+gem 'omniauth-oauth2'
 gem 'rails'
 gem 'rainbow'
 gem 'rsolr'

Gemfile.lock

@@ -1,11 +1,3 @@
-GIT
-  remote: https://github.com/NYULibraries/omniauth-nyulibraries
-  revision: 2a77afefc346669e5a50ff5c84e48a012490dfbc
-  branch: v2.2.0
-  specs:
-    omniauth-nyulibraries (2.2.0)
-      omniauth-oauth2 (~> 1.2.0)
-
 GIT
   remote: https://github.com/mimemagicrb/mimemagic.git
   revision: 01f92d86d15d85cfd0f20dabd025dcbd36a8a60f
@@ -434,7 +426,8 @@ DEPENDENCIES
   jquery-rails
   mimemagic!
   mysql2
-  omniauth-nyulibraries!
+  omniauth
+  omniauth-oauth2
   puma
   rails
   rainbow

app/assets/config/manifest.js

@@ -1,3 +1,4 @@
 //= link_tree ../images
 //= link_tree ../javascripts
+//= link_tree ../stylesheets
 //= link_directory ../stylesheets .css

app/controllers/application_controller.rb

@@ -35,7 +35,7 @@ def after_sign_out_path_for(resource_or_scope)
   end
 
   def logout_path
-    'https://login.library.nyu.edu/logout'
+    Settings.LOGOUT_URL || 'https://qa.auth.it.nyu.edu/oidc/logout'
   end
   private :logout_path
 end

app/controllers/omniauth_callbacks_controller.rb

@@ -0,0 +1,82 @@
+# frozen_string_literal: true
+
+class OmniauthCallbacksController < Devise::OmniauthCallbacksController
+  before_action :require_valid_omniauth, only: :shibboleth
+
+  def shibboleth
+    Rails.logger.info("OmniauthCallbacksController#shibboleth: #{omniauth.inspect}")
+    set_user
+    if @user.persisted?
+      log_in
+    else
+      redirect_to_login
+    end
+  end
+
+  def failure
+    redirect_to root_path
+  end
+
+  private
+
+  def redirect_to_login
+    Rails.logger.error(find_message(:failure, kind: 'NYU Shibboleth', reason: 'User not persisted'))
+    session['devise.shibboleth_data'] = request.env['omniauth.auth']
+    redirect_to root_path
+  end
+
+  def log_in
+    sign_in_and_redirect @user, event: :authentication
+    set_flash_message(:notice, :success, kind: 'NYU Shibboleth')
+    logger.info(find_message(:success, kind: 'NYU Shibboleth'))
+  end
+
+  def require_valid_omniauth
+    head :bad_request unless valid_omniauth?
+  end
+
+  def valid_omniauth?
+    omniauth.present?
+  end
+
+  def omniauth
+    @omniauth ||= request.env['omniauth.auth']
+  end
+
+  def omniauth_provider
+    @omniauth_provider ||= omniauth.provider
+  end
+
+  def attributes_from_omniauth
+    {
+      provider: omniauth.provider,
+      username: omniauth.uid,
+      email: omniauth_email,
+      firstname: omniauth_firstname,
+      lastname: omniauth_lastname
+    }
+  end
+
+  def omniauth_email
+    @omniauth_email ||= omniauth.info.email
+  end
+
+  def omniauth_firstname
+    @omniauth_firstname ||= omniauth.info.first_name
+  end
+
+  def omniauth_lastname
+    @omniauth_lastname ||= omniauth.info.last_name
+  end
+
+  def set_user
+    # Find existing or initialize new user,
+    # and save new attributes each time
+    @user = find_user
+    @user.update_attributes(attributes_from_omniauth)
+  end
+
+  def find_user
+    @find_user ||= User.create_from_provider_data(request.env['omniauth.auth'])
+  end
+end

app/controllers/users/omniauth_callbacks_controller.rb

@@ -7,12 +7,21 @@ class User < ActiveRecord::Base
   # Include default devise modules. Others available are:
   # :confirmable, :lockable, :timeoutable and :omniauthable
 
-  devise :omniauthable, omniauth_providers: [:nyulibraries]
+  devise :omniauthable, omniauth_providers: [:shibboleth]
 
   # Method added by Blacklight; Blacklight uses #to_s on your
   # user class to get a user-displayable login/identifier for
   # the account.
   def to_s
     email
   end
+
+  def self.create_from_provider_data(provider_data)
+    where(provider: provider_data.provider,
+          username: provider_data.uid)
+      .first_or_create do |user|
+      user.email = provider_data.info.email
+      user.username = provider_data.uid
+    end
+  end
 end

app/models/user.rb

@@ -10,8 +10,9 @@
   config.password_length = 8..128
   config.reset_password_within = 60.minutes
   config.sign_out_via = :get
-  config.omniauth :nyulibraries, Settings.APP_ID, Settings.APP_SECRET, client_options: {
-    site: Settings.LOGIN_URL,
-    authorize_path: '/oauth/authorize'
+  config.omniauth :shibboleth, Settings.APP_ID, Settings.APP_SECRET, client_options: {
+    site: (Settings.LOGIN_URL || "https://qa.auth.it.nyu.edu"),
+    authorize_url: "/oauth2/authorize",
+    token_url: "/oauth2/token",
   }
 end

config/initializers/devise.rb

@@ -0,0 +1,46 @@
+module OmniAuth
+  module Strategies
+    require 'omniauth-oauth2'
+    class Shibboleth < OmniAuth::Strategies::OAuth2
+      if defined?(::Rails) && ::Rails.env.development?
+        silence_warnings do
+          OpenSSL::SSL::VERIFY_PEER = OpenSSL::SSL::VERIFY_NONE
+        end
+      end
+      option :name, :shibboleth
+      option :authorize_params, { scope: "openid" }
+
+      uid do
+        raw_info["sub"]
+      end
+
+      info do
+        {
+          email: raw_info["sub"],
+          last_name: last_name,
+          first_name: first_name
+        }
+      end
+
+      def raw_info
+        response = access_token.get("/oauth2/userinfo?schema=openid")
+        Rails.logger.info("Shibboleth raw_info: #{response.parsed}")
+        @raw_info ||= response.parsed
+      end
+
+      # Extract Last Name from identity
+      def last_name
+        @last_name ||= raw_info["lastname"] rescue nil
+      end
+
+      # Extract First Name from identity
+      def first_name
+        @first_name ||= raw_info["firstname"] rescue nil
+      end
+
+      def log(level, message)
+        Rails.logger.send(level, "(#{name}) #{message}")
+      end
+    end
+  end
+end