79 lines
2.0 KiB
Go
79 lines
2.0 KiB
Go
|
|
||
|
|
/*
|
||
|
|
* Copyright (c) 2026 Kaiy Ragur
|
||
|
|
*
|
||
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
||
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
||
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
||
|
|
*/
|
||
|
|
package remote
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/ed25519"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/neilalexander/yggmail/internal/notify"
|
||
|
|
"github.com/neilalexander/yggmail/internal/remote"
|
||
|
|
pb "github.com/neilalexander/yggmail/internal/remote/types"
|
||
|
|
"github.com/neilalexander/yggmail/internal/transport"
|
||
|
|
"github.com/neilalexander/yggmail/internal/utils"
|
||
|
|
)
|
||
|
|
|
||
|
|
type RemoteNotifer struct {
|
||
|
|
Client *remote.Client
|
||
|
|
targetNode ed25519.PublicKey
|
||
|
|
userPk ed25519.PublicKey
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewRemoteNotifer(targetNode, userPk ed25519.PublicKey, signer remote.Signer, tr transport.Transport) *RemoteNotifer {
|
||
|
|
return &RemoteNotifer{
|
||
|
|
Client: remote.NewClient(userPk, signer, tr),
|
||
|
|
targetNode: targetNode,
|
||
|
|
userPk: userPk,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
func (n *RemoteNotifer) Watch(pk, mailbox string) (uint32, uint32, error) {
|
||
|
|
resp, _, err := n.Client.Call(n.targetNode, &pb.Request{
|
||
|
|
Call: &pb.Request_MailWatch{
|
||
|
|
MailWatch: &pb.WatchRequest{Mailbox: mailbox},
|
||
|
|
},
|
||
|
|
}, nil, 0)
|
||
|
|
if err != nil {
|
||
|
|
return 0, 0, err
|
||
|
|
}
|
||
|
|
if !resp.Success {
|
||
|
|
return 0, 0, fmt.Errorf("remote error: %s", resp.Error)
|
||
|
|
}
|
||
|
|
if resp.Error == "HEARTBIT" {
|
||
|
|
return 0, 0, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
status := resp.GetMboxStatus()
|
||
|
|
if status != nil {
|
||
|
|
return status.UidNext, status.MessagesCount, nil
|
||
|
|
}
|
||
|
|
return 0, 0, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (n *RemoteNotifer) StartRemoteNotifer(nf *notify.Notify, pk ed25519.PublicKey, mailbox string) {
|
||
|
|
go func() {
|
||
|
|
for {
|
||
|
|
pk := utils.EncodeString(n.userPk)
|
||
|
|
uidNext, count, err := n.Watch(pk, mailbox)
|
||
|
|
if err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if uidNext == 0 && count == 0 && err == nil {
|
||
|
|
// fmt.Println("RPC Notify: HEARTBIT: Pk", pk)
|
||
|
|
// Heartbit: Update connection
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
if nf != nil {
|
||
|
|
fmt.Println("RPC Notify: Notify: Pk", pk)
|
||
|
|
ev := notify.NewEmailEvent(pk, mailbox, int64(uidNext), int64(count))
|
||
|
|
go nf.NotifyNew(pk, ev)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
}
|