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 // Turn tells if the batch starts a turn: a private batch with a message of
  52 // another person, or a group batch with a message of another person that
  53 // mentions the user or replies to a message of the user.
  54 func (b WhatsAppBatch) Turn() bool {
  55         return slices.ContainsFunc(b.Messages, func(message WhatsAppMessage) bool {
  56                 return !message.Mine && (!b.Chat.Group || message.Mention || message.Reply)
  57         })
  58 }
  59
  60 // Text returns the text of the batch for the log, with the times in location.
  61 // The first line starts with [WhatsApp] and names the chat, its ID, and the day
  62 // of the first message. Each message has a line with its time and its sender,
  63 // and the messages of the user have the sender You. A batch that starts a turn
  64 // ends with the line about the empty reply. Image markers start on their own
  65 // lines so that the shared image reader can find them.
  66 func (b WhatsAppBatch) Text(location *time.Location) string {
  67         kind := "private chat"
  68         if b.Chat.Group {
  69                 kind = "group"
  70         }
  71
  72         lines := []string{"[WhatsApp] " + b.Chat.Name + ", " + kind + " " + b.Chat.ID + ", " + b.Messages[0].Time.In(location).Format(whatsappBatchDayLayout)}
  73         for _, message := range b.Messages {
  74                 sender := message.Sender
  75                 switch {
  76                 case message.Mine:
  77                         sender = "You"
  78                 case message.Mention:
  79                         sender += " (mentions you)"
  80                 case message.Reply:
  81                         sender += " (replies to you)"
  82                 }
  83
  84                 prefix := message.Time.In(location).Format("15:04") + " " + sender + ":"
  85                 if len(imageFiles(message.Text)) > 0 {
  86                         lines = append(lines, prefix, message.Text)
  87                 } else {
  88                         lines = append(lines, prefix+" "+message.Text)
  89                 }
  90         }
  91
  92         if b.Turn() {
  93                 lines = append(lines, whatsappBatchEmptyReply)
  94         }
  95
  96         return strings.Join(lines, "\n")
  97 }