gorc/client/client.go

// gorc project
// Copyright (C) 2021 IllusionMan1212
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License along
// with this program; if not, see https://www.gnu.org/licenses.

package client

import (
	"crypto/tls"
	"fmt"
	"log"
	"strings"
)

type Client struct {
	// tcp connection
	Conn *tls.Conn
}

const CRLF = "\r\n"

func NewClient(host string, port string, tlsEnabled bool) *Client {
	// TODO: distinguish between secure and insecure connections
	addr := fmt.Sprintf("%s:%s", host, port)

	cfg := &tls.Config{ServerName: host}
	conn, err := tls.Dial("tcp", addr, cfg)
	if err != nil {
		// TODO: properly handle the error instead of Fatal-ing (failed to initiate a connection to server)
		log.Fatal(err)
	}

	return &Client{
		Conn: conn,
	}
}

func (c Client) Register(nick string, password string, channel string) {
	if c.Conn == nil {
		// TODO: properly handle the error instead of Fatal-ing
		log.Fatal("Attempted to write data to nil connection")
	}

	c.SendCommand("CAP", "LS")
	if password != "" {
		c.SendCommand("PASS", password)
	}
	// TODO: check if nickname has spaces and remove them
	c.SendCommand("NICK", nick)
	c.SendCommand("USER", nick, "0", "*", nick)
	// TODO: CAP REQ :whatever capability the client recognizes and supports
	c.SendCommand("CAP", "END")
	c.SendCommand("JOIN", channel)
}

func (c Client) SendCommand(cmd string, params ...string) {
	paramsString := ""
	if len(params) > 0 {
		paramsString = " " + strings.Join(params, " ")
	}

	// if we have more than 1 param then replace the last param's space with a " :"
	if len(params) > 1 {
		i := strings.LastIndex(paramsString, " ")
		paramsString = paramsString[:i] + strings.Replace(paramsString[i:], " ", " :", 1)
		// if we have more exactly 1 param and it contains spaces, we prepend colon to the param
	} else if len(params) == 1 && strings.Contains(params[0], " ") {
		paramsString = ":" + paramsString
	}

	c.Conn.Write([]byte(cmd + paramsString + CRLF))
}