refactor(chat): consolidate machine modules
This commit is contained in:
@@ -1,499 +0,0 @@
|
||||
import {
|
||||
setup,
|
||||
type ActionFunction,
|
||||
type DoneActorEvent,
|
||||
type ErrorActorEvent,
|
||||
type EventObject,
|
||||
} from "xstate";
|
||||
|
||||
import { Logger } from "@/utils/logger";
|
||||
import { todayString } from "@/utils/date";
|
||||
|
||||
import type { ChatEvent } from "./chat-events";
|
||||
import {
|
||||
applyHistoryLoadedOutput,
|
||||
applyHttpSendOutput,
|
||||
applyNetworkHistoryLoadedOutput,
|
||||
applyPromotionUnlockOutput,
|
||||
applySingleUnlockOutput,
|
||||
beginPendingReply,
|
||||
countLockedHistoryMessages,
|
||||
createChatPromotionState,
|
||||
finishPendingReply,
|
||||
type HttpSendOutput,
|
||||
type UnlockMessageOutput,
|
||||
} from "./chat-machine.helpers";
|
||||
import { baseChatMachineSetup } from "./chat-machine.setup";
|
||||
import { initialState, type ChatState } from "./chat-state";
|
||||
import type { UnlockHistoryOutput } from "./chat-unlock-flow";
|
||||
|
||||
const sendLog = new Logger("StoresChatChatSendFlow");
|
||||
const mediaLog = new Logger("StoresChatChatMediaFlow");
|
||||
|
||||
const enqueueMessageAction = baseChatMachineSetup.sendTo(
|
||||
"messageQueue",
|
||||
({ event }) => event,
|
||||
);
|
||||
|
||||
const startGuestSessionAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
...initialState,
|
||||
promotion: context.promotion,
|
||||
}),
|
||||
);
|
||||
|
||||
const startUserSessionAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
...initialState,
|
||||
promotion: context.promotion,
|
||||
}),
|
||||
);
|
||||
|
||||
const clearChatSessionAction = baseChatMachineSetup.assign(() => ({
|
||||
...initialState,
|
||||
}));
|
||||
|
||||
const injectPromotionAction = baseChatMachineSetup.assign(({ event }) => {
|
||||
if (event.type !== "ChatPromotionInjected") return {};
|
||||
return {
|
||||
promotion: createChatPromotionState(
|
||||
event.promotion,
|
||||
event.messageId,
|
||||
),
|
||||
};
|
||||
});
|
||||
|
||||
const clearPromotionAction = baseChatMachineSetup.assign({ promotion: null });
|
||||
|
||||
const applyLocalHistoryLoadedAction = baseChatMachineSetup.assign(
|
||||
({ event }) => {
|
||||
if (event.type !== "ChatLocalHistoryLoaded") return {};
|
||||
return applyHistoryLoadedOutput(event.output);
|
||||
},
|
||||
);
|
||||
|
||||
const applyNetworkHistoryLoadedAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatNetworkHistoryLoaded") return {};
|
||||
return applyNetworkHistoryLoadedOutput(context, event.output);
|
||||
},
|
||||
);
|
||||
|
||||
const applyNetworkHistoryLoadedAndClearPaymentAction =
|
||||
baseChatMachineSetup.assign(({ context, event }) => {
|
||||
if (event.type !== "ChatNetworkHistoryLoaded") return {};
|
||||
return {
|
||||
...applyNetworkHistoryLoadedOutput(context, event.output),
|
||||
paymentUnlockPending: false,
|
||||
};
|
||||
});
|
||||
|
||||
const showUnlockHistoryPromptFromNetworkAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatNetworkHistoryLoaded") return {};
|
||||
const nextHistory = applyNetworkHistoryLoadedOutput(context, event.output);
|
||||
return {
|
||||
...nextHistory,
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: true,
|
||||
lockedHistoryCount: countLockedHistoryMessages(nextHistory.messages),
|
||||
unlockHistoryError: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const markHistoryLoadFailedAction = baseChatMachineSetup.assign({
|
||||
historyLoaded: true,
|
||||
});
|
||||
|
||||
const appendGuestUserMessageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendMessage") return {};
|
||||
|
||||
const today = todayString();
|
||||
sendLog.debug("[chat-machine] appendGuestUserMessage", {
|
||||
contentLength: event.content.length,
|
||||
contentPreview: event.content.slice(0, 50),
|
||||
quotaMode: "server controlled",
|
||||
isReplyingAI: context.isReplyingAI,
|
||||
});
|
||||
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: event.content,
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
},
|
||||
],
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const appendUserMessageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendMessage") return {};
|
||||
|
||||
const today = todayString();
|
||||
sendLog.debug("[chat-machine] appendUserMessage", {
|
||||
contentLength: event.content.length,
|
||||
contentPreview: event.content.slice(0, 50),
|
||||
isReplyingAI: context.isReplyingAI,
|
||||
});
|
||||
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: event.content,
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
},
|
||||
],
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const appendQueuedSendErrorMessageAction = baseChatMachineSetup.assign(
|
||||
({ context }) => {
|
||||
const messages = [
|
||||
...context.messages,
|
||||
{
|
||||
content: "Something went wrong. Try sending again?",
|
||||
isFromAI: true,
|
||||
date: todayString(),
|
||||
},
|
||||
];
|
||||
return { messages, ...finishPendingReply(context) };
|
||||
},
|
||||
);
|
||||
|
||||
const markQueuedSendStartedAction = baseChatMachineSetup.assign(
|
||||
({ context }) => beginPendingReply(context),
|
||||
);
|
||||
|
||||
const applyQueuedHttpOutputAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatQueuedHttpDone") return {};
|
||||
return applyHttpSendOutput(context, event.output);
|
||||
},
|
||||
);
|
||||
|
||||
const clearUpgradePromptAction = baseChatMachineSetup.assign(() => ({
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
canSendMessage: true,
|
||||
requiredCredits: 0,
|
||||
shortfallCredits: 0,
|
||||
}));
|
||||
|
||||
const appendGuestUserImageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendImage") return {};
|
||||
|
||||
const today = todayString();
|
||||
mediaLog.debug("[chat-machine] appendGuestUserImage", {
|
||||
oldMessagesCount: context.messages.length,
|
||||
quotaMode: "server controlled",
|
||||
});
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: "[Image]",
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
imageUrl: event.imageBase64,
|
||||
},
|
||||
],
|
||||
...beginPendingReply(context),
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const appendUserImageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendImage") return {};
|
||||
|
||||
const today = todayString();
|
||||
mediaLog.debug("[chat-machine] appendUserImage", {
|
||||
oldMessagesCount: context.messages.length,
|
||||
});
|
||||
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: "[Image]",
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
imageUrl: event.imageBase64,
|
||||
},
|
||||
],
|
||||
...beginPendingReply(context),
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const markPaymentUnlockPendingAction = baseChatMachineSetup.assign(() => ({
|
||||
paymentUnlockPending: true,
|
||||
unlockHistoryError: null,
|
||||
}));
|
||||
|
||||
const showUnlockHistoryPromptAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: true,
|
||||
lockedHistoryCount: countLockedHistoryMessages(context.messages),
|
||||
unlockHistoryError: null,
|
||||
}),
|
||||
);
|
||||
|
||||
const dismissUnlockHistoryPromptAction = baseChatMachineSetup.assign(() => ({
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: false,
|
||||
unlockHistoryError: null,
|
||||
}));
|
||||
|
||||
const markUnlockHistoryStartedAction = baseChatMachineSetup.assign(() => ({
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: false,
|
||||
isUnlockingHistory: true,
|
||||
unlockHistoryError: null,
|
||||
}));
|
||||
|
||||
const markUnlockHistoryFailedAction = baseChatMachineSetup.assign(() => ({
|
||||
isUnlockingHistory: false,
|
||||
unlockHistoryPromptVisible: true,
|
||||
unlockHistoryError: "Failed to unlock messages. Please try again.",
|
||||
}));
|
||||
|
||||
const markUnlockMessageStartedAction = baseChatMachineSetup.assign(
|
||||
({ event }) => {
|
||||
if (event.type !== "ChatUnlockMessageRequested") return {};
|
||||
return {
|
||||
isUnlockingMessage: true,
|
||||
unlockingMessageId: event.messageId,
|
||||
unlockingMessageKind: event.kind,
|
||||
unlockingRemoteMessageId:
|
||||
event.remoteMessageId ?? (event.lockType ? null : event.messageId),
|
||||
unlockingLockType: event.lockType ?? null,
|
||||
unlockingClientLockId: event.clientLockId ?? null,
|
||||
unlockMessageError: null,
|
||||
unlockPaywallRequest: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const applyUnlockMessageSucceededAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
const { output } = event as unknown as DoneActorEvent<UnlockMessageOutput>;
|
||||
return {
|
||||
messages: applySingleUnlockOutput(context.messages, output),
|
||||
promotion: applyPromotionUnlockOutput(context.promotion, output),
|
||||
isUnlockingMessage: false,
|
||||
unlockingMessageId: null,
|
||||
unlockingMessageKind: null,
|
||||
unlockingRemoteMessageId: null,
|
||||
unlockingLockType: null,
|
||||
unlockingClientLockId: null,
|
||||
unlockMessageError: null,
|
||||
unlockPaywallRequest: null,
|
||||
creditBalance: output.response.creditBalance,
|
||||
creditsCharged: output.response.creditsCharged,
|
||||
requiredCredits: 0,
|
||||
shortfallCredits: 0,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const requestUnlockPaymentFromOutputAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
const { output } = event as unknown as DoneActorEvent<UnlockMessageOutput>;
|
||||
return {
|
||||
promotion: applyPromotionUnlockOutput(context.promotion, output),
|
||||
isUnlockingMessage: false,
|
||||
unlockingMessageId: null,
|
||||
unlockingMessageKind: null,
|
||||
unlockingRemoteMessageId: null,
|
||||
unlockingLockType: null,
|
||||
unlockingClientLockId: null,
|
||||
unlockMessageError: output.response.reason,
|
||||
unlockPaywallRequest: {
|
||||
displayMessageId:
|
||||
output.response.messageId || output.displayMessageId,
|
||||
...(output.response.messageId || output.request.messageId
|
||||
? {
|
||||
messageId:
|
||||
output.response.messageId || output.request.messageId,
|
||||
}
|
||||
: {}),
|
||||
kind: context.unlockingMessageKind ?? "private",
|
||||
...(output.request.lockType
|
||||
? { lockType: output.request.lockType }
|
||||
: {}),
|
||||
...(output.request.clientLockId
|
||||
? { clientLockId: output.request.clientLockId }
|
||||
: {}),
|
||||
...(context.promotion?.message.id === output.displayMessageId
|
||||
? { promotion: context.promotion.session }
|
||||
: {}),
|
||||
reason: output.response.reason,
|
||||
creditBalance: output.response.creditBalance,
|
||||
requiredCredits: output.response.requiredCredits,
|
||||
shortfallCredits: output.response.shortfallCredits,
|
||||
},
|
||||
creditBalance: output.response.creditBalance,
|
||||
requiredCredits: output.response.requiredCredits,
|
||||
shortfallCredits: output.response.shortfallCredits,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const requestUnlockPaymentFromErrorAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
isUnlockingMessage: false,
|
||||
unlockMessageError: "unlock_failed",
|
||||
unlockPaywallRequest:
|
||||
context.unlockingMessageId && context.unlockingMessageKind
|
||||
? {
|
||||
displayMessageId: context.unlockingMessageId,
|
||||
...(context.unlockingRemoteMessageId
|
||||
? { messageId: context.unlockingRemoteMessageId }
|
||||
: {}),
|
||||
kind: context.unlockingMessageKind,
|
||||
...(context.unlockingLockType
|
||||
? { lockType: context.unlockingLockType }
|
||||
: {}),
|
||||
...(context.unlockingClientLockId
|
||||
? { clientLockId: context.unlockingClientLockId }
|
||||
: {}),
|
||||
...(context.promotion?.message.id === context.unlockingMessageId
|
||||
? { promotion: context.promotion.session }
|
||||
: {}),
|
||||
reason: "unlock_failed",
|
||||
creditBalance: context.creditBalance,
|
||||
requiredCredits: context.requiredCredits,
|
||||
shortfallCredits: context.shortfallCredits,
|
||||
}
|
||||
: null,
|
||||
unlockingMessageId: null,
|
||||
unlockingMessageKind: null,
|
||||
unlockingRemoteMessageId: null,
|
||||
unlockingLockType: null,
|
||||
unlockingClientLockId: null,
|
||||
}),
|
||||
);
|
||||
|
||||
const clearUnlockPaywallRequestAction = baseChatMachineSetup.assign(() => ({
|
||||
unlockPaywallRequest: null,
|
||||
}));
|
||||
|
||||
type ChatActorAction<TEvent extends EventObject> = ActionFunction<
|
||||
ChatState,
|
||||
TEvent,
|
||||
ChatEvent,
|
||||
undefined,
|
||||
never,
|
||||
never,
|
||||
never,
|
||||
never,
|
||||
never
|
||||
>;
|
||||
|
||||
function createChatActorActionSetup<TEvent extends EventObject>() {
|
||||
const actorActionSetup = setup({
|
||||
types: {
|
||||
context: {} as ChatState,
|
||||
events: {} as TEvent,
|
||||
},
|
||||
});
|
||||
return {
|
||||
assign(
|
||||
assignment: Parameters<typeof actorActionSetup.assign>[0],
|
||||
): ChatActorAction<TEvent> {
|
||||
// Actor lifecycle events are expression events, not public machine events.
|
||||
return actorActionSetup.assign(assignment) as unknown as ChatActorAction<TEvent>;
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
const sendMessageDoneActionSetup =
|
||||
createChatActorActionSetup<DoneActorEvent<HttpSendOutput>>();
|
||||
const unlockHistoryDoneActionSetup =
|
||||
createChatActorActionSetup<DoneActorEvent<UnlockHistoryOutput>>();
|
||||
const actorErrorActionSetup = createChatActorActionSetup<ErrorActorEvent>();
|
||||
|
||||
export const applyHttpSendOutputAction = sendMessageDoneActionSetup.assign(
|
||||
({ context, event }) => applyHttpSendOutput(context, event.output),
|
||||
);
|
||||
|
||||
export const markHttpSendFailedAction = actorErrorActionSetup.assign({
|
||||
isReplyingAI: false,
|
||||
});
|
||||
|
||||
export const applyUnlockHistoryOutputAction =
|
||||
unlockHistoryDoneActionSetup.assign(({ event }) => {
|
||||
const lockedHistoryCount = countLockedHistoryMessages(
|
||||
event.output.messages,
|
||||
);
|
||||
const insufficientBalance =
|
||||
!event.output.unlocked &&
|
||||
event.output.reason === "insufficient_balance";
|
||||
return {
|
||||
messages: event.output.messages,
|
||||
historyLoaded: true,
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: insufficientBalance,
|
||||
lockedHistoryCount,
|
||||
isUnlockingHistory: false,
|
||||
unlockHistoryError: insufficientBalance
|
||||
? `Insufficient credits. You still need ${event.output.shortfallCredits} credits.`
|
||||
: null,
|
||||
};
|
||||
});
|
||||
|
||||
export const chatMachineSetup = baseChatMachineSetup.extend({
|
||||
actions: {
|
||||
enqueueMessage: enqueueMessageAction,
|
||||
startGuestSession: startGuestSessionAction,
|
||||
startUserSession: startUserSessionAction,
|
||||
clearChatSession: clearChatSessionAction,
|
||||
injectPromotion: injectPromotionAction,
|
||||
clearPromotion: clearPromotionAction,
|
||||
applyLocalHistoryLoaded: applyLocalHistoryLoadedAction,
|
||||
applyNetworkHistoryLoaded: applyNetworkHistoryLoadedAction,
|
||||
applyNetworkHistoryLoadedAndClearPayment:
|
||||
applyNetworkHistoryLoadedAndClearPaymentAction,
|
||||
showUnlockHistoryPromptFromNetwork:
|
||||
showUnlockHistoryPromptFromNetworkAction,
|
||||
markHistoryLoadFailed: markHistoryLoadFailedAction,
|
||||
appendGuestUserMessage: appendGuestUserMessageAction,
|
||||
appendUserMessage: appendUserMessageAction,
|
||||
appendQueuedSendErrorMessage: appendQueuedSendErrorMessageAction,
|
||||
markQueuedSendStarted: markQueuedSendStartedAction,
|
||||
applyQueuedHttpOutput: applyQueuedHttpOutputAction,
|
||||
clearUpgradePrompt: clearUpgradePromptAction,
|
||||
appendGuestUserImage: appendGuestUserImageAction,
|
||||
appendUserImage: appendUserImageAction,
|
||||
markPaymentUnlockPending: markPaymentUnlockPendingAction,
|
||||
showUnlockHistoryPrompt: showUnlockHistoryPromptAction,
|
||||
dismissUnlockHistoryPrompt: dismissUnlockHistoryPromptAction,
|
||||
markUnlockHistoryStarted: markUnlockHistoryStartedAction,
|
||||
markUnlockHistoryFailed: markUnlockHistoryFailedAction,
|
||||
markUnlockMessageStarted: markUnlockMessageStartedAction,
|
||||
applyUnlockMessageSucceeded: applyUnlockMessageSucceededAction,
|
||||
requestUnlockPaymentFromOutput: requestUnlockPaymentFromOutputAction,
|
||||
requestUnlockPaymentFromError: requestUnlockPaymentFromErrorAction,
|
||||
clearUnlockPaywallRequest: clearUnlockPaywallRequestAction,
|
||||
},
|
||||
});
|
||||
@@ -1,25 +0,0 @@
|
||||
import { setup } from "xstate";
|
||||
|
||||
import type { ChatEvent } from "./chat-events";
|
||||
import {
|
||||
httpMessageQueueActor,
|
||||
loadHistoryActor,
|
||||
sendMessageHttpActor,
|
||||
unlockHistoryActor,
|
||||
unlockMessageActor,
|
||||
} from "./chat-machine.actors";
|
||||
import type { ChatState } from "./chat-state";
|
||||
|
||||
export const baseChatMachineSetup = setup({
|
||||
types: {
|
||||
context: {} as ChatState,
|
||||
events: {} as ChatEvent,
|
||||
},
|
||||
actors: {
|
||||
loadHistory: loadHistoryActor,
|
||||
sendMessageHttp: sendMessageHttpActor,
|
||||
httpMessageQueue: httpMessageQueueActor,
|
||||
unlockHistory: unlockHistoryActor,
|
||||
unlockMessage: unlockMessageActor,
|
||||
},
|
||||
});
|
||||
@@ -1,17 +1,529 @@
|
||||
import {
|
||||
applyHttpSendOutputAction,
|
||||
applyUnlockHistoryOutputAction,
|
||||
chatMachineSetup,
|
||||
markHttpSendFailedAction,
|
||||
} from "./chat-machine.actions";
|
||||
import { shouldPromptUnlockHistory } from "./chat-machine.helpers";
|
||||
import { initialState } from "./chat-state";
|
||||
setup,
|
||||
type ActionFunction,
|
||||
type DoneActorEvent,
|
||||
type ErrorActorEvent,
|
||||
type EventObject,
|
||||
} from "xstate";
|
||||
|
||||
import { todayString } from "@/utils/date";
|
||||
import { Logger } from "@/utils/logger";
|
||||
|
||||
import type { ChatEvent } from "./chat-events";
|
||||
import {
|
||||
httpMessageQueueActor,
|
||||
loadHistoryActor,
|
||||
sendMessageHttpActor,
|
||||
unlockHistoryActor,
|
||||
unlockMessageActor,
|
||||
} from "./chat-machine.actors";
|
||||
import {
|
||||
applyHistoryLoadedOutput,
|
||||
applyHttpSendOutput,
|
||||
applyNetworkHistoryLoadedOutput,
|
||||
applyPromotionUnlockOutput,
|
||||
applySingleUnlockOutput,
|
||||
beginPendingReply,
|
||||
countLockedHistoryMessages,
|
||||
createChatPromotionState,
|
||||
finishPendingReply,
|
||||
shouldPromptUnlockHistory,
|
||||
type HttpSendOutput,
|
||||
type UnlockMessageOutput,
|
||||
} from "./chat-machine.helpers";
|
||||
import { initialState, type ChatState } from "./chat-state";
|
||||
import type { UnlockHistoryOutput } from "./chat-unlock-flow";
|
||||
|
||||
// 重新导出 State / Event 类型，保持 machine 文件的公共 API 不变
|
||||
export type { ChatState } from "./chat-state";
|
||||
export { initialState } from "./chat-state";
|
||||
export type { ChatEvent } from "./chat-events";
|
||||
|
||||
const baseChatMachineSetup = setup({
|
||||
types: {
|
||||
context: {} as ChatState,
|
||||
events: {} as ChatEvent,
|
||||
},
|
||||
actors: {
|
||||
loadHistory: loadHistoryActor,
|
||||
sendMessageHttp: sendMessageHttpActor,
|
||||
httpMessageQueue: httpMessageQueueActor,
|
||||
unlockHistory: unlockHistoryActor,
|
||||
unlockMessage: unlockMessageActor,
|
||||
},
|
||||
});
|
||||
|
||||
const sendLog = new Logger("StoresChatChatSendFlow");
|
||||
const mediaLog = new Logger("StoresChatChatMediaFlow");
|
||||
|
||||
const enqueueMessageAction = baseChatMachineSetup.sendTo(
|
||||
"messageQueue",
|
||||
({ event }) => event,
|
||||
);
|
||||
|
||||
const startGuestSessionAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
...initialState,
|
||||
promotion: context.promotion,
|
||||
}),
|
||||
);
|
||||
|
||||
const startUserSessionAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
...initialState,
|
||||
promotion: context.promotion,
|
||||
}),
|
||||
);
|
||||
|
||||
const clearChatSessionAction = baseChatMachineSetup.assign(() => ({
|
||||
...initialState,
|
||||
}));
|
||||
|
||||
const injectPromotionAction = baseChatMachineSetup.assign(({ event }) => {
|
||||
if (event.type !== "ChatPromotionInjected") return {};
|
||||
return {
|
||||
promotion: createChatPromotionState(
|
||||
event.promotion,
|
||||
event.messageId,
|
||||
),
|
||||
};
|
||||
});
|
||||
|
||||
const clearPromotionAction = baseChatMachineSetup.assign({ promotion: null });
|
||||
|
||||
const applyLocalHistoryLoadedAction = baseChatMachineSetup.assign(
|
||||
({ event }) => {
|
||||
if (event.type !== "ChatLocalHistoryLoaded") return {};
|
||||
return applyHistoryLoadedOutput(event.output);
|
||||
},
|
||||
);
|
||||
|
||||
const applyNetworkHistoryLoadedAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatNetworkHistoryLoaded") return {};
|
||||
return applyNetworkHistoryLoadedOutput(context, event.output);
|
||||
},
|
||||
);
|
||||
|
||||
const applyNetworkHistoryLoadedAndClearPaymentAction =
|
||||
baseChatMachineSetup.assign(({ context, event }) => {
|
||||
if (event.type !== "ChatNetworkHistoryLoaded") return {};
|
||||
return {
|
||||
...applyNetworkHistoryLoadedOutput(context, event.output),
|
||||
paymentUnlockPending: false,
|
||||
};
|
||||
});
|
||||
|
||||
const showUnlockHistoryPromptFromNetworkAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatNetworkHistoryLoaded") return {};
|
||||
const nextHistory = applyNetworkHistoryLoadedOutput(context, event.output);
|
||||
return {
|
||||
...nextHistory,
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: true,
|
||||
lockedHistoryCount: countLockedHistoryMessages(nextHistory.messages),
|
||||
unlockHistoryError: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const markHistoryLoadFailedAction = baseChatMachineSetup.assign({
|
||||
historyLoaded: true,
|
||||
});
|
||||
|
||||
const appendGuestUserMessageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendMessage") return {};
|
||||
|
||||
const today = todayString();
|
||||
sendLog.debug("[chat-machine] appendGuestUserMessage", {
|
||||
contentLength: event.content.length,
|
||||
contentPreview: event.content.slice(0, 50),
|
||||
quotaMode: "server controlled",
|
||||
isReplyingAI: context.isReplyingAI,
|
||||
});
|
||||
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: event.content,
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
},
|
||||
],
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const appendUserMessageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendMessage") return {};
|
||||
|
||||
const today = todayString();
|
||||
sendLog.debug("[chat-machine] appendUserMessage", {
|
||||
contentLength: event.content.length,
|
||||
contentPreview: event.content.slice(0, 50),
|
||||
isReplyingAI: context.isReplyingAI,
|
||||
});
|
||||
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: event.content,
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
},
|
||||
],
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const appendQueuedSendErrorMessageAction = baseChatMachineSetup.assign(
|
||||
({ context }) => {
|
||||
const messages = [
|
||||
...context.messages,
|
||||
{
|
||||
content: "Something went wrong. Try sending again?",
|
||||
isFromAI: true,
|
||||
date: todayString(),
|
||||
},
|
||||
];
|
||||
return { messages, ...finishPendingReply(context) };
|
||||
},
|
||||
);
|
||||
|
||||
const markQueuedSendStartedAction = baseChatMachineSetup.assign(
|
||||
({ context }) => beginPendingReply(context),
|
||||
);
|
||||
|
||||
const applyQueuedHttpOutputAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatQueuedHttpDone") return {};
|
||||
return applyHttpSendOutput(context, event.output);
|
||||
},
|
||||
);
|
||||
|
||||
const clearUpgradePromptAction = baseChatMachineSetup.assign(() => ({
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
canSendMessage: true,
|
||||
requiredCredits: 0,
|
||||
shortfallCredits: 0,
|
||||
}));
|
||||
|
||||
const appendGuestUserImageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendImage") return {};
|
||||
|
||||
const today = todayString();
|
||||
mediaLog.debug("[chat-machine] appendGuestUserImage", {
|
||||
oldMessagesCount: context.messages.length,
|
||||
quotaMode: "server controlled",
|
||||
});
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: "[Image]",
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
imageUrl: event.imageBase64,
|
||||
},
|
||||
],
|
||||
...beginPendingReply(context),
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const appendUserImageAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
if (event.type !== "ChatSendImage") return {};
|
||||
|
||||
const today = todayString();
|
||||
mediaLog.debug("[chat-machine] appendUserImage", {
|
||||
oldMessagesCount: context.messages.length,
|
||||
});
|
||||
|
||||
return {
|
||||
messages: [
|
||||
...context.messages,
|
||||
{
|
||||
content: "[Image]",
|
||||
isFromAI: false,
|
||||
date: today,
|
||||
imageUrl: event.imageBase64,
|
||||
},
|
||||
],
|
||||
...beginPendingReply(context),
|
||||
upgradePromptVisible: false,
|
||||
upgradeReason: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const markPaymentUnlockPendingAction = baseChatMachineSetup.assign(() => ({
|
||||
paymentUnlockPending: true,
|
||||
unlockHistoryError: null,
|
||||
}));
|
||||
|
||||
const showUnlockHistoryPromptAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: true,
|
||||
lockedHistoryCount: countLockedHistoryMessages(context.messages),
|
||||
unlockHistoryError: null,
|
||||
}),
|
||||
);
|
||||
|
||||
const dismissUnlockHistoryPromptAction = baseChatMachineSetup.assign(() => ({
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: false,
|
||||
unlockHistoryError: null,
|
||||
}));
|
||||
|
||||
const markUnlockHistoryStartedAction = baseChatMachineSetup.assign(() => ({
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: false,
|
||||
isUnlockingHistory: true,
|
||||
unlockHistoryError: null,
|
||||
}));
|
||||
|
||||
const markUnlockHistoryFailedAction = baseChatMachineSetup.assign(() => ({
|
||||
isUnlockingHistory: false,
|
||||
unlockHistoryPromptVisible: true,
|
||||
unlockHistoryError: "Failed to unlock messages. Please try again.",
|
||||
}));
|
||||
|
||||
const markUnlockMessageStartedAction = baseChatMachineSetup.assign(
|
||||
({ event }) => {
|
||||
if (event.type !== "ChatUnlockMessageRequested") return {};
|
||||
return {
|
||||
isUnlockingMessage: true,
|
||||
unlockingMessageId: event.messageId,
|
||||
unlockingMessageKind: event.kind,
|
||||
unlockingRemoteMessageId:
|
||||
event.remoteMessageId ?? (event.lockType ? null : event.messageId),
|
||||
unlockingLockType: event.lockType ?? null,
|
||||
unlockingClientLockId: event.clientLockId ?? null,
|
||||
unlockMessageError: null,
|
||||
unlockPaywallRequest: null,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const applyUnlockMessageSucceededAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
const { output } = event as unknown as DoneActorEvent<UnlockMessageOutput>;
|
||||
return {
|
||||
messages: applySingleUnlockOutput(context.messages, output),
|
||||
promotion: applyPromotionUnlockOutput(context.promotion, output),
|
||||
isUnlockingMessage: false,
|
||||
unlockingMessageId: null,
|
||||
unlockingMessageKind: null,
|
||||
unlockingRemoteMessageId: null,
|
||||
unlockingLockType: null,
|
||||
unlockingClientLockId: null,
|
||||
unlockMessageError: null,
|
||||
unlockPaywallRequest: null,
|
||||
creditBalance: output.response.creditBalance,
|
||||
creditsCharged: output.response.creditsCharged,
|
||||
requiredCredits: 0,
|
||||
shortfallCredits: 0,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const requestUnlockPaymentFromOutputAction = baseChatMachineSetup.assign(
|
||||
({ context, event }) => {
|
||||
const { output } = event as unknown as DoneActorEvent<UnlockMessageOutput>;
|
||||
return {
|
||||
promotion: applyPromotionUnlockOutput(context.promotion, output),
|
||||
isUnlockingMessage: false,
|
||||
unlockingMessageId: null,
|
||||
unlockingMessageKind: null,
|
||||
unlockingRemoteMessageId: null,
|
||||
unlockingLockType: null,
|
||||
unlockingClientLockId: null,
|
||||
unlockMessageError: output.response.reason,
|
||||
unlockPaywallRequest: {
|
||||
displayMessageId:
|
||||
output.response.messageId || output.displayMessageId,
|
||||
...(output.response.messageId || output.request.messageId
|
||||
? {
|
||||
messageId:
|
||||
output.response.messageId || output.request.messageId,
|
||||
}
|
||||
: {}),
|
||||
kind: context.unlockingMessageKind ?? "private",
|
||||
...(output.request.lockType
|
||||
? { lockType: output.request.lockType }
|
||||
: {}),
|
||||
...(output.request.clientLockId
|
||||
? { clientLockId: output.request.clientLockId }
|
||||
: {}),
|
||||
...(context.promotion?.message.id === output.displayMessageId
|
||||
? { promotion: context.promotion.session }
|
||||
: {}),
|
||||
reason: output.response.reason,
|
||||
creditBalance: output.response.creditBalance,
|
||||
requiredCredits: output.response.requiredCredits,
|
||||
shortfallCredits: output.response.shortfallCredits,
|
||||
},
|
||||
creditBalance: output.response.creditBalance,
|
||||
requiredCredits: output.response.requiredCredits,
|
||||
shortfallCredits: output.response.shortfallCredits,
|
||||
};
|
||||
},
|
||||
);
|
||||
|
||||
const requestUnlockPaymentFromErrorAction = baseChatMachineSetup.assign(
|
||||
({ context }) => ({
|
||||
isUnlockingMessage: false,
|
||||
unlockMessageError: "unlock_failed",
|
||||
unlockPaywallRequest:
|
||||
context.unlockingMessageId && context.unlockingMessageKind
|
||||
? {
|
||||
displayMessageId: context.unlockingMessageId,
|
||||
...(context.unlockingRemoteMessageId
|
||||
? { messageId: context.unlockingRemoteMessageId }
|
||||
: {}),
|
||||
kind: context.unlockingMessageKind,
|
||||
...(context.unlockingLockType
|
||||
? { lockType: context.unlockingLockType }
|
||||
: {}),
|
||||
...(context.unlockingClientLockId
|
||||
? { clientLockId: context.unlockingClientLockId }
|
||||
: {}),
|
||||
...(context.promotion?.message.id === context.unlockingMessageId
|
||||
? { promotion: context.promotion.session }
|
||||
: {}),
|
||||
reason: "unlock_failed",
|
||||
creditBalance: context.creditBalance,
|
||||
requiredCredits: context.requiredCredits,
|
||||
shortfallCredits: context.shortfallCredits,
|
||||
}
|
||||
: null,
|
||||
unlockingMessageId: null,
|
||||
unlockingMessageKind: null,
|
||||
unlockingRemoteMessageId: null,
|
||||
unlockingLockType: null,
|
||||
unlockingClientLockId: null,
|
||||
}),
|
||||
);
|
||||
|
||||
const clearUnlockPaywallRequestAction = baseChatMachineSetup.assign(() => ({
|
||||
unlockPaywallRequest: null,
|
||||
}));
|
||||
|
||||
type ChatActorAction<TEvent extends EventObject> = ActionFunction<
|
||||
ChatState,
|
||||
TEvent,
|
||||
ChatEvent,
|
||||
undefined,
|
||||
never,
|
||||
never,
|
||||
never,
|
||||
never,
|
||||
never
|
||||
>;
|
||||
|
||||
function createChatActorActionSetup<TEvent extends EventObject>() {
|
||||
const actorActionSetup = setup({
|
||||
types: {
|
||||
context: {} as ChatState,
|
||||
events: {} as TEvent,
|
||||
},
|
||||
});
|
||||
return {
|
||||
assign(
|
||||
assignment: Parameters<typeof actorActionSetup.assign>[0],
|
||||
): ChatActorAction<TEvent> {
|
||||
// Actor lifecycle events are expression events, not public machine events.
|
||||
return actorActionSetup.assign(assignment) as unknown as ChatActorAction<TEvent>;
|
||||
},
|
||||
};
|
||||
}
|
||||
|
||||
const sendMessageDoneActionSetup =
|
||||
createChatActorActionSetup<DoneActorEvent<HttpSendOutput>>();
|
||||
const unlockHistoryDoneActionSetup =
|
||||
createChatActorActionSetup<DoneActorEvent<UnlockHistoryOutput>>();
|
||||
const actorErrorActionSetup = createChatActorActionSetup<ErrorActorEvent>();
|
||||
|
||||
const applyHttpSendOutputAction = sendMessageDoneActionSetup.assign(
|
||||
({ context, event }) => applyHttpSendOutput(context, event.output),
|
||||
);
|
||||
|
||||
const markHttpSendFailedAction = actorErrorActionSetup.assign({
|
||||
isReplyingAI: false,
|
||||
});
|
||||
|
||||
const applyUnlockHistoryOutputAction =
|
||||
unlockHistoryDoneActionSetup.assign(({ event }) => {
|
||||
const lockedHistoryCount = countLockedHistoryMessages(
|
||||
event.output.messages,
|
||||
);
|
||||
const insufficientBalance =
|
||||
!event.output.unlocked &&
|
||||
event.output.reason === "insufficient_balance";
|
||||
return {
|
||||
messages: event.output.messages,
|
||||
historyLoaded: true,
|
||||
paymentUnlockPending: false,
|
||||
unlockHistoryPromptVisible: insufficientBalance,
|
||||
lockedHistoryCount,
|
||||
isUnlockingHistory: false,
|
||||
unlockHistoryError: insufficientBalance
|
||||
? `Insufficient credits. You still need ${event.output.shortfallCredits} credits.`
|
||||
: null,
|
||||
};
|
||||
});
|
||||
|
||||
const chatMachineSetup = baseChatMachineSetup.extend({
|
||||
actions: {
|
||||
enqueueMessage: enqueueMessageAction,
|
||||
startGuestSession: startGuestSessionAction,
|
||||
startUserSession: startUserSessionAction,
|
||||
clearChatSession: clearChatSessionAction,
|
||||
injectPromotion: injectPromotionAction,
|
||||
clearPromotion: clearPromotionAction,
|
||||
applyLocalHistoryLoaded: applyLocalHistoryLoadedAction,
|
||||
applyNetworkHistoryLoaded: applyNetworkHistoryLoadedAction,
|
||||
applyNetworkHistoryLoadedAndClearPayment:
|
||||
applyNetworkHistoryLoadedAndClearPaymentAction,
|
||||
showUnlockHistoryPromptFromNetwork:
|
||||
showUnlockHistoryPromptFromNetworkAction,
|
||||
markHistoryLoadFailed: markHistoryLoadFailedAction,
|
||||
appendGuestUserMessage: appendGuestUserMessageAction,
|
||||
appendUserMessage: appendUserMessageAction,
|
||||
appendQueuedSendErrorMessage: appendQueuedSendErrorMessageAction,
|
||||
markQueuedSendStarted: markQueuedSendStartedAction,
|
||||
applyQueuedHttpOutput: applyQueuedHttpOutputAction,
|
||||
clearUpgradePrompt: clearUpgradePromptAction,
|
||||
appendGuestUserImage: appendGuestUserImageAction,
|
||||
appendUserImage: appendUserImageAction,
|
||||
markPaymentUnlockPending: markPaymentUnlockPendingAction,
|
||||
showUnlockHistoryPrompt: showUnlockHistoryPromptAction,
|
||||
dismissUnlockHistoryPrompt: dismissUnlockHistoryPromptAction,
|
||||
markUnlockHistoryStarted: markUnlockHistoryStartedAction,
|
||||
markUnlockHistoryFailed: markUnlockHistoryFailedAction,
|
||||
markUnlockMessageStarted: markUnlockMessageStartedAction,
|
||||
applyUnlockMessageSucceeded: applyUnlockMessageSucceededAction,
|
||||
requestUnlockPaymentFromOutput: requestUnlockPaymentFromOutputAction,
|
||||
requestUnlockPaymentFromError: requestUnlockPaymentFromErrorAction,
|
||||
clearUnlockPaywallRequest: clearUnlockPaywallRequestAction,
|
||||
},
|
||||
});
|
||||
|
||||
export const chatMachine = chatMachineSetup.createMachine({
|
||||
id: "chat",
|
||||
initial: "idle",
|
||||
|
||||
Reference in New Issue
Block a user