nimona/go-nimona

package nimona

import (
    "context"
    "fmt"
    "time"
)

type (
    Ping struct {
        _     string `nimona:"$type,type=test/ping"`
        Nonce string `nimona:"nonce"`
    }
    Pong struct {
        _     string `nimona:"$type,type=test/pong"`
        Nonce string `nimona:"nonce"`
    }
)

func RequestPing(
    ctx context.Context,
    ses *Session,
) (*Pong, error) {
    req := &Ping{}
    res := &Pong{}
    msgRes, err := ses.Request(ctx, req.Document())
    if err != nil {
        return nil, fmt.Errorf("error sending message: %w", err)
    }
    err = res.FromDocument(msgRes.Document)
    if err != nil {
        return nil, fmt.Errorf("error decoding message: %w", err)
    }
    return res, nil
}

func HandlePingRequest(
    sesManager *SessionManager,
) {
    handler := func(
        ctx context.Context,
        msg *Request,
    ) error {
        req := Ping{}
        err := req.FromDocument(msg.Document)
        if err != nil {
            return fmt.Errorf("error unmarshaling request: %w", err)
        }
        res := &Pong{
            Nonce: time.Now().Format(time.RFC3339Nano),
        }
        err = msg.Respond(res.Document())
        if err != nil {
            return fmt.Errorf("error replying: %w", err)
        }
        fmt.Println("Handled ping request")
        return nil
    }
    sesManager.RegisterHandler("test/ping", handler)
}