package sandbox

import (
	"fmt"

	"github.com/yaoapp/yao/config"
)

const taiHost = "host.tai.internal"

// BuildGRPCEnv builds the gRPC environment variables for a sandbox container.
//
// All containers reach the host via "host.tai.internal" (injected by Tai at
// container creation). The port depends on the mode:
//
//   - local:          Yao gRPC port (Tai and Yao on the same machine)
//   - tunnel/direct:  Tai gRPC port (Tai Gateway forwards to Yao)
//
// taiGRPCPort is the Tai node's gRPC port from registration (Ports.GRPC).
func BuildGRPCEnv(mode string, taiGRPCPort int, sandboxID string) map[string]string {
	env := map[string]string{
		"YAO_SANDBOX_ID": sandboxID,
	}

	switch mode {
	case "local":
		port := config.Conf.GRPC.Port
		if port == 0 {
			port = 9099
		}
		env["YAO_GRPC_ADDR"] = fmt.Sprintf("%s:%d", taiHost, port)

	case "tunnel", "direct":
		port := taiGRPCPort
		if port == 0 {
			port = 19100
		}
		env["YAO_GRPC_ADDR"] = fmt.Sprintf("%s:%d", taiHost, port)

	default:
		port := config.Conf.GRPC.Port
		if port == 0 {
			port = 9099
		}
		env["YAO_GRPC_ADDR"] = fmt.Sprintf("%s:%d", taiHost, port)
	}
	return env
}