csm_impl

var CSMClusterIdentifierTask = coretask.NewTask(
	csm.CSMClusterIdentifierTaskID,
	[]coretask.Dependency{k8scommon.NEGToBackendServiceInventoryTaskID.Ref()},
	func(ctx context.Context) ([]string, error) {
		inventory := coretask.GetTaskResult(ctx, k8scommon.NEGToBackendServiceInventoryTaskID.Ref())

		uniqueIds := make(map[string]struct{})

		for _, bsName := range inventory {
			if strings.HasPrefix(bsName, "gsmrsvd-") {

				parts := strings.Split(bsName, "-")
				if len(parts) >= 3 {
					uniqueIds[parts[1]] = struct{}{}
				}
			}
		}

		result := make([]string, 0, len(uniqueIds))
		for id := range uniqueIds {
			result = append(result, id)
		}
		return result, nil
	},
)

var CSMTrafficDirectorLogGrouperTask = inspectiontaskbase.NewLogGrouperTask(
	csm.CSMTrafficDirectorLogGrouperTaskID,
	csm.ListCSMTrafficDirectorLogEntriesTaskID.Ref(),
	func(ctx context.Context, l *log.Log) string {
		audit, err := gcpcommon.ExtractGCPAuditLog(l.NodeReader)
		if err != nil {
			return "unknown"
		}
		return audit.ResourceName
	},
)

var CSMTrafficDirectorLogIngesterTask = gcpcommon.NewGCPOperationLogIngesterTask(
	csm.CSMTrafficDirectorLogIngesterTaskID,
	csm.ListCSMTrafficDirectorLogEntriesTaskID.Ref(),
	csm.LogTypeCSMTrafficLog,
)

var CSMTrafficDirectorLogToTimelineMapperTask = inspectiontaskbase.NewLogToTimelineMapperTask(
	csm.CSMTrafficDirectorLogToTimelineMapperTaskID,
	&CSMTrafficDirectorLogToTimelineMapper{},
	inspectioncore.FeatureTaskLabel(
		"CSM Resource Audit Logs",
		"Gather audit logs for Traffic Director resources created by the TD-based CSM to map them to timelines alongside associated Kubernetes resource logs.",
		10100,
		false,
	),
)

var ClusterIdentityAliasTask = coretask.NewAliasTask(
	csm.ClusterIdentityTaskID,
	k8scommon.ClusterIdentityTaskID.Ref(),
)

var InputCSMResponseFlagsTask = formtask.NewSetFormTaskBuilder(csm.InputCSMResponseFlagsTaskID, priorityForCSMGroup+1000, "Envoy response flags").
	WithDefaultValueConstant([]string{"@any", "-OK"}, true).
	WithAllowAddAll(false).
	WithAllowRemoveAll(false).
	WithAllowCustomValue(true).
	WithDescription("Response flags used for filtering CSM traffic logs. Note '-' in response flags is corresponded to 'OK' in this form.").
	WithOptionsFunc(func(ctx context.Context, previousValues []string) ([]inspectionmetadata.SetParameterFormFieldOptionItem, error) {
		result := []inspectionmetadata.SetParameterFormFieldOptionItem{
			{ID: "@any", Description: "[Alias] Matches any response flag"},
		}
		ids := make([]string, 0, len(logutil.EnvoyResponseFlagDescriptions))
		for flag := range logutil.EnvoyResponseFlagDescriptions {
			ids = append(ids, string(flag))
		}
		sort.Strings(ids)
		for _, id := range ids {
			message := logutil.EnvoyResponseFlagDescriptions[logutil.EnvoyResponseFlag(id)]
			if id == "-" {
				id = "OK"
				message = "It's '-' in the response flag field because '-' means subtracting operator in this form."
			}
			result = append(result, inspectionmetadata.SetParameterFormFieldOptionItem{ID: id, Description: message})
		}
		return result, nil
	}).
	WithValidator(func(ctx context.Context, value []string) (string, error) {
		strFilter := strings.Join(value, " ")
		result, err := gcpqueryutil.ParseSetFilter(strFilter, inputCSMAliasMap, true, true, true)
		if err != nil {
			return "", err
		}
		if result.ValidationError == "" {
			err = verifyResponseFlags(convertInputOnlyResponseFlagToActualFlag(result.Additives))
			if err != nil {
				return err.Error(), nil
			}
			err = verifyResponseFlags(convertInputOnlyResponseFlagToActualFlag(result.Subtractives))
			if err != nil {
				return err.Error(), nil
			}
		}
		return result.ValidationError, nil
	}).
	WithConverter(func(ctx context.Context, value []string) (*gcpqueryutil.SetFilterParseResult, error) {
		strFilter := strings.Join(value, " ")
		result, err := gcpqueryutil.ParseSetFilter(strFilter, inputCSMAliasMap, true, true, true)
		if err != nil {
			return nil, err
		}
		result.Additives = convertInputOnlyResponseFlagToActualFlag(result.Additives)
		result.Subtractives = convertInputOnlyResponseFlagToActualFlag(result.Subtractives)
		return result, nil
	}).
	Build()

var InputFleetProjectIDTask = formtask.NewTextFormTaskBuilder(csm.InputFleetProjectIDTaskID, priorityForCSMGroup+500, "Fleet project ID").
	WithDependencies([]coretask.Dependency{
		csm.ClusterIdentityTaskID.Ref(),
	}).
	WithDescription("The project ID where the Fleet is hosted and CSM control plane logs are stored. Default is the cluster's project ID.").
	WithDefaultValueFunc(func(ctx context.Context, previousValues []string) (string, error) {
		if len(previousValues) > 0 {
			return previousValues[0], nil
		}
		cluster := coretask.GetTaskResult(ctx, csm.ClusterIdentityTaskID.Ref())
		return cluster.ProjectID, nil
	}).
	Build()

var ListCSMTrafficDirectorLogEntriesTask = gcpcommon.NewStructuredListLogEntriesTask(&CSMTrafficDirectorListLogEntryTaskSetting{})

var ListLogEntriesTask = gcpcommon.NewStructuredListLogEntriesTask(&CSMTrafficLogListLogEntryTaskSetting{})

var LogGrouperTask = inspectiontaskbase.NewLogGrouperTask(csm.LogGrouperTaskID, csm.ListLogEntriesTaskID.Ref(),
	func(ctx context.Context, l *log.Log) string {
		istioAccessLogFieldSet, err := csm.ExtractIstioAccessLog(l.NodeReader)
		if err != nil {
			return "unknown"
		}
		return fmt.Sprintf("%s-%s", istioAccessLogFieldSet.ReporterPodNamespace, istioAccessLogFieldSet.ReporterPodName)
	},
)

var LogIngesterTask = inspectiontaskbase.NewLogIngesterTask(
	csm.LogIngesterTaskID,
	&CSMTrafficLogLogIngester{},
)

var LogToTimelineMapperTask = inspectiontaskbase.NewLogToTimelineMapperTask(
	csm.LogToTimelineMapperTaskID,
	&CSMTrafficLogLogToTimelineMapper{},
	inspectioncore.FeatureTaskLabel(
		"CSM Traffic Logs",
		"Gather CSM traffic logs to visualize network traffic flows and latency under client or server Pod timelines.",
		10000,
		false,
	),
)