package alicloud

import (
	"errors"
	"testing"

	privatelink "github.com/alibabacloud-go/privatelink-20200415/v5/client"
	"github.com/stretchr/testify/assert"
	"k8s.io/utils/ptr"

	"github.com/tidbcloud/observability/agent-control-server/pkg/consts"
)

type mockPrivateLinkClient struct {
	listVpcEndpointsFunc func(*privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error)
}

func (m *mockPrivateLinkClient) ListVpcEndpoints(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
	if m.listVpcEndpointsFunc != nil {
		return m.listVpcEndpointsFunc(req)
	}
	return nil, nil
}

func (m *mockPrivateLinkClient) TagResources(req *privatelink.TagResourcesRequest) (*privatelink.TagResourcesResponse, error) {
	return &privatelink.TagResourcesResponse{}, nil
}

func TestFindVPCEndpointByClusterTag(t *testing.T) {
	region := "cn-hangzhou"
	clusterName := "test-cluster"
	endpointID := "ep-xxx"
	endpointDomain := "ep.domain.com"

	tests := []struct {
		name       string
		mockFunc   func(*privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error)
		wantID     string
		wantDomain string
		wantErr    bool
		errMsg     string
	}{
		{
			name: "success with active endpoint",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				// Verify tag filter is set correctly
				assert.Len(t, req.Tag, 1)
				assert.Equal(t, consts.LabelKeyRelatedCluster, *req.Tag[0].Key)
				assert.Equal(t, clusterName, *req.Tag[0].Value)
				assert.Equal(t, VpcEndpointTypeInterface, *req.EndpointType)

				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{
							{
								EndpointId:     ptr.To(endpointID),
								EndpointDomain: ptr.To(endpointDomain),
								EndpointStatus: ptr.To("Active"),
							},
						},
					},
				}, nil
			},
			wantID:     endpointID,
			wantDomain: endpointDomain,
		},
		{
			name: "success with available endpoint",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{
							{
								EndpointId:     ptr.To(endpointID),
								EndpointDomain: ptr.To(endpointDomain),
								EndpointStatus: ptr.To("Available"),
							},
						},
					},
				}, nil
			},
			wantID:     endpointID,
			wantDomain: endpointDomain,
		},
		{
			name: "skip pending endpoint and return active one",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{
							{
								EndpointId:     ptr.To("ep-pending"),
								EndpointDomain: ptr.To("pending.domain.com"),
								EndpointStatus: ptr.To("Pending"),
							},
							{
								EndpointId:     ptr.To(endpointID),
								EndpointDomain: ptr.To(endpointDomain),
								EndpointStatus: ptr.To("Active"),
							},
						},
					},
				}, nil
			},
			wantID:     endpointID,
			wantDomain: endpointDomain,
		},
		{
			name: "no endpoints found",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{},
					},
				}, nil
			},
			wantID:     "",
			wantDomain: "",
		},
		{
			name: "nil body",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: nil,
				}, nil
			},
			wantID:     "",
			wantDomain: "",
		},
		{
			name: "endpoint with nil domain",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{
							{
								EndpointId:     ptr.To(endpointID),
								EndpointDomain: nil,
								EndpointStatus: ptr.To("Active"),
							},
						},
					},
				}, nil
			},
			wantID:     "",
			wantDomain: "",
		},
		{
			name: "endpoint with nil id",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{
							{
								EndpointId:     nil,
								EndpointDomain: ptr.To(endpointDomain),
								EndpointStatus: ptr.To("Active"),
							},
						},
					},
				}, nil
			},
			wantID:     "",
			wantDomain: "",
		},
		{
			name: "endpoint with nil status",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{
							{
								EndpointId:     ptr.To(endpointID),
								EndpointDomain: ptr.To(endpointDomain),
								EndpointStatus: nil,
							},
						},
					},
				}, nil
			},
			wantID:     "",
			wantDomain: "",
		},
		{
			name: "all endpoints in non-active status",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return &privatelink.ListVpcEndpointsResponse{
					Body: &privatelink.ListVpcEndpointsResponseBody{
						Endpoints: []*privatelink.ListVpcEndpointsResponseBodyEndpoints{
							{
								EndpointId:     ptr.To("ep-1"),
								EndpointDomain: ptr.To("domain1.com"),
								EndpointStatus: ptr.To("Pending"),
							},
							{
								EndpointId:     ptr.To("ep-2"),
								EndpointDomain: ptr.To("domain2.com"),
								EndpointStatus: ptr.To("Deleting"),
							},
						},
					},
				}, nil
			},
			wantID:     "",
			wantDomain: "",
		},
		{
			name: "api error",
			mockFunc: func(req *privatelink.ListVpcEndpointsRequest) (*privatelink.ListVpcEndpointsResponse, error) {
				return nil, errors.New("api error")
			},
			wantErr: true,
			errMsg:  "failed to list vpc endpoints by tag",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			mockCli := &mockPrivateLinkClient{listVpcEndpointsFunc: tt.mockFunc}
			gotID, gotDomain, err := FindVPCEndpointByClusterTag(mockCli, region, clusterName)
			if tt.wantErr {
				assert.Error(t, err)
				if tt.errMsg != "" {
					assert.Contains(t, err.Error(), tt.errMsg)
				}
			} else {
				assert.NoError(t, err)
				assert.Equal(t, tt.wantID, gotID)
				assert.Equal(t, tt.wantDomain, gotDomain)
			}
		})
	}
}

func TestIsVPCEndpointReusableNative(t *testing.T) {
	svc := "com.aliyuncs.privatelink.cn-hangzhou.epsrv-test"
	vpc := "vpc-123"
	et := VpcEndpointTypeInterface

	tests := []struct {
		name         string
		mockEndpoint func() *privatelink.ListVpcEndpointsResponseBodyEndpoints
		want         bool
	}{
		{
			name: "active and matching",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Active"),
				}
			},
			want: true,
		},
		{
			name: "available and matching",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Available"),
				}
			},
			want: true,
		},
		{
			name: "nil status",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:  &svc,
					VpcId:        &vpc,
					EndpointType: &et,
					// EndpointStatus is nil
				}
			},
			want: false,
		},
		{
			name: "deleting state",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Deleting"),
				}
			},
			want: false,
		},
		{
			name: "deleted state",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Deleted"),
				}
			},
			want: false,
		},
		{
			name: "creating state",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Creating"),
				}
			},
			want: false,
		},
		{
			name: "failed state",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Failed"),
				}
			},
			want: false,
		},
		{
			name: "servicedeleted state",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("ServiceDeleted"),
				}
			},
			want: false,
		},
		{
			name: "mismatch type",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				wrongType := "Gateway"
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   &wrongType,
					EndpointStatus: ptr.To("Active"),
				}
			},
			want: false,
		},
		{
			name: "mismatch service",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				wrongSvc := "com.aliyuncs.privatelink.cn-hangzhou.epsrv-other"
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &wrongSvc,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Active"),
				}
			},
			want: false,
		},
		{
			name: "mismatch vpc",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				wrongVpc := "vpc-other"
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &wrongVpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Active"),
				}
			},
			want: false,
		},
		{
			name: "nil service name",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    nil,
					VpcId:          &vpc,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Active"),
				}
			},
			want: false,
		},
		{
			name: "nil vpc id",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          nil,
					EndpointType:   &et,
					EndpointStatus: ptr.To("Active"),
				}
			},
			want: false,
		},
		{
			name: "nil endpoint type",
			mockEndpoint: func() *privatelink.ListVpcEndpointsResponseBodyEndpoints {
				return &privatelink.ListVpcEndpointsResponseBodyEndpoints{
					ServiceName:    &svc,
					VpcId:          &vpc,
					EndpointType:   nil,
					EndpointStatus: ptr.To("Active"),
				}
			},
			want: false,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			assert.Equal(t, tt.want, isVPCEndpointReusableNative(tt.mockEndpoint(), et, svc, vpc))
		})
	}
}