migrate additional dependencies to ergochat

vendor/github.com/ergochat/go-ident/client.go

@@ -0,0 +1,107 @@
+// Package ident implements an RFC 1413 client
+package ident
+
+import (
+	"bufio"
+	"fmt"
+	"net"
+	"strings"
+	"time"
+)
+
+// Response is a successful answer to our query to the identd server.
+type Response struct {
+	OS         string
+	Charset    string
+	Identifier string
+}
+
+// ResponseError indicates that the identd server returned an error rather than an
+// identifying string.
+type ResponseError struct {
+	Type string
+}
+
+func (e ResponseError) Error() string {
+	return fmt.Sprintf("Ident error: %s", e.Type)
+}
+
+// ProtocolError indicates that an error occurred with the protocol itself, that the response
+// could not be successfully parsed or was malformed.
+type ProtocolError struct {
+	Line string
+}
+
+func (e ProtocolError) Error() string {
+	return fmt.Sprintf("Unexpected response from server: %s", e.Line)
+}
+
+// Query makes an Ident query, if timeout is >0 the query is timed out after that many seconds.
+func Query(ip string, portOnServer, portOnClient int, timeout time.Duration) (response Response, err error) {
+	var conn net.Conn
+	if timeout > 0 {
+		conn, err = net.DialTimeout("tcp", net.JoinHostPort(ip, "113"), timeout)
+	} else {
+		conn, err = net.Dial("tcp", net.JoinHostPort(ip, "113"))
+	}
+	if err != nil {
+		return
+	}
+
+	// stop the ident read after <timeout> seconds
+	if timeout > 0 {
+		conn.SetDeadline(time.Now().Add(timeout))
+	}
+
+	_, err = conn.Write([]byte(fmt.Sprintf("%d, %d", portOnClient, portOnServer) + "\r\n"))
+	if err != nil {
+		return
+	}
+
+	r := bufio.NewReaderSize(conn, 1024)
+	respBytes, err := r.ReadSlice('\n')
+	if err != nil {
+		return
+	}
+	resp := string(respBytes)
+
+	fields := strings.SplitN(resp, ":", 4)
+	if len(fields) < 3 {
+		return response, ProtocolError{resp}
+	}
+	for i, field := range fields {
+		fields[i] = strings.TrimSpace(field)
+	}
+
+	switch fields[1] {
+	case "USERID":
+		if len(fields) != 4 {
+			return response, ProtocolError{resp}
+		}
+
+		var os, charset string
+		osAndCharset := strings.SplitN(fields[2], ",", 2)
+		if len(osAndCharset) == 2 {
+			os = osAndCharset[0]
+			charset = osAndCharset[1]
+		} else {
+			os = osAndCharset[0]
+			charset = "US-ASCII"
+		}
+
+		return Response{
+			OS:         os,
+			Charset:    charset,
+			Identifier: fields[3],
+		}, nil
+	case "ERROR":
+		if len(fields) != 3 {
+			return response, ProtocolError{resp}
+		}
+
+		return response, ResponseError{fields[2]}
+	default:
+		err = ProtocolError{resp}
+	}
+	return
+}