git.giacomolli.uk - onechat/blob - whatsapp_batch.go

   1 package main
   2
   3 import (
   4         "slices"
   5         "strings"
   6         "time"
   7 )
   8
   9 const (
  10         // whatsappBatchEmptyReply is the last line of a batch that starts a
  11         // turn.
  12         whatsappBatchEmptyReply = "If nothing in this batch needs the user, end your turn with an empty reply."
  13
  14         // whatsappBatchDayLayout is the layout of the day in the first line of
  15         // a batch.
  16         whatsappBatchDayLayout = "Monday 2006-01-02"
  17 )
  18
  19 // WhatsAppBatch is one batch of the messages of one chat, with the arrival
  20 // times of its first and its last message.
  21 type WhatsAppBatch struct {
  22         Number   int               `json:"number"`
  23         Chat     WhatsAppChat      `json:"chat"`
  24         Closed   bool              `json:"closed"`
  25         First    time.Time         `json:"first"`
  26         Last     time.Time         `json:"last"`
  27         Messages []WhatsAppMessage `json:"messages"`
  28 }
  29
  30 // WhatsAppChat is a chat: its ID, its name, and whether it is a group.
  31 type WhatsAppChat struct {
  32         ID    string `json:"id"`
  33         Name  string `json:"name"`
  34         Group bool   `json:"group,omitempty"`
  35 }
  36
  37 // WhatsAppMessage is one message of a batch: its ID, the name of its sender,
  38 // its WhatsApp time, and its text. Mine marks a message of the user. Mention
  39 // and Reply mark a message that mentions the user or replies to a message of
  40 // the user.
  41 type WhatsAppMessage struct {
  42         ID      string    `json:"id"`
  43         Sender  string    `json:"sender"`
  44         Time    time.Time `json:"time"`
  45         Text    string    `json:"text"`
  46         Mine    bool      `json:"mine,omitempty"`
  47         Mention bool      `json:"mention,omitempty"`
  48         Reply   bool      `json:"reply,omitempty"`
  49 }
  50
  51 // Text returns the text of the batch for the log, with the times in location.
  52 // The first line starts with [WhatsApp] and names the chat, its ID, and the day
  53 // of the first message. Each message has a line with its time and its sender,
  54 // and the messages of the user have the sender You. A batch that starts a turn
  55 // ends with the line about the empty reply. Image markers start on their own
  56 // lines so that the shared image reader can find them.
  57 func (b WhatsAppBatch) Text(location *time.Location) string {
  58         kind := "private chat"
  59         if b.Chat.Group {
  60                 kind = "group"
  61         }
  62
  63         lines := []string{"[WhatsApp] " + b.Chat.Name + ", " + kind + " " + b.Chat.ID + ", " + b.Messages[0].Time.In(location).Format(whatsappBatchDayLayout)}
  64         for _, message := range b.Messages {
  65                 sender := message.Sender
  66                 switch {
  67                 case message.Mine:
  68                         sender = "You"
  69                 case message.Mention:
  70                         sender += " (mentions you)"
  71                 case message.Reply:
  72                         sender += " (replies to you)"
  73                 }
  74
  75                 prefix := message.Time.In(location).Format("15:04") + " " + sender + ":"
  76                 if len(imageFiles(message.Text)) > 0 {
  77                         lines = append(lines, prefix, message.Text)
  78                 } else {
  79                         lines = append(lines, prefix+" "+message.Text)
  80                 }
  81         }
  82
  83         if b.Turn() {
  84                 lines = append(lines, whatsappBatchEmptyReply)
  85         }
  86
  87         return strings.Join(lines, "\n")
  88 }
  89
  90 // Turn tells if the batch starts a turn: a private batch with a message of
  91 // another person, or a group batch with a message of another person that
  92 // mentions the user or replies to a message of the user.
  93 func (b WhatsAppBatch) Turn() bool {
  94         return slices.ContainsFunc(b.Messages, func(message WhatsAppMessage) bool {
  95                 return !message.Mine && (!b.Chat.Group || message.Mention || message.Reply)
  96         })
  97 }