Laniakea/scene_handler.go

package laniakea

import (
	"fmt"
	"strings"
	"time"
)

func (bot *Bot[T]) tryHandleScene(ctx *MessageContext) (bool, error) {
	for _, scope := range bot.sceneScopePriority {
		key, ok := buildSceneKey(scope, ctx)
		if !ok {
			continue
		}

		unlock := bot.sceneLocks.lock([]string{key})
		session, err := bot.sessionStore.Get(key)
		if err != nil {
			unlock()
			return false, err
		}
		if session.Scene == "" {
			unlock()
			continue
		}

		handled, err := bot.tryHandleSceneSession(ctx, key, session)
		unlock()
		return handled, err
	}
	return false, nil
}

func (bot *Bot[T]) tryHandleSceneSession(ctx *MessageContext, key string, session SceneSession) (bool, error) {
	for _, plugin := range bot.plugins {
		scene, ok := plugin.scenes[session.Scene]
		if !ok {
			continue
		}
		if scene.pluginName != "" && scene.pluginName != plugin.name {
			continue
		}
		if !plugin.executeMiddlewares(ctx, bot.appData) {
			return false, nil
		}
		sceneCtx := &SceneContext{
			MessageContext: ctx,
			sess:           session,
			key:            key,
		}

		return bot.executeScene(sceneCtx, scene)
	}
	return false, ErrSceneNotFound
}

func (bot *Bot[T]) executeScene(ctx *SceneContext, scene *Scene[T]) (bool, error) {
	if ctx.MessageContext == nil || ctx.sess.Scene == "" {
		return false, nil
	}

	var text string
	if ctx.Msg != nil {
		text = ctx.Msg.Text
		if text == "" {
			text = ctx.Msg.Caption
		}
	}

	text = strings.TrimSpace(text)
	prefix, cmd, args := bot.parseCommand(text)
	if cmd != "" {
		ctx.Prefix = prefix
		ctx.Text = args
		ctx.Args = strings.Fields(args)

		if _, ok := scene.commands[cmd]; ok {
			startTime := time.Now()
			bot.emitSceneStarted(ctx, scene, HandlerSceneCommandKind, cmd)
			res, _, err := scene.executeCommand(cmd, ctx, bot.appData)
			if err != nil {
				bot.emitSceneFinished(ctx, scene, HandlerSceneCommandKind, cmd, startTime, err)
				return false, bot.emitSceneError(ctx, scene, HandlerSceneCommandKind, cmd, err)
			}
			from := ctx.sess.Step
			ok, err := bot.applySceneResult(scene, ctx, res)
			bot.emitSceneFinished(ctx, scene, HandlerSceneCommandKind, cmd, startTime, err)
			if err != nil {
				err = bot.emitSceneError(ctx, scene, HandlerSceneCommandKind, cmd, err)
			}
			if ok {
				bot.emitSceneTransition(ctx, scene, from, res)
			}
			return ok, err
		}

		// Unmatched slash-commands should continue through normal bot command routing
		// instead of also triggering the active scene step or fallback handler.
		return false, nil
	}

	query := ctx.Update.CallbackQuery
	if query != nil {
		data, err := bot.decodePayload(query.Data)
		if err != nil {
			return false, err
		}
		ctx.Args = data.Args
		cmd := data.Command
		if _, ok := scene.payloads[cmd]; ok {
			startTime := time.Now()
			bot.emitSceneStarted(ctx, scene, HandlerScenePayloadKind, cmd)
			res, _, err := scene.executePayload(cmd, ctx, bot.appData)
			if err != nil {
				bot.emitSceneFinished(ctx, scene, HandlerScenePayloadKind, cmd, startTime, err)
				return false, bot.emitSceneError(ctx, scene, HandlerScenePayloadKind, cmd, err)
			}
			from := ctx.sess.Step
			ok, err := bot.applySceneResult(scene, ctx, res)
			bot.emitSceneFinished(ctx, scene, HandlerScenePayloadKind, cmd, startTime, err)
			if err != nil {
				err = bot.emitSceneError(ctx, scene, HandlerScenePayloadKind, cmd, err)
			}
			if ok {
				bot.emitSceneTransition(ctx, scene, from, res)
			}
			return ok, err
		}

		// Unmatched payloads should not trigger the active scene step or fallback handler.
		// This allows using payloads for other bot features like pagination without interfering with active scenes.
		return false, nil
	}

	ctx.Text = text
	ctx.Args = nil
	ctx.Prefix = ""
	if ctx.sess.Step != "" {
		step := ctx.sess.Step
		if _, ok := scene.steps[step]; ok {
			startTime := time.Now()
			bot.emitSceneStarted(ctx, scene, HandlerSceneStepKind, step)
			res, _, err := scene.executeStep(step, ctx, bot.appData)
			if err != nil {
				bot.emitSceneFinished(ctx, scene, HandlerSceneStepKind, step, startTime, err)
				return false, bot.emitSceneError(ctx, scene, HandlerSceneStepKind, step, err)
			}
			from := step
			ok, err := bot.applySceneResult(scene, ctx, res)
			bot.emitSceneFinished(ctx, scene, HandlerSceneStepKind, step, startTime, err)
			if err != nil {
				err = bot.emitSceneError(ctx, scene, HandlerSceneStepKind, from, err)
			}
			if ok {
				bot.emitSceneTransition(ctx, scene, from, res)
			}
			return ok, err
		}
	}

	if scene.message != nil {
		startTime := time.Now()
		bot.emitSceneStarted(ctx, scene, HandlerSceneMessageKind, "message_fallback")
		res, _, err := scene.executeMessage(ctx, bot.appData)
		if err != nil {
			bot.emitSceneFinished(ctx, scene, HandlerSceneMessageKind, "message_fallback", startTime, err)
			return false, bot.emitSceneError(ctx, scene, HandlerSceneMessageKind, "message_fallback", err)
		}
		from := ctx.sess.Step
		ok, err := bot.applySceneResult(scene, ctx, res)
		bot.emitSceneFinished(ctx, scene, HandlerSceneMessageKind, "message_fallback", startTime, err)
		if err != nil {
			err = bot.emitSceneError(ctx, scene, HandlerSceneMessageKind, "message_fallback", err)
		}
		if ok {
			bot.emitSceneTransition(ctx, scene, from, res)
		}
		return ok, err
	}

	return false, nil
}

func (bot *Bot[T]) emitSceneStarted(ctx *SceneContext, scene *Scene[T], kind HandlerEventKind, name string) {
	bot.safeEmitEvent(ctx.Context(), HandlerStartedEvent{
		UpdateID:    ctx.Update.UpdateID,
		UpdateType:  ctx.Update.Type,
		Plugin:      scene.pluginName,
		HandlerKind: kind,
		HandlerName: name,
		FromID:      ctx.FromID,
		ChatID:      ctx.ChatID,
	})
}

func (bot *Bot[T]) emitSceneFinished(ctx *SceneContext, scene *Scene[T], kind HandlerEventKind, name string, startedAt time.Time, err error) {
	bot.safeEmitEvent(ctx.Context(), HandlerFinishedEvent{
		UpdateID:    ctx.Update.UpdateID,
		UpdateType:  ctx.Update.Type,
		Plugin:      scene.pluginName,
		HandlerKind: kind,
		HandlerName: name,
		FromID:      ctx.FromID,
		ChatID:      ctx.ChatID,
		Duration:    time.Since(startedAt),
		Err:         err,
		UserFacing:  IsUserError(err),
	})
}

type reportedSceneError struct {
	err error
}

func (e *reportedSceneError) Error() string { return e.err.Error() }
func (e *reportedSceneError) Unwrap() error { return e.err }

func (bot *Bot[T]) emitSceneError(ctx *SceneContext, scene *Scene[T], kind HandlerEventKind, name string, err error) error {
	bot.safeEmitEvent(ctx.Context(), ErrorEvent{
		UpdateID:    ctx.Update.UpdateID,
		UpdateType:  ctx.Update.Type,
		Plugin:      scene.pluginName,
		HandlerKind: kind,
		HandlerName: name,
		FromID:      ctx.FromID,
		ChatID:      ctx.ChatID,
		Err:         err,
		UserFacing:  IsUserError(err),
	})
	return &reportedSceneError{err: err}
}

func (bot *Bot[T]) emitSceneTransition(ctx *SceneContext, scene *Scene[T], from string, result SceneResult) {
	if result.Action == SceneActionPass {
		return
	}

	var to string
	switch result.Action {
	case SceneActionNext:
		to = result.Next
	case SceneActionExit:
		to = ""
	default:
		to = from
	}

	bot.safeEmitEvent(ctx.Context(), SceneTransitionEvent{
		Plugin: scene.pluginName,
		Scene:  scene.name,
		From:   from,
		To:     to,
		Action: result.Action,
		FromID: ctx.FromID,
		ChatID: ctx.ChatID,
	})
}

func (bot *Bot[T]) applySceneResult(scene *Scene[T], ctx *SceneContext, result SceneResult) (bool, error) {
	switch result.Action {
	case SceneActionStay:
		if err := bot.sessionStore.Set(ctx.key, ctx.sess); err != nil {
			return false, err
		}
		return true, nil
	case SceneActionNext:
		if result.Next == "" {
			return false, ErrSceneStepNotFound
		}
		if _, ok := scene.steps[result.Next]; !ok {
			return false, ErrSceneStepNotFound
		}
		ctx.sess.Step = result.Next
		if err := bot.sessionStore.Set(ctx.key, ctx.sess); err != nil {
			return false, err
		}
		return true, nil
	case SceneActionExit:
		if err := bot.sessionStore.Delete(ctx.key); err != nil {
			return false, err
		}
		return true, nil
	case SceneActionPass:
		return false, nil
	default:
		return false, fmt.Errorf("%w: %v", ErrInvalidSceneAction, result.Action)
	}
}
func buildSceneKey(scope SceneScope, ctx *MessageContext) (string, bool) {
	if ctx == nil {
		return "", false
	}

	switch scope {
	case SceneScopeUserChat:
		if ctx.Msg == nil || ctx.Msg.Chat == nil || ctx.FromID == 0 {
			return "", false
		}
		return fmt.Sprintf("user_id:%d:chat_id:%d", ctx.FromID, ctx.Msg.Chat.ID), true
	case SceneScopeChat:
		if ctx.Msg == nil || ctx.Msg.Chat == nil {
			return "", false
		}
		return fmt.Sprintf("chat_id:%d", ctx.Msg.Chat.ID), true
	case SceneScopeUser:
		if ctx.FromID == 0 {
			return "", false
		}
		return fmt.Sprintf("user_id:%d", ctx.FromID), true
	default:
		return "", false
	}
}