まず、Gemfile に以下の行を追加します。

gem 'omniauth'
gem 'omniauth-openid-connect'

その後、以下のコマンドを実行して gem をインストールします。

bash
bundle install

.env に CLIENT_ID , CLIENT_SECRET を記載します。

COGNITO_DOMAIN= 'https://aaaa.com'
CLIENT_ID= 'xxxxx'
CLIENT_SECRET= 'yyyy'

COGNITO_DOMAIN: Cognito のドメイン

CLIENT_ID: Cognito のクライアント ID

CLIENT_SECRET: Cognito のクライアントシークレット

次に、config/initializers/omniauth.rb に以下の設定を追加します。

Rails.application.config.middleware.use OmniAuth::Builder do
  provider :openid_connect, {
    name: :cognito,
    scope: [:openid, :email, :profile],
    response_type: :code,
    discovery: true,
    client_options: {
      port: 443,
      scheme: 'https',
      host: ENV['COGNITO_DOMAIN'],
      identifier: ENV['CLIENT_ID'],
      secret: ENV['CLIENT_SECRET'],
      redirect_uri: 'http://localhost:3000/auth/cognito/callback'
    }
  }
end

redirect_uri: 認証後のリダイレクト先 (ローカル環境の場合: http://localhost:3000/auth/cognito/callback)

app/sesson_controller.rb を設定します。

class SessionsController < ApplicationController
  def create
    reset_session
    auth = request.env['omniauth.auth']
    user = User.find_or_create_by(uid: auth.uid, provider: auth.provider) do |u|
      u.email = auth.info.email
      u.name = auth.info.name
    end

    session[:user_id] = user.id
    redirect_to root_path, notice: 'Signed in!'
  end

  def destroy
    reset_session
    redirect_to root_path, notice: 'Signed out!'
  end

  def failure
    redirect_to root_path, alert: 'Authentication failed, please try again.'
  end
end

ルーティングに OmniAuth のコールバック URL を追加するために、config/routes.rb を編集します。

Rails.application.routes.draw do
  get    '/auth/:provider/callback', to: 'sessions#create'
  get    '/auth/failure', to: 'sessions#failure'
  delete '/logout', to: 'sessions#destroy'

  root 'home#index'
end

ユーザー情報を保存するための User モデルを作成します。

rails generate model User provider:string uid:string name:string email:string
rails db:migrate

ログインボタンやログアウトリンクをビューに追加します。

例えば、app/views/home/index.html.erb に以下のように記述します。

<% if session[:user_id] %>
  <p>Signed in as <%= current_user.name %></p>
  <%= link_to 'Sign out', logout_path, method: :delete %>
<% else %>
  <%= link_to 'Sign in with Cognito', '/auth/cognito' %>
<% end %>

current_user ヘルパーを追加するために、

app/controllers/application_controller.rb に以下のメソッドを追加します。

class ApplicationController < ActionController::Base
  helper_method :current_user

  def current_user
    @current_user ||= User.find(session[:user_id]) if session[:user_id]
  end
end

以上で、SSO認証(openid_connect) の実装が完了します。

簡単に認証が実装できていいですね！