fix: Add basic graph functionality

pkg/graphhelper/graphhelper.go

@@ -0,0 +1,77 @@
+package graphhelper
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"strings"
+
+	"github.com/Azure/azure-sdk-for-go/sdk/azcore/policy"
+	"github.com/Azure/azure-sdk-for-go/sdk/azidentity"
+	auth "github.com/microsoft/kiota-authentication-azure-go"
+	msgraphsdk "github.com/microsoftgraph/msgraph-sdk-go"
+	// "github.com/microsoftgraph/msgraph-sdk-go/models"
+	// "github.com/microsoftgraph/msgraph-sdk-go/users"
+)
+
+type GraphHelper struct {
+	deviceCodeCredential *azidentity.DeviceCodeCredential
+	userClient           *msgraphsdk.GraphServiceClient
+	graphUserScopes      []string
+}
+
+func NewGraphHelper() *GraphHelper {
+	g := &GraphHelper{}
+	return g
+}
+
+func (g *GraphHelper) InitializeGraphForUserAuth() error {
+	clientId := os.Getenv("CLIENT_ID")
+	tenantId := os.Getenv("TENANT_ID")
+	scopes := os.Getenv("GRAPH_USER_SCOPES")
+	g.graphUserScopes = strings.Split(scopes, ",")
+
+	// Create the device code credential
+	credential, err := azidentity.NewDeviceCodeCredential(&azidentity.DeviceCodeCredentialOptions{
+		ClientID: clientId,
+		TenantID: tenantId,
+		UserPrompt: func(ctx context.Context, message azidentity.DeviceCodeMessage) error {
+			fmt.Println(message.Message)
+			return nil
+		},
+	})
+	if err != nil {
+		return err
+	}
+
+	g.deviceCodeCredential = credential
+
+	// Create an auth provider using the credential
+	authProvider, err := auth.NewAzureIdentityAuthenticationProviderWithScopes(credential, g.graphUserScopes)
+	if err != nil {
+		return err
+	}
+
+	// Create a request adapter using the auth provider
+	adapter, err := msgraphsdk.NewGraphRequestAdapter(authProvider)
+	if err != nil {
+		return err
+	}
+
+	// Create a Graph client using request adapter
+	client := msgraphsdk.NewGraphServiceClient(adapter)
+	g.userClient = client
+
+	return nil
+}
+
+func (g *GraphHelper) GetUserToken() (*string, error) {
+	token, err := g.deviceCodeCredential.GetToken(context.Background(), policy.TokenRequestOptions{
+		Scopes: g.graphUserScopes,
+	})
+	if err != nil {
+		return nil, err
+	}
+
+	return &token.Token, nil
+}

pkg/spamasaurusrex/main.go

@@ -7,6 +7,7 @@ import (
 	"net/http"
 	"os"
 	"os/signal"
+	"pkg/spamasaurusrex/pkg/graphhelper"
 	"syscall"
 	"time"
 
@@ -16,6 +17,12 @@ import (
 func handler(w http.ResponseWriter, r *http.Request) {
 	query := r.URL.Query()
 	name := query.Get("name")
+
+	graphHelper := graphhelper.NewGraphHelper()
+
+	initializeGraph(graphHelper)
+	greetUser(graphHelper)
+
 	if name == "" {
 		name = "Guest"
 	}
@@ -73,3 +80,28 @@ func waitForShutdown(srv *http.Server) {
 	log.Println("Shutting down")
 	os.Exit(0)
 }
+
+func initializeGraph(graphHelper *graphhelper.GraphHelper) {
+	err := graphHelper.InitializeGraphForUserAuth()
+	if err != nil {
+		log.Panicf("Error initializing Graph for user auth: %v\n", err)
+	}
+}
+
+func greetUser(graphHelper *graphhelper.GraphHelper) {
+	// TODO
+}
+
+func displayAccessToken(graphHelper *graphhelper.GraphHelper) {
+	token, err := graphHelper.GetUserToken()
+	if err != nil {
+		log.Panicf("Error getting user token: %v\n", err)
+	}
+
+	fmt.Printf("User token: %s", *token)
+	fmt.Println()
+}
+
+func makeGraphCall(graphHelper *graphhelper.GraphHelper) {
+	// TODO
+}