diff --git a/commands.go b/commands.go index b7d51c3..5e1eb2c 100644 --- a/commands.go +++ b/commands.go @@ -18,6 +18,8 @@ package main import ( "fmt" + "math" + "sort" "strconv" "strings" @@ -528,18 +530,65 @@ func (handler *CommandHandler) CommandDeleteAllPortals(ce *CommandEvent) { const cmdListHelp = `list - Get a list of all contacts and groups.` -func (handler *CommandHandler) CommandList(ce *CommandEvent) { - var contacts strings.Builder - var groups strings.Builder +func formatContacts(contacts bool, skip, max int, input map[string]whatsapp.Contact) (result []string, total int) { + skipped := 0 + for jid, contact := range input { + if strings.HasSuffix(jid, whatsappExt.NewUserSuffix) != contacts { + continue + } - for jid, contact := range ce.User.Conn.Store.Contacts { - if strings.HasSuffix(jid, whatsappExt.NewUserSuffix) { - _, _ = fmt.Fprintf(&contacts, "* %s / %s - `%s`\n", contact.Name, contact.Notify, contact.Jid[:len(contact.Jid)-len(whatsappExt.NewUserSuffix)]) + total++ + if skipped < skip || len(result) >= max { + skipped++ + continue + } + + if contacts { + result = append(result, fmt.Sprintf("* %s / %s - `%s`", contact.Name, contact.Notify, contact.Jid[:len(contact.Jid)-len(whatsappExt.NewUserSuffix)])) } else { - _, _ = fmt.Fprintf(&groups, "* %s - `%s`\n", contact.Name, contact.Jid) + result = append(result, fmt.Sprintf("* %s - `%s`", contact.Name, contact.Jid)) } } - ce.Reply("### Contacts\n%s\n\n### Groups\n%s", contacts.String(), groups.String()) + sort.Sort(sort.StringSlice(result)) + return +} + +func (handler *CommandHandler) CommandList(ce *CommandEvent) { + if len(ce.Args) == 0 { + ce.Reply("**Usage:** `list [page] [items per page]`") + return + } + mode := strings.ToLower(ce.Args[0]) + if mode[0] != 'g' && mode[0] != 'c' { + ce.Reply("**Usage:** `list [page] [items per page]`") + } + var err error + page := 0 + max := 100 + if len(ce.Args) > 1 { + page, err = strconv.Atoi(ce.Args[1]) + if err != nil || page <= 0 { + ce.Reply("\"%s\" isn't a valid page number", ce.Args[1]) + return + } + } + if len(ce.Args) > 2 { + max, err = strconv.Atoi(ce.Args[2]) + if err != nil || max <= 0 { + ce.Reply("\"%s\" isn't a valid number of items per page", ce.Args[2]) + return + } else if max > 400 { + ce.Reply("Warning: a high number of items per page may fail to send a reply") + } + } + contacts := mode[0] == 'c' + typeName := "Groups" + if contacts { + typeName = "Contacts" + } + result, totalItems := formatContacts(contacts, (page-1)*max, max, ce.User.Conn.Store.Contacts) + pages := int(math.Ceil(float64(totalItems) / float64(max))) + ce.Reply("### %s (page %d of %d)\n\n%s", typeName, page, pages, strings.Join(result, "\n")) } const cmdOpenHelp = `open <_group JID_> - Open a group chat portal.`