package goSam

import (
	"net"
	"strings"
)

// Dial implements the net.Dial function and can be used for http.Transport
func (c *Client) Dial(network, addr string) (net.Conn, error) {
	portIdx := strings.Index(addr, ":")
	if portIdx >= 0 {
		addr = addr[:portIdx]
	}
	addr, err := c.Lookup(addr)
	if err != nil {
		return nil, err
	}

	id, _, err := c.CreateStreamSession("")
	if err != nil {
		return nil, err
	}

	newC, err := NewClient(c.samaddr())
	if err != nil {
		return nil, err
	}

	err = newC.StreamConnect(id, addr)
	if err != nil {
		return nil, err
	}

	return newC.SamConn, nil
}