package experiments

import (
	"fmt"
	"testing"
	"time"

	"github.com/litmuschaos/chaos-ci-lib/pkg"
	"github.com/litmuschaos/chaos-ci-lib/pkg/environment"
	"github.com/litmuschaos/chaos-ci-lib/pkg/infrastructure"
	"github.com/litmuschaos/chaos-ci-lib/pkg/log"
	"github.com/litmuschaos/chaos-ci-lib/pkg/types"
	"github.com/litmuschaos/chaos-ci-lib/pkg/workflow"
	"github.com/litmuschaos/litmus-go-sdk/pkg/sdk"
	models "github.com/litmuschaos/litmus/chaoscenter/graphql/server/graph/model"
	. "github.com/onsi/ginkgo"
	. "github.com/onsi/gomega"
	_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
	"k8s.io/klog"
)

func TestNodeIOStress(t *testing.T) {
	RegisterFailHandler(Fail)
	RunSpecs(t, "BDD test")
}

//BDD for running node-io-stress experiment
var _ = Describe("BDD of running node-io-stress experiment", func() {

	Context("Check for node-io-stress experiment via SDK", func() {
		// Define variables accessible to It and AfterEach
		var (
			experimentsDetails types.ExperimentDetails
			sdkClient          sdk.Client
			err                error
		)

		BeforeEach(func() {
			experimentsDetails = types.ExperimentDetails{}
			err = nil

			//Fetching all the default ENV
			By("[PreChaos]: Fetching all default ENVs")
			log.Infof("[PreReq]: Getting the ENVs for the %v experiment", experimentsDetails.ExperimentName)
			environment.GetENV(&experimentsDetails, "node-io-stress", "node-io-stress-engine")

			// Initialize SDK client
			By("[PreChaos]: Initializing SDK client")
			sdkClient, err = environment.GenerateClientSetFromSDK()
			Expect(err).To(BeNil(), "Unable to generate Litmus SDK client, due to {%v}", err)

			// Setup infrastructure 
			By("[PreChaos]: Setting up infrastructure")
			err = infrastructure.SetupInfrastructure(&experimentsDetails, sdkClient)
			Expect(err).To(BeNil(), "Failed to setup infrastructure, due to {%v}", err)
			
			// Validate that infrastructure ID is properly set
			Expect(experimentsDetails.ConnectedInfraID).NotTo(BeEmpty(), "Setup failed: ConnectedInfraID is empty after connection attempt.")
			
			// Setup probe if configured to do so
			if experimentsDetails.CreateProbe {
				By("[PreChaos]: Setting up probe")
				err = workflow.CreateProbe(&experimentsDetails, sdkClient, experimentsDetails.LitmusProjectID)
				Expect(err).To(BeNil(), "Failed to create probe, due to {%v}", err)
				// Validate that probe was created successfully
				Expect(experimentsDetails.CreatedProbeID).NotTo(BeEmpty(), "Probe creation failed: CreatedProbeID is empty")
			}
		})

		It("Should run the node io stress experiment via SDK", func() {

			// Ensure pre-checks passed from BeforeEach
			Expect(err).To(BeNil(), "Error during BeforeEach setup: %v", err)
			klog.Info("Executing V3 SDK Path for Experiment")

			// 1. Construct Experiment Request
			By("[SDK Prepare]: Constructing Chaos Experiment Request")
			experimentName := pkg.GenerateUniqueExperimentName("node-io-stress")
			experimentsDetails.ExperimentName = experimentName
			experimentID := pkg.GenerateExperimentID()
			experimentRequest, errConstruct := workflow.ConstructNodeIOStressExperimentRequest(&experimentsDetails, experimentID, experimentName)
			Expect(errConstruct).To(BeNil(), "Failed to construct experiment request: %v", errConstruct)

			// 2. Create and Run Experiment via SDK
			By("[SDK Prepare]: Creating and Running Chaos Experiment")
			createResponse, err := sdkClient.Experiments().Create(experimentsDetails.LitmusProjectID, *experimentRequest)
			Expect(err).To(BeNil(), "Failed to create experiment via SDK: %v", err)
			klog.Infof("Created experiment: %s", createResponse)
			
			// 3. Get the experiment run ID
			By("[SDK Query]: Polling for experiment run to become available")
			var experimentRunID string
			maxRetries := 10
			found := false
			
			for i := 0; i < maxRetries; i++ {
				time.Sleep(3 * time.Second)
				
				listExperimentRunsReq := models.ListExperimentRunRequest{
					ExperimentIDs: []*string{&experimentID},
				}
				
				runsList, err := sdkClient.Experiments().ListRuns(listExperimentRunsReq)
				if err != nil {
					klog.Warningf("Error fetching experiment runs: %v", err)
					continue
				}
				
				klog.Infof("Attempt %d: Found %d experiment runs", i+1, 
					len(runsList.ExperimentRuns))
				
				if len(runsList.ExperimentRuns) > 0 {
					experimentRunID = runsList.ExperimentRuns[0].ExperimentRunID
					klog.Infof("Found experiment run ID: %s", experimentRunID)
					found = true
					break
				}
				
				klog.Infof("Retrying after delay...")
			}
			
			Expect(found).To(BeTrue(), "No experiment runs found for experiment after %d retries", maxRetries)
			
			// 4. Poll for Experiment Run Status
			By("[SDK Status]: Polling for Experiment Run Status")
			var finalPhase string
			var pollError error
			timeout := time.After(time.Duration(experimentsDetails.ExperimentTimeout) * time.Minute)
			ticker := time.NewTicker(time.Duration(experimentsDetails.ExperimentPollingInterval) * time.Second)
			defer ticker.Stop()
			
		pollLoop:
			for {
				select {
				case <-timeout:
					pollError = fmt.Errorf("timed out waiting for experiment run %s to complete after %d minutes", experimentRunID, experimentsDetails.ExperimentTimeout)
					klog.Error(pollError)
					break pollLoop
				case <-ticker.C:
					phase, errStatus := sdkClient.Experiments().GetRunPhase(experimentRunID)
					if errStatus != nil {
						klog.Errorf("Error fetching experiment run status for %s: %v", experimentRunID, errStatus)
						continue
					}
					klog.Infof("Experiment Run %s current phase: %s", experimentRunID, phase)
					finalPhases := []string{"Completed", "Completed_With_Error", "Failed", "Error", "Stopped", "Skipped", "Aborted", "Timeout", "Terminated"}
					if pkg.ContainsString(finalPhases, phase) {
						finalPhase = phase
						klog.Infof("Experiment Run %s reached final phase: %s", experimentRunID, phase)
						break pollLoop
					}
				}
			}
			
			// 5. Post Validation / Verdict Check
			By("[SDK Verdict]: Checking Experiment Run Verdict")
			Expect(pollError).To(BeNil())
			Expect(finalPhase).NotTo(BeEmpty(), "Final phase should not be empty after polling")
			Expect(finalPhase).To(Equal("Completed"), fmt.Sprintf("Experiment Run phase should be Completed, but got %s", finalPhase))
		})
		// Cleanup using AfterEach
		AfterEach(func() {
			// Disconnect infrastructure using the new module
			By("[CleanUp]: Cleaning up infrastructure")
			errDisconnect := infrastructure.DisconnectInfrastructure(&experimentsDetails, sdkClient)
			Expect(errDisconnect).To(BeNil(), "Failed to clean up infrastructure, due to {%v}", errDisconnect)
		})
	})
})