package utils

import (
	"context"
	"encoding/json"
	"fmt"
	"os"

	"github.com/databricks/cli/bundle"
	"github.com/databricks/cli/cmd/root"
	"github.com/databricks/cli/libs/diag"
	"github.com/spf13/cobra"
)

func configureVariables(cmd *cobra.Command, b *bundle.Bundle, variables []string) diag.Diagnostics {
	return bundle.ApplyFunc(cmd.Context(), b, func(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
		err := b.Config.InitializeVariables(variables)
		return diag.FromErr(err)
	})
}

func configureVariablesFromFile(cmd *cobra.Command, b *bundle.Bundle, filePath string) diag.Diagnostics {
	return bundle.ApplyFunc(cmd.Context(), b, func(ctx context.Context, b *bundle.Bundle) diag.Diagnostics {
		f, err := os.ReadFile(filePath)
		if err != nil {
			return diag.FromErr(fmt.Errorf("failed to read variables file: %w", err))
		}

		vars := map[string]any{}
		err = json.Unmarshal(f, &vars)
		if err != nil {
			return diag.FromErr(fmt.Errorf("failed to parse variables file: %w", err))
		}

		if len(vars) > 0 {
			err = b.Config.InitializeAnyTypeVariables(vars)
			if err != nil {
				return diag.FromErr(err)
			}
		}

		return nil
	})
}

func ConfigureBundleWithVariables(cmd *cobra.Command) (*bundle.Bundle, diag.Diagnostics) {
	// Load bundle config and apply target
	b, diags := root.MustConfigureBundle(cmd)
	if diags.HasError() {
		return b, diags
	}

	variables, err := cmd.Flags().GetStringSlice("var")
	if err != nil {
		return b, diag.FromErr(err)
	}
	variableFilePath, err := cmd.Flags().GetString("vars-file-path")
	if err != nil {
		return b, diag.FromErr(err)
	}

	if len(variables) > 0 && variableFilePath != "" {
		return b, diag.Errorf("cannot specify both --var and --vars-file-path flags")
	} else if len(variables) > 0 {
		// Initialize variables by assigning them values passed as command line flags
		diags = diags.Extend(configureVariables(cmd, b, variables))
	} else if variableFilePath != "" {
		// Initialize variables by loading them from a file
		diags = diags.Extend(configureVariablesFromFile(cmd, b, variableFilePath))
	}

	return b, diags
}