go-oidcを使ってGo言語アプリケーション実装 Azure AD(Microsoft Entra ID)認証

$ sudo apt update -y
$ sudo apt install golang-go -y
$ go version
go version go1.18.1 linux/amd64

$ cd ~/
$ mkdir go-oidc-azure-example
$ cd go-oidc-azure-example
$ vi app.go

package main // このコードがmainパッケージに属していることを示します。

import (
	"crypto/rand"     // 暗号学的に安全な乱数を生成するためのパッケージ
	"encoding/base64" // base64エンコーディングとデコーディングを行うためのパッケージ
	"encoding/json"   // JSONエンコーディングとデコーディングを行うためのパッケージ
	"io"              // 入出力操作を行うためのパッケージ
	"log"             // ログメッセージを出力するためのパッケージ
	"net/http"        // HTTPクライアントとサーバーの実装を提供するパッケージ
	"time"            // 時間を扱うための関数や型を提供するパッケージ

	"github.com/coreos/go-oidc/v3/oidc" // OpenID Connectクライアントを作成するためのパッケージ
	"golang.org/x/net/context"          // コンテキストを管理するためのパッケージ
	"golang.org/x/oauth2"               // OAuth2クライアントを作成するためのパッケージ
)

var (
	// clientID     = os.Getenv("GOOGLE_OAUTH2_CLIENT_ID")
	// clientSecret = os.Getenv("GOOGLE_OAUTH2_CLIENT_SECRET")
	clientID     = "＜アプリケーション (クライアント) ID＞"     // OAuth2クライアントID
	clientSecret = "＜証明書とシークレットの「値」＞" // OAuth2クライアントシークレット
)

func randString(nByte int) (string, error) { // nByte長のランダムな文字列を生成する関数
	b := make([]byte, nByte)
	if _, err := io.ReadFull(rand.Reader, b); err != nil {
		return "", err
	}
	return base64.RawURLEncoding.EncodeToString(b), nil
}

func setCallbackCookie(w http.ResponseWriter, r *http.Request, name, value string) {
	c := &http.Cookie{
		Name:     name,
		Value:    value,
		MaxAge:   int(time.Hour.Seconds()),
		Secure:   r.TLS != nil,
		HttpOnly: true,
		// Path:     "/",
	}
	http.SetCookie(w, c) // クッキーを設定
	// クッキーにより、リクエスト時のstateを保持して、リダイレクトURLのstateとの一致を後で確認できます。
	// 例： Set-Cookie: state=1L7bJe0tMcF7xEDxTrLDnA; Max-Age=3600; HttpOnly
}

func main() { // メイン関数
	ctx := context.Background() // 新しい背景コンテキストを作成

	// provider, err := oidc.NewProvider(ctx, "https://accounts.google.com")
	provider, err := oidc.NewProvider(ctx, "https://login.microsoftonline.com/＜ディレクトリ (テナント) ID＞/v2.0") // 新しいOpenID Connectプロバイダーを作成
	if err != nil {
		log.Fatal(err)
	}
	config := oauth2.Config{ // OAuth2クライアントの設定を作成
		ClientID:     clientID,
		ClientSecret: clientSecret,
		Endpoint:     provider.Endpoint(),
		// RedirectURL:  "http://127.0.0.1:5556/auth/google/callback",
		RedirectURL: "http://localhost:5556/auth/callback",// Azure AD アプリの登録で設定した リダイレクト URI←一致している必要あり
		Scopes:      []string{oidc.ScopeOpenID, "profile", "email"},
	}

	http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) { // ルートパスに対するハンドラを登録
		state, err := randString(16)
		if err != nil {
			http.Error(w, "Internal error", http.StatusInternalServerError)
			return
		}
		setCallbackCookie(w, r, "state", state)

		http.Redirect(w, r, config.AuthCodeURL(state), http.StatusFound) // ユーザーをOAuth2プロバイダーの認証ページにリダイレクト
	})

	// http.HandleFunc("/auth/google/callback", func(w http.ResponseWriter, r *http.Request) {
	http.HandleFunc("/auth/callback", func(w http.ResponseWriter, r *http.Request) { // 認証コールバックに対するハンドラを登録
		// リクエスト時state（クッキー）とコールバックURLのstateと一致しするか確認します。
		state, err := r.Cookie("state") // Cookieからstateパラメータを取得
		if err != nil {
			http.Error(w, "state not found", http.StatusBadRequest) // stateパラメータが見つからない場合はエラーを返す
			return
		}
		// ここで、アプリケーションは以前に送信したstateパラメータと一致するか確認します。
		if r.URL.Query().Get("state") != state.Value {
			http.Error(w, "state did not match", http.StatusBadRequest) // stateパラメータが一致しない場合はエラーを返す
			return
		}

		oauth2Token, err := config.Exchange(ctx, r.URL.Query().Get("code")) // 認証コードをアクセストークンに交換
		if err != nil {
			http.Error(w, "Failed to exchange token: "+err.Error(), http.StatusInternalServerError) // トークンの交換に失敗した場合はエラーを返す
			return
		}

		userInfo, err := provider.UserInfo(ctx, oauth2.StaticTokenSource(oauth2Token)) // ユーザー情報を取得
		if err != nil {
			http.Error(w, "Failed to get userinfo: "+err.Error(), http.StatusInternalServerError) // ユーザー情報の取得に失敗した場合はエラーを返す
			return
		}

		resp := struct { // レスポンスを作成
			OAuth2Token *oauth2.Token
			UserInfo    *oidc.UserInfo
		}{oauth2Token, userInfo}
		data, err := json.MarshalIndent(resp, "", "    ") // レスポンスをJSONに変換
		if err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError) // JSONの変換に失敗した場合はエラーを返す
			return
		}
		w.Write(data) // レスポンスを書き込む
	})

	log.Printf("listening on http://%s/", "127.0.0.1:5556") // サーバーがリッスンしているアドレスをログに出力
	log.Fatal(http.ListenAndServe("127.0.0.1:5556", nil))   // HTTPサーバーを起動
}

$ go mod init example.com/go-oidc-azure-example
$ go mod tidy
$ go run app.go
2023/12/29 22:06:13 listening on http://127.0.0.1:5556/