X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/786c2b0551517d4befaa35be2c7743f65bd5f1c4..016d46008fc9726d2617e4e3ea098113f34c7636:/whatsapp.go?ds=inline

diff --git a/whatsapp.go b/whatsapp.go
index c3db573..6e1e1e6 100644
--- a/whatsapp.go
+++ b/whatsapp.go
@@ -116,6 +116,7 @@ func openWhatsApp(ctx context.Context, dir string, clock func() time.Time, logge
 	}
 
 	var err error
+
 	w.state, err = openWhatsAppState(filepath.Join(w.dir, whatsappStateFileName), whatsappBatchQuiet, whatsappBatchWindow)
 	if err != nil {
 		return nil, err
@@ -144,6 +145,7 @@ func openWhatsApp(ctx context.Context, dir string, clock func() time.Time, logge
 	if w.device.ID != nil {
 		w.linked = true
 		w.inbox.waitForCatchup()
+
 		w.own = whatsappOwnIDs(w.device.ID.User, w.device.LID.User)
 	}
 
@@ -215,6 +217,7 @@ type WhatsApp struct {
 // would be lost.
 func (w *WhatsApp) Run(ctx context.Context, l *TurnLoop, notify func(Event)) error {
 	w.mutex.Lock()
+
 	w.notify = notify
 	linked := w.linked
 	w.retrying = linked
@@ -225,11 +228,13 @@ func (w *WhatsApp) Run(ctx context.Context, l *TurnLoop, notify func(Event)) err
 	defer timer.Stop()
 
 	var retry <-chan time.Time
+
 	delay := whatsappRetryDelay
 	connect := func() {
 		retry = nil
 		if w.connect(ctx) {
 			timer.Reset(delay)
+
 			retry = timer.C
 			delay = min(2*delay, whatsappRetryLimit)
 		}
@@ -237,13 +242,17 @@ func (w *WhatsApp) Run(ctx context.Context, l *TurnLoop, notify func(Event)) err
 
 	stopRetry := func() {
 		timer.Stop()
+
 		retry = nil
+
 		w.mutex.Lock()
+
 		w.retrying = false
 		w.mutex.Unlock()
 	}
 
 	var pairing sync.WaitGroup
+
 	cancel := func() {}
 	defer func() {
 		cancel()
@@ -272,6 +281,7 @@ func (w *WhatsApp) Run(ctx context.Context, l *TurnLoop, notify func(Event)) err
 			connect()
 		case <-w.logouts:
 			stopRetry()
+
 			err = w.loggedOut(ctx, l)
 			if err != nil {
 				return err
@@ -279,11 +289,13 @@ func (w *WhatsApp) Run(ctx context.Context, l *TurnLoop, notify func(Event)) err
 		case request := <-w.requests:
 			cancel()
 			pairing.Wait()
+
 			cancel = func() {}
 
 			switch {
 			case request.unlink:
 				stopRetry()
+
 				err = w.unlink(ctx, l, request.answer)
 				if err != nil {
 					return err
@@ -315,6 +327,7 @@ func (w *WhatsApp) Command(args string, answer func(text string)) {
 	switch {
 	case len(fields) == 1 && fields[0] == "status":
 		answer(w.Status())
+
 		return
 	case len(fields) == 1 && fields[0] == "unlink":
 		request.unlink = true
@@ -322,12 +335,14 @@ func (w *WhatsApp) Command(args string, answer func(text string)) {
 		phone, valid := whatsappPhone(strings.Join(fields[1:], ""))
 		if !valid {
 			answer("PHONE must be the number of your WhatsApp account with its country code: 7 to 15 digits, such as +15550001111.")
+
 			return
 		}
 
 		request.phone = phone
 	default:
 		answer(whatsappUsage)
+
 		return
 	}
 
@@ -442,8 +457,10 @@ func (w *WhatsApp) link() WhatsAppLink {
 // authentication.
 func (w *WhatsApp) connect(ctx context.Context) bool {
 	w.mutex.Lock()
+
 	if ctx.Err() != nil || !w.linked || !w.retrying || w.closed {
 		w.mutex.Unlock()
+
 		return false
 	}
 
@@ -455,16 +472,20 @@ func (w *WhatsApp) connect(ctx context.Context) bool {
 	w.mutex.Unlock()
 
 	err := client.ConnectContext(ctx)
+
 	w.mutex.Lock()
+
 	if err == nil || ctx.Err() != nil || !w.retrying || !w.linked {
 		w.retrying = false
 		w.mutex.Unlock()
+
 		return false
 	}
 
 	first := w.startOutage()
 	cause := whatsappRedact(w.own, err.Error())
 	w.mutex.Unlock()
+
 	if first {
 		w.logger.Printf("first connect failed, retrying: %s", cause)
 	}
@@ -493,6 +514,7 @@ func (w *WhatsApp) startOutage() bool {
 // close ends the connection and the timer of the notice, and stops all notices.
 func (w *WhatsApp) close() {
 	w.mutex.Lock()
+
 	w.closed = true
 	if w.timer != nil {
 		w.timer.Stop()
@@ -517,6 +539,7 @@ func (w *WhatsApp) loggedOut(ctx context.Context, l *TurnLoop) error {
 	}
 
 	w.logger.Print("logged out by the phone, run /whatsapp pair PHONE")
+
 	err := w.inbox.deliver(ctx, l, w.link(), true)
 	if err != nil {
 		return err
@@ -544,12 +567,14 @@ func (w *WhatsApp) loggedOut(ctx context.Context, l *TurnLoop) error {
 // file.
 func (w *WhatsApp) unlink(ctx context.Context, l *TurnLoop, answer func(text string)) error {
 	w.mutex.Lock()
+
 	linked := w.linked
 	client := w.client
 	w.mutex.Unlock()
 
 	if !linked {
 		answer("WhatsApp is not linked.")
+
 		return nil
 	}
 
@@ -559,6 +584,7 @@ func (w *WhatsApp) unlink(ctx context.Context, l *TurnLoop, answer func(text str
 	}
 
 	var logoutErr error
+
 	if client != nil {
 		ctx, cancel := context.WithTimeout(ctx, whatsappLogoutTime)
 		defer cancel()
@@ -569,10 +595,12 @@ func (w *WhatsApp) unlink(ctx context.Context, l *TurnLoop, answer func(text str
 	err = w.removeSession()
 	if err != nil {
 		answer("The unlink failed: " + w.redact(err.Error()))
+
 		return nil
 	}
 
 	w.logger.Print("unlinked")
+
 	err = l.note("WhatsApp is unlinked.")
 	if err != nil {
 		w.logger.Printf("chat note failed: %v", err)
@@ -580,6 +608,7 @@ func (w *WhatsApp) unlink(ctx context.Context, l *TurnLoop, answer func(text str
 
 	if client == nil || logoutErr != nil {
 		answer("WhatsApp is unlinked here, but the phone did not get the logout. In WhatsApp on your phone, open Linked devices and remove the device.")
+
 		return nil
 	}
 
@@ -592,6 +621,7 @@ func (w *WhatsApp) unlink(ctx context.Context, l *TurnLoop, answer func(text str
 // its files.
 func (w *WhatsApp) removeSession() error {
 	w.mutex.Lock()
+
 	client := w.client
 	w.client = nil
 	w.linked = false
@@ -616,6 +646,7 @@ func (w *WhatsApp) removeSession() error {
 	}
 
 	var err error
+
 	if container != nil {
 		err = container.Close()
 	}
@@ -645,6 +676,7 @@ func (w *WhatsApp) pair(ctx, runCtx context.Context, l *TurnLoop, phone string,
 	device, err := w.newDevice(ctx)
 	if err != nil {
 		answer("The pairing failed: " + err.Error())
+
 		return
 	}
 
@@ -663,6 +695,7 @@ func (w *WhatsApp) pair(ctx, runCtx context.Context, l *TurnLoop, phone string,
 
 	if err != nil {
 		answer("The connection to WhatsApp failed: " + err.Error())
+
 		return
 	}
 
@@ -670,6 +703,7 @@ func (w *WhatsApp) pair(ctx, runCtx context.Context, l *TurnLoop, phone string,
 	case item, open := <-qr:
 		if !open || item.Event != whatsmeow.QRChannelEventCode {
 			answer("WhatsApp did not start the pairing. Try again later.")
+
 			return
 		}
 	case <-ctx.Done():
@@ -685,6 +719,7 @@ func (w *WhatsApp) pair(ctx, runCtx context.Context, l *TurnLoop, phone string,
 	code, err := client.PairPhone(ctx, phone, true, whatsmeow.PairClientChrome, "Chrome (Linux)")
 	if err != nil {
 		answer("WhatsApp refused the pairing: " + err.Error() + ". Check the phone number and run /whatsapp pair PHONE again.")
+
 		return
 	}
 
@@ -693,6 +728,7 @@ func (w *WhatsApp) pair(ctx, runCtx context.Context, l *TurnLoop, phone string,
 
 	for {
 		var item whatsmeow.QRChannelItem
+
 		open := true
 		select {
 		case item, open = <-qr:
@@ -707,11 +743,13 @@ func (w *WhatsApp) pair(ctx, runCtx context.Context, l *TurnLoop, phone string,
 		switch {
 		case !open || item.Event == whatsmeow.QRChannelTimeout.Event:
 			answer(whatsappPairExpired)
+
 			return
 		case item.Event == whatsmeow.QRChannelEventCode:
 			continue
 		case item.Event == whatsmeow.QRChannelSuccess.Event:
 			linked = true
+
 			w.linkedBy(client)
 			w.logger.Print("linked")
 
@@ -725,6 +763,7 @@ func (w *WhatsApp) pair(ctx, runCtx context.Context, l *TurnLoop, phone string,
 			return
 		default:
 			answer("The pairing failed: " + whatsappItemError(item) + ". Run /whatsapp pair PHONE again.")
+
 			return
 		}
 	}
@@ -797,16 +836,20 @@ func (w *WhatsApp) handle(event any) bool {
 		return true
 	case *events.JoinedGroup:
 		w.groups.set(event.JID.String(), event.Name)
+
 		return true
 	case *events.OfflineSyncCompleted:
 		w.inbox.endCatchup()
+
 		return true
 	case *events.Connected:
 		w.inbox.startCatchup()
 	}
 
 	var notice, line string
+
 	w.mutex.Lock()
+
 	switch event := event.(type) {
 	case *events.PairSuccess:
 		w.own = whatsappOwnIDs(event.ID.User, event.LID.User)
@@ -875,6 +918,7 @@ func (w *WhatsApp) handle(event any) bool {
 // disconnection with the number disconnections ended.
 func (w *WhatsApp) disconnectedLong(disconnections int) {
 	w.mutex.Lock()
+
 	long := !w.closed && w.linked && !w.connected && !w.noticed && w.disconnections == disconnections
 	if long {
 		w.noticed = true
@@ -891,6 +935,7 @@ func (w *WhatsApp) disconnectedLong(disconnections int) {
 // broadcast gives text as a notice to all clients.
 func (w *WhatsApp) broadcast(text string) {
 	w.mutex.Lock()
+
 	notify := w.notify
 	closed := w.closed
 	w.mutex.Unlock()
@@ -935,6 +980,7 @@ type WhatsAppRequest struct {
 // name OneChat and a desktop icon only when the platform type is known.
 func whatsappSetDeviceProps() {
 	store.SetOSInfo("OneChat", [3]uint32{0, 1, 0})
+
 	store.DeviceProps.PlatformType = waCompanionReg.DeviceProps_DESKTOP.Enum()
 }
 
@@ -998,6 +1044,7 @@ func whatsappContactName(ctx context.Context, device *store.Device, jid types.JI
 // whatsappOwnIDs returns the own number and the own LID that are not empty.
 func whatsappOwnIDs(ids ...string) []string {
 	var own []string
+
 	for _, id := range ids {
 		if id != "" {
 			own = append(own, id)