mikl/telegram-mcp-read-only
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add unread arg to dialogs

Browse source
This commit is contained in:
John Doe 2025-04-07 16:24:51 +03:00
parent e2eca96c46
commit 2d629fd3bd
2 changed files with 21 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
internal/tg/dialogs.go
View file

@ -29,7 +29,8 @@ const (
// nolint:lll // nolint:lll
type DialogsArguments struct { type DialogsArguments struct {
Offset string `json:"offset,omitempty" jsonschema:"description=Offset for continuation"` Offset string `json:"offset,omitempty" jsonschema:"description=Offset for continuation"`
OnlyUnread bool `json:"only_unread,omitempty" jsonschema:"description=Include only dialogs with unread mark"`
} }
type MessageInfo struct { type MessageInfo struct {
@ -87,7 +88,7 @@ func (c *Client) GetDialogs(args DialogsArguments) (*mcp.ToolResponse, error) {
return nil, errors.Wrap(err, "failed to get dialogs") return nil, errors.Wrap(err, "failed to get dialogs")
} }
d, err := newDialogs(dc) d, err := newDialogs(dc, args.OnlyUnread)
if err != nil { if err != nil {
return nil, errors.Wrap(err, "failed to get dialogs") return nil, errors.Wrap(err, "failed to get dialogs")
} }
@ -114,9 +115,12 @@ type dialogs struct {
//dialogs map[string]*tg.Dialog //dialogs map[string]*tg.Dialog
chats map[int64]*tg.Chat chats map[int64]*tg.Chat
channels map[int64]*tg.Channel channels map[int64]*tg.Channel
//opts
onlyUnread bool
} }
func newDialogs(rawD tg.MessagesDialogsClass) (*dialogs, error) { func newDialogs(rawD tg.MessagesDialogsClass, onlyUnread bool) (*dialogs, error) {
var d dialogs var d dialogs
switch dT := rawD.(type) { switch dT := rawD.(type) {
case *tg.MessagesDialogs: case *tg.MessagesDialogs:
@ -167,6 +171,8 @@ func newDialogs(rawD tg.MessagesDialogsClass) (*dialogs, error) {
} }
} }
d.onlyUnread = onlyUnread
return &d, nil return &d, nil
} }
@ -174,7 +180,16 @@ func (d *dialogs) Info() []DialogInfo {
ds := make([]DialogInfo, 0, len(d.Dialogs)) ds := make([]DialogInfo, 0, len(d.Dialogs))
for _, dItem := range d.Dialogs { for _, dItem := range d.Dialogs {
info, err := d.processDialog(dItem) dialogItem, ok := dItem.(*tg.Dialog)
if !ok {
continue
}
if d.onlyUnread && dialogItem.UnreadCount == 0 {
continue
}
info, err := d.processDialog(dialogItem)
if err != nil { if err != nil {
log.Debug().Err(err).Str("dialog", dItem.String()).Msg("failed process dialog") log.Debug().Err(err).Str("dialog", dItem.String()).Msg("failed process dialog")
continue continue
@ -215,12 +230,7 @@ func (d *dialogs) Offset() DialogsOffset {
return DialogsOffset{} return DialogsOffset{}
} }
func (d *dialogs) processDialog(rawD tg.DialogClass) (DialogInfo, error) { func (d *dialogs) processDialog(dialogItem *tg.Dialog) (DialogInfo, error) {
dialogItem, ok := rawD.(*tg.Dialog)
if !ok {
return DialogInfo{}, errors.Errorf("newDialogs(%T): invalid dialog type", rawD)
}
var info DialogInfo var info DialogInfo
if msg, ok := d.messages[getPeerID(dialogItem.Peer)]; ok { if msg, ok := d.messages[getPeerID(dialogItem.Peer)]; ok {

2
serve.go
View file

@ -41,7 +41,7 @@ func serve(ctx context.Context, cmd *cli.Command) error {
log.Info().RawJSON("answer", data).Msg("Check GetMe: OK") log.Info().RawJSON("answer", data).Msg("Check GetMe: OK")
answer, err = client.GetDialogs(tg.DialogsArguments{Offset: ""}) answer, err = client.GetDialogs(tg.DialogsArguments{Offset: "", OnlyUnread: true})
if err != nil { if err != nil {
return fmt.Errorf("get dialogs: %w", err) return fmt.Errorf("get dialogs: %w", err)
} }