gochat/internal/service/conversation_maintenance_worker.go

package service

import (
	"context"
	"encoding/json"
	"fmt"
	"strings"
	"sync"
	"time"

	"github.com/gochat/gochat/internal/campaign"
	"github.com/gochat/gochat/internal/model"
	"github.com/gochat/gochat/internal/worker"
	"gorm.io/gorm"
)

const (
	TaskTypeScheduledTriggerItems            = "scheduled:trigger_items"
	TaskTypeCampaignTriggerOneoff            = "campaign:trigger_oneoff"
	TaskTypeConversationReopenSnoozed        = "conversation:reopen_snoozed"
	TaskTypeConversationResolutionScheduler  = "account:conversations_resolution_scheduler"
	TaskTypeConversationResolutionForAccount = "conversation:resolution"
	TaskTypeConversationUpdateMessageStatus  = "conversation:update_message_status"
	TaskTypeConversationBulkAction           = "conversation:bulk_action"
)

const (
	scheduledItemsInterval      = time.Hour
	scheduledItemsLookback      = 3 * 24 * time.Hour
	conversationResolutionLimit = 100
)

type campaignTriggerOneoffJob struct {
	CampaignID uint `json:"campaign_id"`
}

type conversationResolutionJob struct {
	AccountID uint `json:"account_id"`
}

type conversationUpdateMessageStatusJob struct {
	ConversationID uint   `json:"conversation_id"`
	Timestamp      int64  `json:"timestamp"`
	Status         string `json:"status"`
}

type ConversationBulkActionParams struct {
	Type         string                       `json:"type"`
	ActionName   string                       `json:"action_name,omitempty"`
	IDs          []uint                       `json:"ids"`
	Fields       ConversationBulkActionFields `json:"fields,omitempty"`
	Labels       ConversationBulkActionLabels `json:"labels,omitempty"`
	SnoozedUntil string                       `json:"snoozed_until,omitempty"`
}

type ConversationBulkActionFields struct {
	Status     *string `json:"status,omitempty"`
	AssigneeID *uint   `json:"assignee_id,omitempty"`
	TeamID     *uint   `json:"team_id,omitempty"`
}

type ConversationBulkActionLabels struct {
	Add    []string `json:"add,omitempty"`
	Remove []string `json:"remove,omitempty"`
}

type conversationBulkActionJob struct {
	AccountID uint                         `json:"account_id"`
	UserID    uint                         `json:"user_id,omitempty"`
	Params    ConversationBulkActionParams `json:"params"`
}

var conversationMaintenanceRegistrations sync.Map

// RegisterConversationMaintenanceJobs wires Chatwoot scheduled maintenance jobs
// into the durable worker: TriggerScheduledItemsJob fans out to campaign,
// snooze-reopen, and auto-resolution jobs.
func RegisterConversationMaintenanceJobs(wp *worker.WorkerPool, db *gorm.DB) {
	registerConversationMaintenanceJobsWithNow(wp, db, time.Now)
}

func registerConversationMaintenanceJobsWithNow(wp *worker.WorkerPool, db *gorm.DB, now func() time.Time) {
	if wp == nil || db == nil {
		return
	}
	if _, loaded := conversationMaintenanceRegistrations.LoadOrStore(wp, struct{}{}); loaded {
		return
	}
	runner := &conversationMaintenanceRunner{wp: wp, db: db, now: now}
	wp.Register(TaskTypeScheduledTriggerItems, runner.performScheduledTriggerItems)
	wp.Register(TaskTypeCampaignTriggerOneoff, runner.performCampaignTriggerOneoff)
	wp.Register(TaskTypeConversationReopenSnoozed, runner.performReopenSnoozed)
	wp.Register(TaskTypeConversationResolutionScheduler, runner.performResolutionScheduler)
	wp.Register(TaskTypeConversationResolutionForAccount, runner.performResolutionForAccount)
	wp.Register(TaskTypeConversationUpdateMessageStatus, runner.performUpdateMessageStatus)
	wp.Register(TaskTypeConversationBulkAction, runner.performConversationBulkAction)
}

func EnqueueScheduledItemsTrigger(ctx context.Context, wp *worker.WorkerPool, scheduledAt time.Time) (*model.BackgroundJob, error) {
	if wp == nil {
		return nil, nil
	}
	return wp.Enqueue(ctx, TaskTypeScheduledTriggerItems, nil,
		worker.WithQueue("scheduled_jobs"),
		worker.WithScheduledAt(scheduledAt),
		worker.WithMaxAttempts(3),
		worker.WithIdempotencyKey(scheduledItemsIdempotencyKey(scheduledAt)),
	)
}

func EnqueueConversationMessageStatusUpdate(ctx context.Context, wp *worker.WorkerPool, conversationID uint, timestamp time.Time, status string) (*model.BackgroundJob, error) {
	if wp == nil {
		return nil, nil
	}
	if status == "" {
		status = string(model.MessageStatusRead)
	}
	payload := conversationUpdateMessageStatusJob{ConversationID: conversationID, Timestamp: timestamp.UTC().Unix(), Status: status}
	return wp.Enqueue(ctx, TaskTypeConversationUpdateMessageStatus, payload,
		worker.WithQueue("deferred"),
		worker.WithMaxAttempts(3),
		worker.WithIdempotencyKey(fmt.Sprintf("conversation:update_message_status:%d:%s:%d", conversationID, status, payload.Timestamp)),
	)
}

func EnqueueConversationBulkAction(ctx context.Context, wp *worker.WorkerPool, accountID, userID uint, params ConversationBulkActionParams) (*model.BackgroundJob, error) {
	if wp == nil {
		return nil, nil
	}
	if accountID == 0 || len(params.IDs) == 0 {
		return nil, fmt.Errorf("invalid conversation bulk action payload: account_id=%d ids=%v", accountID, params.IDs)
	}
	return wp.Enqueue(ctx, TaskTypeConversationBulkAction, conversationBulkActionJob{AccountID: accountID, UserID: userID, Params: params},
		worker.WithQueue("medium"),
		worker.WithMaxAttempts(3),
	)
}

func scheduledItemsIdempotencyKey(scheduledAt time.Time) string {
	bucket := scheduledAt.UTC().Truncate(scheduledItemsInterval).Unix()
	return fmt.Sprintf("scheduled:trigger_items:%d", bucket)
}

type conversationMaintenanceRunner struct {
	wp  *worker.WorkerPool
	db  *gorm.DB
	now func() time.Time
}

func (r *conversationMaintenanceRunner) performScheduledTriggerItems(ctx context.Context, job *model.BackgroundJob) error {
	now := r.now()
	var campaignIDs []uint
	if err := r.db.WithContext(ctx).Model(&campaign.Campaign{}).
		Where("campaign_type = ? AND campaign_status = ? AND enabled = ?", campaign.CampaignTypeOneOff, campaign.CampaignStatusActive, true).
		Where("scheduled_at BETWEEN ? AND ?", now.Add(-scheduledItemsLookback), now).
		Pluck("id", &campaignIDs).Error; err != nil {
		return fmt.Errorf("find due one-off campaigns: %w", err)
	}
	for _, campaignID := range campaignIDs {
		_, err := r.wp.Enqueue(ctx, TaskTypeCampaignTriggerOneoff, campaignTriggerOneoffJob{CampaignID: campaignID},
			worker.WithQueue("low"),
			worker.WithMaxAttempts(3),
			worker.WithIdempotencyKey(fmt.Sprintf("campaign:trigger_oneoff:%d", campaignID)),
		)
		if err != nil {
			return fmt.Errorf("enqueue campaign %d: %w", campaignID, err)
		}
	}
	if _, err := r.wp.Enqueue(ctx, TaskTypeConversationReopenSnoozed, nil, worker.WithQueue("low"), worker.WithMaxAttempts(3)); err != nil {
		return fmt.Errorf("enqueue reopen snoozed conversations: %w", err)
	}
	if _, err := r.wp.Enqueue(ctx, TaskTypeConversationResolutionScheduler, nil, worker.WithQueue("scheduled_jobs"), worker.WithMaxAttempts(3)); err != nil {
		return fmt.Errorf("enqueue conversation resolution scheduler: %w", err)
	}
	_, err := EnqueueScheduledItemsTrigger(ctx, r.wp, now.Add(scheduledItemsInterval))
	return err
}

func (r *conversationMaintenanceRunner) performCampaignTriggerOneoff(ctx context.Context, job *model.BackgroundJob) error {
	var payload campaignTriggerOneoffJob
	if err := json.Unmarshal(job.Payload, &payload); err != nil {
		return fmt.Errorf("unmarshal one-off campaign job: %w", err)
	}
	if payload.CampaignID == 0 {
		return fmt.Errorf("invalid one-off campaign job payload: %#v", payload)
	}

	claimed := false
	if err := r.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
		result := tx.Model(&campaign.Campaign{}).
			Where("id = ? AND campaign_type = ? AND campaign_status = ? AND enabled = ?", payload.CampaignID, campaign.CampaignTypeOneOff, campaign.CampaignStatusActive, true).
			Update("campaign_status", campaign.CampaignStatusProcessing)
		if result.Error != nil {
			return result.Error
		}
		claimed = result.RowsAffected == 1
		return nil
	}); err != nil {
		return fmt.Errorf("claim one-off campaign %d: %w", payload.CampaignID, err)
	}
	if !claimed {
		return nil
	}
	if err := campaign.NewCampaignService(r.db).TriggerCampaign(ctx, payload.CampaignID); err != nil {
		_ = r.db.WithContext(ctx).Model(&campaign.Campaign{}).Where("id = ?", payload.CampaignID).Update("campaign_status", campaign.CampaignStatusActive).Error
		return err
	}
	return r.db.WithContext(ctx).Model(&campaign.Campaign{}).Where("id = ?", payload.CampaignID).Update("campaign_status", campaign.CampaignStatusCompleted).Error
}

func (r *conversationMaintenanceRunner) performReopenSnoozed(ctx context.Context, job *model.BackgroundJob) error {
	now := r.now()
	nowUnix := now.Unix()
	lookbackUnix := now.Add(-scheduledItemsLookback).Unix()
	updates := map[string]any{
		"status":        string(model.ConversationStatusOpen),
		"snoozed_until": nil,
		"resumed_at":    now,
	}
	return r.db.WithContext(ctx).Model(&model.Conversation{}).
		Where("status = ?", string(model.ConversationStatusSnoozed)).
		Where("snoozed_until BETWEEN ? AND ?", lookbackUnix, nowUnix).
		Updates(updates).Error
}

func (r *conversationMaintenanceRunner) performResolutionScheduler(ctx context.Context, job *model.BackgroundJob) error {
	var accountIDs []uint
	if err := r.db.WithContext(ctx).Model(&model.Account{}).
		Where("auto_resolve_duration > 0").
		Pluck("id", &accountIDs).Error; err != nil {
		return fmt.Errorf("find auto-resolve accounts: %w", err)
	}
	for _, accountID := range accountIDs {
		if _, err := r.wp.Enqueue(ctx, TaskTypeConversationResolutionForAccount, conversationResolutionJob{AccountID: accountID}, worker.WithQueue("low"), worker.WithMaxAttempts(3)); err != nil {
			return fmt.Errorf("enqueue account conversation resolution %d: %w", accountID, err)
		}
	}
	return nil
}

func (r *conversationMaintenanceRunner) performResolutionForAccount(ctx context.Context, job *model.BackgroundJob) error {
	var payload conversationResolutionJob
	if err := json.Unmarshal(job.Payload, &payload); err != nil {
		return fmt.Errorf("unmarshal conversation resolution job: %w", err)
	}
	if payload.AccountID == 0 {
		return fmt.Errorf("invalid conversation resolution job payload: %#v", payload)
	}
	var account model.Account
	if err := r.db.WithContext(ctx).First(&account, payload.AccountID).Error; err != nil {
		return fmt.Errorf("load auto-resolve account %d: %w", payload.AccountID, err)
	}
	if account.AutoResolveDuration <= 0 {
		return nil
	}
	cutoff := r.now().Add(-time.Duration(account.AutoResolveDuration) * time.Minute).Unix()
	now := r.now()
	updates := map[string]any{
		"status":      string(model.ConversationStatusResolved),
		"resolved_at": now,
	}
	return r.db.WithContext(ctx).Model(&model.Conversation{}).
		Where("account_id = ? AND status = ? AND contact_id <> 0", account.ID, string(model.ConversationStatusOpen)).
		Where("last_activity_at IS NOT NULL AND last_activity_at < ?", cutoff).
		Limit(conversationResolutionLimit).
		Updates(updates).Error
}

func (r *conversationMaintenanceRunner) performUpdateMessageStatus(ctx context.Context, job *model.BackgroundJob) error {
	var payload conversationUpdateMessageStatusJob
	if err := json.Unmarshal(job.Payload, &payload); err != nil {
		return fmt.Errorf("unmarshal conversation message status job: %w", err)
	}
	if payload.ConversationID == 0 || payload.Timestamp == 0 {
		return fmt.Errorf("invalid conversation message status job payload: %#v", payload)
	}
	if !validConversationMessageStatus(payload.Status) {
		return nil
	}

	var conversation model.Conversation
	if err := r.db.WithContext(ctx).First(&conversation, payload.ConversationID).Error; err != nil {
		if err == gorm.ErrRecordNotFound {
			return nil
		}
		return fmt.Errorf("load conversation %d for message status update: %w", payload.ConversationID, err)
	}

	return r.db.WithContext(ctx).Model(&model.Message{}).
		Where("conversation_id = ?", conversation.ID).
		Where("status IN ?", []string{string(model.MessageStatusSent), string(model.MessageStatusDelivered)}).
		Where("message_type <> ?", "incoming").
		Where("created_at <= ?", time.Unix(payload.Timestamp, 0).UTC()).
		Update("status", payload.Status).Error
}

func validConversationMessageStatus(status string) bool {
	switch status {
	case string(model.MessageStatusRead), string(model.MessageStatusDelivered):
		return true
	default:
		return false
	}
}

func (r *conversationMaintenanceRunner) performConversationBulkAction(ctx context.Context, job *model.BackgroundJob) error {
	var payload conversationBulkActionJob
	if err := json.Unmarshal(job.Payload, &payload); err != nil {
		return fmt.Errorf("unmarshal conversation bulk action job: %w", err)
	}
	if payload.AccountID == 0 || len(payload.Params.IDs) == 0 {
		return fmt.Errorf("invalid conversation bulk action job payload: %#v", payload)
	}
	if payload.Params.Type != "" && payload.Params.Type != "Conversation" {
		return nil
	}

	var conversations []model.Conversation
	if err := r.db.WithContext(ctx).
		Where("account_id = ? AND display_id IN ?", payload.AccountID, payload.Params.IDs).
		Find(&conversations).Error; err != nil {
		return fmt.Errorf("load bulk action conversations: %w", err)
	}
	for i := range conversations {
		conversation := conversations[i]
		updates := map[string]any{}
		if payload.Params.Fields.Status != nil && *payload.Params.Fields.Status != "" {
			updates["status"] = *payload.Params.Fields.Status
			now := r.now()
			switch *payload.Params.Fields.Status {
			case string(model.ConversationStatusResolved):
				updates["resolved_at"] = now
			case string(model.ConversationStatusOpen):
				updates["resumed_at"] = now
				updates["snoozed_until"] = nil
			}
		}
		if payload.Params.Fields.AssigneeID != nil {
			updates["assignee_id"] = payload.Params.Fields.AssigneeID
		}
		if payload.Params.Fields.TeamID != nil {
			updates["team_id"] = payload.Params.Fields.TeamID
		}
		if payload.Params.SnoozedUntil != "" {
			if snoozedUntil, ok := parseBulkActionTime(payload.Params.SnoozedUntil); ok {
				updates["snoozed_until"] = snoozedUntil.Unix()
			}
		}
		if len(payload.Params.Labels.Add) > 0 || len(payload.Params.Labels.Remove) > 0 {
			updates["labels"] = mergeConversationLabels(conversation.Labels, payload.Params.Labels.Add, payload.Params.Labels.Remove)
		}
		if len(updates) == 0 {
			continue
		}
		if err := r.db.WithContext(ctx).Model(&model.Conversation{}).Where("id = ?", conversation.ID).Updates(updates).Error; err != nil {
			return fmt.Errorf("bulk update conversation %d: %w", conversation.ID, err)
		}
	}
	return nil
}

func parseBulkActionTime(value string) (time.Time, bool) {
	if value == "" {
		return time.Time{}, false
	}
	if ts, err := time.Parse(time.RFC3339, value); err == nil {
		return ts.UTC(), true
	}
	if ts, err := time.Parse("2006-01-02T15:04:05.000Z", value); err == nil {
		return ts.UTC(), true
	}
	return time.Time{}, false
}

func mergeConversationLabels(existing string, add, remove []string) string {
	labels := map[string]bool{}
	order := []string{}
	for _, label := range splitConversationLabels(existing) {
		if !labels[label] {
			labels[label] = true
			order = append(order, label)
		}
	}
	for _, label := range add {
		label = strings.TrimSpace(label)
		if label == "" || labels[label] {
			continue
		}
		labels[label] = true
		order = append(order, label)
	}
	for _, label := range remove {
		delete(labels, strings.TrimSpace(label))
	}
	merged := make([]string, 0, len(order))
	for _, label := range order {
		if labels[label] {
			merged = append(merged, label)
		}
	}
	return strings.Join(merged, ",")
}

func splitConversationLabels(existing string) []string {
	parts := strings.Split(existing, ",")
	labels := make([]string, 0, len(parts))
	for _, part := range parts {
		label := strings.TrimSpace(part)
		if label != "" {
			labels = append(labels, label)
		}
	}
	return labels
}