onyx_online/q3rcon
1
0
Fork 0
mirror of https://github.com/onyx-and-iris/q3rcon.git synced 2025-01-18 02:30:47 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

{Rcon}.login no longer exported.

Browse Source 
split the cmd from its args when checking timeout map.

remove final TrimPrefix, it's already handled in {Rcon}.listen()
This commit is contained in:
onyx-and-iris 2024-11-24 13:13:07 +00:00
parent 65ab17b9c9
commit 3e1088d625
2 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@ -25,4 +25,4 @@ go.work.sum
# env file
.env
cmd/aeiou
cmd/codrcon

15
q3rcon.go
View File

@ -7,6 +7,8 @@ import (
"strings"
"time"
log "github.com/sirupsen/logrus"
"github.com/onyx-and-iris/q3rcon/internal/conn"
"github.com/onyx-and-iris/q3rcon/internal/packet"
)
@ -47,14 +49,14 @@ func New(host string, port int, password string, options ...Option) (*Rcon, erro
o(r)
}
if err = r.Login(); err != nil {
if err = r.login(); err != nil {
return nil, err
}
return r, nil
}
func (r Rcon) Login() error {
func (r Rcon) login() error {
timeout := time.After(r.loginTimeout)
for {
select {
@ -78,10 +80,13 @@ func (r Rcon) Login() error {
}
}
func (r Rcon) Send(cmd string) (string, error) {
func (r Rcon) Send(cmdWithArgs string) (string, error) {
cmd, _, _ := strings.Cut(string(cmdWithArgs), " ")
timeout, ok := r.timeouts[cmd]
if !ok {
timeout = r.defaultTimeout
} else {
log.Debugf("%s in timeouts map, using timeout %v", cmd, timeout)
}
respChan := make(chan string)
@ -89,7 +94,7 @@ func (r Rcon) Send(cmd string) (string, error) {
go r.listen(timeout, respChan, errChan)
_, err := r.conn.Write(r.request.Encode(cmd))
_, err := r.conn.Write(r.request.Encode(cmdWithArgs))
if err != nil {
return "", err
}
@ -98,7 +103,7 @@ func (r Rcon) Send(cmd string) (string, error) {
case err := <-errChan:
return "", err
case resp := <-respChan:
return strings.TrimPrefix(resp, string(r.response.Header())), nil
return resp, nil
}
}