googleapis--mcp-toolbox/internal/sources/serverlessspark/url.go

// Copyright 2026 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package serverlessspark

import (
	"fmt"
	"net/url"
	"regexp"
	"time"

	"cloud.google.com/go/dataproc/v2/apiv1/dataprocpb"
)

var batchFullNameRegex = regexp.MustCompile(`projects/(?P<project>[^/]+)/locations/(?P<location>[^/]+)/batches/(?P<batch_id>[^/]+)`)
var sessionTemplateFullNameRegex = regexp.MustCompile(`projects/(?P<project>[^/]+)/locations/(?P<location>[^/]+)/sessionTemplates/(?P<template_id>[^/]+)`)

const (
	logTimeBufferBefore = 1 * time.Minute
	logTimeBufferAfter  = 10 * time.Minute
)

// Extract BatchDetails extracts the project ID, location, and batch ID from a fully qualified batch name.
func ExtractBatchDetails(batchName string) (projectID, location, batchID string, err error) {
	matches := batchFullNameRegex.FindStringSubmatch(batchName)
	if len(matches) < 4 {
		return "", "", "", fmt.Errorf("failed to parse batch name: %s", batchName)
	}
	return matches[1], matches[2], matches[3], nil
}

// BatchConsoleURL builds a URL to the Google Cloud Console linking to the batch summary page.
func BatchConsoleURL(projectID, location, batchID string) string {
	return fmt.Sprintf("https://console.cloud.google.com/dataproc/batches/%s/%s/summary?project=%s", location, batchID, projectID)
}

// BatchLogsURL builds a URL to the Google Cloud Console showing Cloud Logging for the given batch and time range.
//
// The implementation adds some buffer before and after the provided times.
func BatchLogsURL(projectID, location, batchID string, startTime, endTime time.Time) string {
	advancedFilterTemplate := `resource.type="cloud_dataproc_batch"
resource.labels.project_id="%s"
resource.labels.location="%s"
resource.labels.batch_id="%s"`
	advancedFilter := fmt.Sprintf(advancedFilterTemplate, projectID, location, batchID)
	if !startTime.IsZero() {
		actualStart := startTime.Add(-1 * logTimeBufferBefore)
		advancedFilter += fmt.Sprintf("\ntimestamp>=\"%s\"", actualStart.Format(time.RFC3339Nano))
	}
	if !endTime.IsZero() {
		actualEnd := endTime.Add(logTimeBufferAfter)
		advancedFilter += fmt.Sprintf("\ntimestamp<=\"%s\"", actualEnd.Format(time.RFC3339Nano))
	}

	v := url.Values{}
	v.Add("resource", "cloud_dataproc_batch/batch_id/"+batchID)
	v.Add("advancedFilter", advancedFilter)
	v.Add("project", projectID)

	return "https://console.cloud.google.com/logs/viewer?" + v.Encode()
}

// BatchConsoleURLFromProto builds a URL to the Google Cloud Console linking to the batch summary page.
func BatchConsoleURLFromProto(batchPb *dataprocpb.Batch) (string, error) {
	projectID, location, batchID, err := ExtractBatchDetails(batchPb.GetName())
	if err != nil {
		return "", err
	}
	return BatchConsoleURL(projectID, location, batchID), nil
}

// BatchLogsURLFromProto builds a URL to the Google Cloud Console showing Cloud Logging for the given batch and time range.
func BatchLogsURLFromProto(batchPb *dataprocpb.Batch) (string, error) {
	projectID, location, batchID, err := ExtractBatchDetails(batchPb.GetName())
	if err != nil {
		return "", err
	}
	createTime := batchPb.GetCreateTime().AsTime()
	stateTime := batchPb.GetStateTime().AsTime()
	return BatchLogsURL(projectID, location, batchID, createTime, stateTime), nil
}

// ExtractSessionTemplateDetails extracts the project ID, location, and session template ID from a fully qualified sessionTemplateName.
func ExtractSessionTemplateDetails(sessionTemplateName string) (projectID, location, sessionTemplateID string, err error) {
	matches := sessionTemplateFullNameRegex.FindStringSubmatch(sessionTemplateName)
	if len(matches) < 4 {
		return "", "", "", fmt.Errorf("failed to parse session template name: %s", sessionTemplateName)
	}
	return matches[1], matches[2], matches[3], nil
}

var sessionFullNameRegex = regexp.MustCompile(`projects/(?P<project>[^/]+)/locations/(?P<location>[^/]+)/sessions/(?P<session_id>[^/]+)`)

// ExtractSessionDetails extracts the project ID, location, and session ID from a fully qualified session name.
func ExtractSessionDetails(sessionName string) (projectID, location, sessionID string, err error) {
	matches := sessionFullNameRegex.FindStringSubmatch(sessionName)
	if len(matches) < 4 {
		return "", "", "", fmt.Errorf("failed to parse session name: %s", sessionName)
	}
	return matches[1], matches[2], matches[3], nil
}

// SessionConsoleURL builds a URL to the Google Cloud Console linking to the session summary page.
func SessionConsoleURL(projectID, location, sessionID string) string {
	return fmt.Sprintf("https://console.cloud.google.com/dataproc/interactive/%s/%s/details?project=%s", location, sessionID, projectID)
}

// SessionLogsURL builds a URL to the Google Cloud Console showing Cloud Logging for the given session and time range.
func SessionLogsURL(projectID, location, sessionID string, startTime, endTime time.Time) string {
	advancedFilterTemplate := `resource.type="cloud_dataproc_session"
resource.labels.session_id=%q
resource.labels.project_id=%q
resource.labels.location=%q`

	advancedFilter := fmt.Sprintf(advancedFilterTemplate, sessionID, projectID, location)

	if !startTime.IsZero() {
		actualStart := startTime.Add(-1 * logTimeBufferBefore)
		advancedFilter += fmt.Sprintf("\ntimestamp>=\"%s\"", actualStart.Format(time.RFC3339Nano))
	}
	if !endTime.IsZero() {
		actualEnd := endTime.Add(logTimeBufferAfter)
		advancedFilter += fmt.Sprintf("\ntimestamp<=\"%s\"", actualEnd.Format(time.RFC3339Nano))
	}

	v := url.Values{}
	v.Add("advancedFilter", advancedFilter)
	v.Add("project", projectID)

	return "https://console.cloud.google.com/logs/viewer?" + v.Encode()
}

// SessionConsoleURLFromProto builds a URL to the Google Cloud Console linking to the session summary page.
func SessionConsoleURLFromProto(sessionPb *dataprocpb.Session) (string, error) {
	projectID, location, sessionID, err := ExtractSessionDetails(sessionPb.GetName())
	if err != nil {
		return "", err
	}
	return SessionConsoleURL(projectID, location, sessionID), nil
}

// SessionLogsURLFromProto builds a URL to the Google Cloud Console showing Cloud Logging for the given session and time range.
func SessionLogsURLFromProto(sessionPb *dataprocpb.Session) (string, error) {
	projectID, location, sessionID, err := ExtractSessionDetails(sessionPb.GetName())
	if err != nil {
		return "", err
	}
	createTime := sessionPb.GetCreateTime().AsTime()
	stateTime := sessionPb.GetStateTime().AsTime()
	return SessionLogsURL(projectID, location, sessionID, createTime, stateTime), nil
}