package client

import (
	"net"

	"github.com/admin-else/strom/proto"
	"github.com/admin-else/strom/proto_base"
)

// Connect connects to the given address note that it does not resolve the SRV records it's just a raw net.Dial
func Connect(addr string) (ret *proto.Conn, err error) {
	s, err := Status(addr)
	if err != nil {
		return
	}
	ret, err = ConnectVersionLess(addr)
	if err != nil {
		return
	}
	err = ret.SetProtocolVersion(s.Version.Protocol)
	return
}

// ConnectVersionLess connects to the given address without setting the version you should set it
// when using this function.
func ConnectVersionLess(addr string) (ret *proto.Conn, err error) {
	ret = &proto.Conn{}
	ret.SetState(proto_base.Handshaking)
	ret.SetCompressionThreshold(-1)
	ret.Actor = proto_base.Client
	ret.Conn, err = net.Dial("tcp", addr)
	if err != nil {
		return
	}
	ret.R = ret.Conn
	ret.W = ret.Conn
	return
}