package wgconf

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

func TestParseConfig(t *testing.T) {
	content := `[Interface]
PrivateKey = ABC123XYZ
Address = 10.0.0.1/24
DNS = 1.1.1.1

[Peer]
PublicKey = DEF456UVW
Endpoint = 1.2.3.4:51820
AllowedIPs = 0.0.0.0/0

[Peer]
PublicKey = GHI789TSR
Endpoint = 5.6.7.8:51820
AllowedIPs = 192.168.1.0/24, 192.168.2.0/24`

	tmpFile := filepath.Join(t.TempDir(), "test.conf")
	if err := os.WriteFile(tmpFile, []byte(content), 0644); err != nil {
		t.Fatal(err)
	}

	cfg, err := Parse(tmpFile)
	if err != nil {
		t.Fatalf("Parse failed: %v", err)
	}

	if cfg.PrivateKey != "ABC123XYZ" {
		t.Errorf("expected PrivateKey ABC123XYZ, got %s", cfg.PrivateKey)
	}
	if cfg.Address != "10.0.0.1/24" {
		t.Errorf("expected Address 10.0.0.1/24, got %s", cfg.Address)
	}
	if cfg.DNS != "1.1.1.1" {
		t.Errorf("expected DNS 1.1.1.1, got %s", cfg.DNS)
	}
	if len(cfg.Peers) != 2 {
		t.Fatalf("expected 2 peers, got %d", len(cfg.Peers))
	}

	p1 := cfg.Peers[0]
	if p1.PublicKey != "DEF456UVW" || p1.Endpoint != "1.2.3.4:51820" || len(p1.AllowedIPs) != 1 || p1.AllowedIPs[0] != "0.0.0.0/0" {
		t.Errorf("Peer 1 mismatch: %+v", p1)
	}

	p2 := cfg.Peers[1]
	if p2.PublicKey != "GHI789TSR" || p2.Endpoint != "5.6.7.8:51820" || len(p2.AllowedIPs) != 2 || p2.AllowedIPs[0] != "192.168.1.0/24" {
		t.Errorf("Peer 2 mismatch: %+v", p2)
	}
}

func TestParseInvalidConfig(t *testing.T) {
	content := `[Interface]
InvalidLineWithoutEquals`
	tmpFile := filepath.Join(t.TempDir(), "invalid.conf")
	if err := os.WriteFile(tmpFile, []byte(content), 0644); err != nil {
		t.Fatal(err)
	}

	_, err := Parse(tmpFile)
	if err == nil {
		t.Error("expected error for invalid line format, got nil")
	}
}