summaryrefslogtreecommitdiff
path: root/packages/cli/src/ui/contexts
diff options
context:
space:
mode:
Diffstat (limited to 'packages/cli/src/ui/contexts')
-rw-r--r--packages/cli/src/ui/contexts/SessionContext.test.tsx29
-rw-r--r--packages/cli/src/ui/contexts/SessionContext.tsx38
2 files changed, 67 insertions, 0 deletions
diff --git a/packages/cli/src/ui/contexts/SessionContext.test.tsx b/packages/cli/src/ui/contexts/SessionContext.test.tsx
new file mode 100644
index 00000000..3b5454cf
--- /dev/null
+++ b/packages/cli/src/ui/contexts/SessionContext.test.tsx
@@ -0,0 +1,29 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { render } from 'ink-testing-library';
+import { Text } from 'ink';
+import { SessionProvider, useSession } from './SessionContext.js';
+import { describe, it, expect } from 'vitest';
+
+const TestComponent = () => {
+ const { startTime } = useSession();
+ return <Text>{startTime.toISOString()}</Text>;
+};
+
+describe('SessionContext', () => {
+ it('should provide a start time', () => {
+ const { lastFrame } = render(
+ <SessionProvider>
+ <TestComponent />
+ </SessionProvider>,
+ );
+
+ const frameText = lastFrame();
+ // Check if the output is a valid ISO string, which confirms it's a Date object.
+ expect(new Date(frameText!).toString()).not.toBe('Invalid Date');
+ });
+});
diff --git a/packages/cli/src/ui/contexts/SessionContext.tsx b/packages/cli/src/ui/contexts/SessionContext.tsx
new file mode 100644
index 00000000..c511aa46
--- /dev/null
+++ b/packages/cli/src/ui/contexts/SessionContext.tsx
@@ -0,0 +1,38 @@
+/**
+ * @license
+ * Copyright 2025 Google LLC
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import React, { createContext, useContext, useState, useMemo } from 'react';
+
+interface SessionContextType {
+ startTime: Date;
+}
+
+const SessionContext = createContext<SessionContextType | null>(null);
+
+export const SessionProvider: React.FC<{ children: React.ReactNode }> = ({
+ children,
+}) => {
+ const [startTime] = useState(new Date());
+
+ const value = useMemo(
+ () => ({
+ startTime,
+ }),
+ [startTime],
+ );
+
+ return (
+ <SessionContext.Provider value={value}>{children}</SessionContext.Provider>
+ );
+};
+
+export const useSession = () => {
+ const context = useContext(SessionContext);
+ if (!context) {
+ throw new Error('useSession must be used within a SessionProvider');
+ }
+ return context;
+};