[Mia] How to use Google Calendar API with Go: From Refresh Token to Event Acquisition

[refresh_token] --> [Google Token Endpoint] --> [access_token]
[access_token] --> [Google Calendar API] --> [イベント取得]

package google

import (
	"bytes"
	"context"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"os"
	"time"

	"golang.org/x/oauth2"
	"google.golang.org/api/calendar/v3"
	"google.golang.org/api/option"
)

const TokenEndpoint = "https://oauth2.googleapis.com/token"

// AccessTokenResponse GoogleのトークンAPIレスポンス
type AccessTokenResponse struct {
	AccessToken string `json:"access_token"`
	ExpiresIn   int    `json:"expires_in"`
	TokenType   string `json:"token_type"`
	Scope       string `json:"scope"`
}

// GoogleCalendarService Google Calendar APIを利用するサービス
type GoogleCalendarService struct {
	httpClient  *http.Client
	accessToken string
	expiry      time.Time
}

// NewGoogleCalendarService 初期化
func NewGoogleCalendarService() *GoogleCalendarService {
	return &GoogleCalendarService{
		httpClient: &http.Client{},
	}
}

// GetAccessToken リフレッシュトークンを使ってアクセストークンを取得
func (s *GoogleCalendarService) GetAccessToken(refreshToken string) (string, error) {
	// トークンが有効であれば再利用
	if s.accessToken != "" && time.Now().Before(s.expiry) {
		return s.accessToken, nil
	}

	clientID := os.Getenv("GOOGLE_CLIENT_ID")
	clientSecret := os.Getenv("GOOGLE_CLIENT_SECRET")

	if clientID == "" || clientSecret == "" {
		return "", fmt.Errorf("環境変数 GOOGLE_CLIENT_ID または GOOGLE_CLIENT_SECRET が設定されていません")
	}

	payload := map[string]string{
		"client_id":     clientID,
		"client_secret": clientSecret,
		"refresh_token": refreshToken,
		"grant_type":    "refresh_token",
	}
	body, _ := json.Marshal(payload)

	req, err := http.NewRequest("POST", TokenEndpoint, bytes.NewBuffer(body))
	if err != nil {
		return "", fmt.Errorf("リクエスト作成エラー: %v", err)
	}
	req.Header.Set("Content-Type", "application/json")

	resp, err := s.httpClient.Do(req)
	if err != nil {
		return "", fmt.Errorf("HTTPリクエストエラー: %v", err)
	}
	defer resp.Body.Close()

	if resp.StatusCode != http.StatusOK {
		responseBody, _ := ioutil.ReadAll(resp.Body)
		return "", fmt.Errorf("アクセストークン取得失敗: ステータスコード %d, レスポンス %s", resp.StatusCode, string(responseBody))
	}

	var tokenResp AccessTokenResponse
	if err := json.NewDecoder(resp.Body).Decode(&tokenResp); err != nil {
		return "", fmt.Errorf("レスポンスデコードエラー: %v", err)
	}

	s.accessToken = tokenResp.AccessToken
	s.expiry = time.Now().Add(time.Duration(tokenResp.ExpiresIn) * time.Second)
	return s.accessToken, nil
}

// GetEvents 指定された時間範囲のイベントを取得
func (s *GoogleCalendarService) GetEvents(ctx context.Context, accessToken string, timeMin, timeMax time.Time) ([]*calendar.Event, error) {
	srv, err := calendar.NewService(ctx, option.WithTokenSource(oauth2.StaticTokenSource(&oauth2.Token{
		AccessToken: accessToken,
	})))
	if err != nil {
		return nil, fmt.Errorf("Google Calendar APIクライアント作成エラー: %v", err)
	}

	events, err := srv.Events.List("primary").
		ShowDeleted(false).
		SingleEvents(true).
		TimeMin(timeMin.Format(time.RFC3339)).
		TimeMax(timeMax.Format(time.RFC3339)).
		OrderBy("startTime").
		Do()
	if err != nil {
		return nil, fmt.Errorf("Google Calendar APIイベント取得エラー: %v", err)
	}

	return events.Items, nil
}

clientID := os.Getenv("GOOGLE_CLIENT_ID")
clientSecret := os.Getenv("GOOGLE_CLIENT_SECRET")

if clientID == "" || clientSecret == "" {
    return "", fmt.Errorf("環境変数 GOOGLE_CLIENT_ID または GOOGLE_CLIENT_SECRET が設定されていません")
}

// GoogleCalendarTokenServiceに全ユーザー取得メソッドを追加
func (s *GoogleCalendarTokenService) GetAllTokens(ctx context.Context) ([]mia.GoogleCalendarToken, error) {
	var tokens []mia.GoogleCalendarToken
	query := `
		SELECT user_id, refresh_token, created_at, updated_at
		FROM google_calendar_tokens
	`
	err := s.db.SelectContext(ctx, &tokens, query)
	if err != nil {
		return nil, fmt.Errorf("failed to fetch all tokens: %w", err)
	}
	return tokens, nil
}

func main(){
	// Google Calendar Token Service の初期化
	googleTokenService := rdb.NewGoogleCalendarTokenService(db)
	googleTokenValidator, err := google.NewGoogleCalendarTokenValidator()
	if err != nil {
		slog.Error("Failed to initialize Google Calendar Token Validator", "error", err)
		os.Exit(1)
	}
	// Google Calendar Serviceの初期化
	googleCalendarService := google.NewGoogleCalendarService()
	
	
	// 5分ごとにGoogle Calendarのイベントを取得するタスク
	c.AddFunc("0 */5 * * * *", func() {
		ctx := context.Background()
		fetchGoogleCalendarEvents(ctx, googleTokenService, googleCalendarService)
	})
}


// google calendar eventsを取得
func fetchGoogleCalendarEvents(
	ctx context.Context,
	googleTokenService *rdb.GoogleCalendarTokenService,
	googleCalendarService *google.GoogleCalendarService,
) {
	slog.Info("Starting Google Calendar events fetch task")

	// データベースから対象ユーザーを取得
	userTokens, err := googleTokenService.GetAllTokens(ctx)
	if err != nil {
		slog.Error("Failed to fetch user tokens from database", "error", err)
		return
	}

	for _, token := range userTokens {
		// リフレッシュトークンを使ってアクセストークンを取得
		accessToken, err := googleCalendarService.GetAccessToken(token.RefreshToken)
		if err != nil {
			slog.Error("Failed to fetch Google Calendar access token", "userID", token.UserID, "error", err)
			continue
		}

		// Google Calendar APIからイベントを取得
		now := time.Now().UTC()
		fiveMinutesLater := now.Add(5 * time.Minute)
		events, err := googleCalendarService.GetEvents(ctx, accessToken, now, fiveMinutesLater)
		if err != nil {
			slog.Error("Failed to fetch Google Calendar events", "userID", token.UserID, "error", err)
			continue
		}

		// イベントをログに出力
		for _, event := range events {
			slog.Info("Fetched Google Calendar event",
				"userID", token.UserID,
				"event", event.Summary,
				"startTime", event.Start.DateTime,
			)
		}
	}
}

clocky_api_local    | {"time":"2024-11-26T04:00:02.038511052Z","level":"INFO","msg":"Fetched Google Calendar event","userID":4,"event":"テスト：デザイン定例","startTime":"2024-11-26T13:00:00+09:00"}