package provider

import (
	"github.com/Dreamacro/clash/constant"
	rules "github.com/Dreamacro/clash/rule"

	"github.com/stretchr/testify/assert"
	"net"
	"testing"
	"time"
)

func setup() {
	SetClassicalRuleParser(func(ruleType, rule string, params []string) (constant.Rule, error) {
		if params == nil {
			params = make([]string, 0)
		}

		return rules.ParseRule(ruleType, rule, "", params)
	})
}

func TestDomain(t *testing.T) {
	setup()
	domainProvider := NewRuleSetProvider("test", Domain,
		time.Duration(uint(100000)), NewFileVehicle("./domain.txt"))
	assert.Nil(t, domainProvider.Initial())
	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "youtube.com"}))
	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "www.youtube.com"}))
	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "test.youtube.com"}))
	assert.True(t, domainProvider.Match(&constant.Metadata{Host: "bcovlive-a.akamaihd.net"}))
	assert.False(t, domainProvider.Match(&constant.Metadata{Host: "baidu.com"}))
}

func TestClassical(t *testing.T) {
	setup()
	classicalProvider := NewRuleSetProvider("test", Classical,
		time.Duration(uint(100000)), NewFileVehicle("./classical.txt"))
	assert.Nil(t, classicalProvider.Initial())
	assert.True(t, classicalProvider.Match(&constant.Metadata{Host: "www.10010.com", AddrType: constant.AtypDomainName}))
	assert.False(t, classicalProvider.Match(&constant.Metadata{Host: "google.com", AddrType: constant.AtypDomainName}))
	assert.True(t, classicalProvider.Match(&constant.Metadata{Host: "analytics.strava.com", AddrType: constant.AtypDomainName}))
	assert.True(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2a0b:b580::1")}))
	assert.False(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2a0b:c582::1")}))
	assert.True(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("1.255.62.34")}))
	assert.False(t, classicalProvider.Match(&constant.Metadata{DstIP: net.ParseIP("103.65.41.199")}))
}

func TestIpCidr(t *testing.T) {
	setup()
	domainProvider := NewRuleSetProvider("test", IPCIDR,
		time.Duration(uint(100000)), NewFileVehicle("./ipcidr.txt"))
	assert.Nil(t, domainProvider.Initial())
	assert.True(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("91.108.22.10")}))
	assert.False(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("149.190.220.251")}))
	assert.True(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2001:b28:f23f:f005::a")}))
	assert.False(t, domainProvider.Match(&constant.Metadata{DstIP: net.ParseIP("2006:b28:f23f:f005::a")}))
}