package ipv4

import (
	"crypto/tls"
	"fmt"
	"net"
	"time"
)

type serviceConnector interface {
	Type() string
	Dial(ip string, port int) (net.Conn, error)
}

type normalServiceConn struct {
	t string
}

func (nsc *normalServiceConn) Type() string {
	return nsc.t
}

func (nsc *normalServiceConn) Dial(ip string, port int) (net.Conn, error) {
	addr := fmt.Sprintf("%s:%d", ip, port)
	conn, err := net.DialTimeout("tcp", addr, time.Duration(3)*time.Second)
	if err != nil {
		return nil, err
	}

	err = conn.SetDeadline(time.Now().Add(3 * time.Second))
	if err != nil {
		return nil, err
	}
	return conn, err
}

type tlsServiceConn struct {
	t string
}

func (tsc *tlsServiceConn) Type() string {
	return tsc.t
}

func (tsc *tlsServiceConn) Dial(ip string, port int) (net.Conn, error) {
	addr := fmt.Sprintf("%s:%d", ip, port)
	dialer := &net.Dialer{
		Timeout: 3 * time.Second,
	}
	conn, err := tls.DialWithDialer(
		dialer,
		"tcp",
		addr,
		&tls.Config{
			InsecureSkipVerify: true,
			ServerName:         ip,
		},
	)
	if err != nil {
		return nil, err
	}

	err = conn.SetDeadline(time.Now().Add(3 * time.Second))
	if err != nil {
		return nil, err
	}
	return conn, err
}