package generator

import (
	"fmt"
	"os"
	"path/filepath"
	"testing"

	"katenary.io/internal/generator/labels"
	"katenary.io/internal/utils"

	"github.com/compose-spec/compose-go/cli"
)

func TestSplitPorts(t *testing.T) {
	composeFileContent := `
services:
  foo:
    image: nginx:latest
    labels:
      %[1]s/ports: 80,443
`
	composeFileContent = fmt.Sprintf(composeFileContent, labels.KatenaryLabelPrefix)
	tmpDir, err := os.MkdirTemp("", "katenary-test-override")
	if err != nil {
		t.Fatal(err.Error())
	}
	composeFile := filepath.Join(tmpDir, "compose.yaml")

	os.MkdirAll(tmpDir, utils.DirectoryPermission)
	if err := os.WriteFile(composeFile, []byte(composeFileContent), 0644); err != nil {
		t.Log(err)
	}
	defer os.RemoveAll(tmpDir)

	// chand dir to this directory
	os.Chdir(tmpDir)
	options, _ := cli.NewProjectOptions(nil,
		cli.WithWorkingDirectory(tmpDir),
		cli.WithDefaultConfigPath,
	)
	project, err := cli.ProjectFromOptions(options)
	if err != nil {
		t.Fatal(err)
	}
	if err := fixPorts(&project.Services[0]); err != nil {
		t.Errorf("Expected no error, got %s", err)
	}
	found := 0
	for _, p := range project.Services[0].Ports {
		switch p.Target {
		case 80, 443:
			found++
		}
	}
	if found != 2 {
		t.Errorf("Expected 2 ports, got %d", found)
	}
}

func TestSplitPortsWithDefault(t *testing.T) {
	composeFileContent := `
services:
  foo:
    image: nginx:latest
    ports:
      - 8080
    labels:
      %[1]s/ports: 80,443
`
	composeFileContent = fmt.Sprintf(composeFileContent, labels.KatenaryLabelPrefix)
	tmpDir, err := os.MkdirTemp("", "katenary-test-override")
	if err != nil {
		t.Fatal(err)
	}
	composeFile := filepath.Join(tmpDir, "compose.yaml")

	os.MkdirAll(tmpDir, utils.DirectoryPermission)
	if err := os.WriteFile(composeFile, []byte(composeFileContent), 0644); err != nil {
		t.Log(err)
	}
	defer os.RemoveAll(tmpDir)

	// chand dir to this directory
	os.Chdir(tmpDir)
	options, _ := cli.NewProjectOptions(nil,
		cli.WithWorkingDirectory(tmpDir),
		cli.WithDefaultConfigPath,
	)
	project, err := cli.ProjectFromOptions(options)
	if err != nil {
		t.Fatal(err)
	}
	if err := fixPorts(&project.Services[0]); err != nil {
		t.Errorf("Expected no error, got %s", err)
	}
	found := 0
	for _, p := range project.Services[0].Ports {
		switch p.Target {
		case 80, 443, 8080:
			found++
		}
	}
	if found != 3 {
		t.Errorf("Expected 3 ports, got %d", found)
	}
}