package utils

import (
	"fmt"
	"net"
)

func InitTCP(conn net.Conn, readBufferSize, writeBufferSize int) error {
	tcpConn := conn.(*net.TCPConn) //nolint: forcetypeassert

	if err := tcpConn.SetNoDelay(true); err != nil {
		return fmt.Errorf("cannot set TCP_NO_DELAY: %w", err)
	}

	if err := tcpConn.SetReadBuffer(readBufferSize); err != nil {
		return fmt.Errorf("cannot set read buffer size: %w", err)
	}

	if err := tcpConn.SetWriteBuffer(writeBufferSize); err != nil {
		return fmt.Errorf("cannot set write buffer size: %w", err)
	}

	if err := tcpConn.SetKeepAlive(true); err != nil {
		return fmt.Errorf("cannot enable keep-alive: %w", err)
	}

	return nil
}