use InputPeerUser by custom username to 'user[%d]'

This commit is contained in:
Huanan 2025-04-29 19:07:40 +08:00
parent 911e761430
commit 98b3e1da4c
3 changed files with 11 additions and 3 deletions

View file

@ -35,12 +35,12 @@ func (c *Client) SendDraft(args DraftArguments) (*mcp.ToolResponse, error) {
Message: args.Text, Message: args.Text,
}) })
if err != nil { if err != nil {
return fmt.Errorf("failed to get history: %w", err) return fmt.Errorf("failed to send draft: %w", err)
} }
return nil return nil
}); err != nil { }); err != nil {
return nil, errors.Wrap(err, "failed to get history") return nil, errors.Wrap(err, "failed to send draft")
} }
jsonData, err := json.Marshal(DraftResponse{Success: ok}) jsonData, err := json.Marshal(DraftResponse{Success: ok})

View file

@ -30,7 +30,7 @@ func getUsername(source any) string {
var username string var username string
switch u := source.(type) { switch u := source.(type) {
case *tg.User: case *tg.User:
username = u.Username username = fmt.Sprintf("user[%d]", u.ID)
case *tg.Chat: case *tg.Chat:
username = fmt.Sprintf("cht[%d]", u.ID) username = fmt.Sprintf("cht[%d]", u.ID)
case *tg.Channel: case *tg.Channel:

View file

@ -89,6 +89,14 @@ func getInputPeerFromName(ctx context.Context, api *tg.Client, name string) (tg.
} }
return &chatPeer, nil return &chatPeer, nil
case strings.HasPrefix(name, "user") && isCustom:
var userPeer tg.InputPeerUser
_, err := fmt.Sscanf(name, "user[%d]", &userPeer.UserID)
if err != nil {
return nil, errors.Wrapf(err, "scan user peer(%q)", name)
}
return &userPeer, nil
default: default:
sender := message.NewSender(api) sender := message.NewSender(api)
inputPeer, err := sender.Resolve(name).AsInputPeer(ctx) inputPeer, err := sender.Resolve(name).AsInputPeer(ctx)