newpr 1.0.10 → 1.0.12

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "newpr",
-  "version": "1.0.10",
+  "version": "1.0.12",
   "description": "AI-powered large PR review tool - understand PRs with 1000+ lines of changes",
   "module": "src/cli/index.ts",
   "type": "module",

package/src/web/client/App.tsx

@@ -200,7 +200,7 @@ export function App() {
 					cartoonEnabled={features.cartoon}
 					sessionId={diffSessionId}
 					onTabChange={handleTabChange}
-					onReanalyze={(prUrl: string) => { analysis.start(prUrl); }}
+					onReanalyze={(prUrl: string) => { analysis.start(prUrl, { reuseSessionId: diffSessionId }); }}
 					enabledPlugins={features.enabledPlugins}
 					onTrackAnalysis={bgAnalyses.track}
 				/>

package/src/web/client/components/ChatSection.tsx

@@ -92,35 +92,6 @@ function segmentsFromMessage(msg: ChatMessage): ChatSegment[] {
 	return segs;
 }
 
-function ThrottledMarkdown({ content, onAnchorClick, activeId }: {
-	content: string;
-	onAnchorClick?: (kind: "group" | "file" | "line", id: string) => void;
-	activeId?: string | null;
-}) {
-	const [rendered, setRendered] = useState(content);
-	const pendingRef = useRef(content);
-	const timerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
-
-	useEffect(() => {
-		pendingRef.current = content;
-		if (!timerRef.current) {
-			timerRef.current = setTimeout(() => {
-				setRendered(pendingRef.current);
-				timerRef.current = null;
-			}, 150);
-		}
-	}, [content]);
-
-	useEffect(() => {
-		return () => {
-			if (timerRef.current) clearTimeout(timerRef.current);
-			setRendered(pendingRef.current);
-		};
-	}, []);
-
-	return <Markdown onAnchorClick={onAnchorClick} activeId={activeId}>{rendered}</Markdown>;
-}
-
 function AssistantMessage({ segments, activeToolName, isStreaming, onAnchorClick, activeId }: {
 	segments: ChatSegment[];
 	activeToolName?: string;
@@ -137,10 +108,12 @@ function AssistantMessage({ segments, activeToolName, isStreaming, onAnchorClick
 					return <ToolCallDisplay key={seg.toolCall.id} tc={seg.toolCall} />;
 				}
 				if (!seg.content) return null;
+				const isTrailingSegment = i === segments.length - 1;
+				const isStreamingTail = isStreaming && isTrailingSegment;
 				return (
 					<div key={`text-${i}`} className="text-xs leading-relaxed">
-						{isStreaming ? (
-							<ThrottledMarkdown content={seg.content} onAnchorClick={onAnchorClick} activeId={activeId} />
+						{isStreamingTail ? (
+							<Markdown streaming onAnchorClick={onAnchorClick} activeId={activeId}>{seg.content}</Markdown>
 						) : (
 							<Markdown onAnchorClick={onAnchorClick} activeId={activeId}>{seg.content}</Markdown>
 						)}

package/src/web/client/components/Markdown.tsx

@@ -1,4 +1,4 @@
-import { useState, useEffect } from "react";
+import { memo, useState, useEffect, useMemo } from "react";
 import ReactMarkdown from "react-markdown";
 import remarkGfm from "remark-gfm";
 import remarkMath from "remark-math";
@@ -12,6 +12,7 @@ interface MarkdownProps {
 	children: string;
 	onAnchorClick?: (kind: "group" | "file" | "line", id: string) => void;
 	activeId?: string | null;
+	streaming?: boolean;
 }
 
 function useHighlighter(): Highlighter | null {
@@ -185,8 +186,8 @@ function preprocess(text: string): string {
 	return processed.replace(/\x00MATH_BLOCK_(\d+)\x00/g, (_, idx) => mathBlocks[Number(idx)]!);
 }
 
-export function Markdown({ children, onAnchorClick, activeId }: MarkdownProps) {
-	const processed = preprocess(children);
+export const Markdown = memo(function Markdown({ children, onAnchorClick, activeId, streaming = false }: MarkdownProps) {
+	const processed = useMemo(() => preprocess(children), [children]);
 	const hl = useHighlighter();
 	const dark = useDarkMode();
 
@@ -202,6 +203,12 @@ export function Markdown({ children, onAnchorClick, activeId }: MarkdownProps) {
 		strong: ({ children }) => <strong className="font-semibold text-foreground">{children}</strong>,
 		em: ({ children }) => <em className="italic">{children}</em>,
 		code: ({ children, className }) => {
+			if (streaming) {
+				if (className?.includes("language-")) {
+					return <code className="text-xs font-mono">{children}</code>;
+				}
+				return <code className="px-1.5 py-0.5 rounded bg-muted text-xs font-mono">{children}</code>;
+			}
 			const lang = langFromClassName(className);
 			if (lang && hl) {
 				const code = String(children).replace(/\n$/, "");
@@ -330,5 +337,11 @@ export function Markdown({ children, onAnchorClick, activeId }: MarkdownProps) {
 		td: ({ children }) => <td className="border-b border-border px-3 py-1.5 text-sm">{children}</td>,
 	};
 
-	return <ReactMarkdown remarkPlugins={[[remarkMath, { singleDollarTextMath: true }], remarkGfm]} rehypePlugins={[[rehypeRaw, { passThrough: ["math", "inlineMath"] }], [rehypeKatex, { throwOnError: false, strict: false, output: "html" }]]} components={components}>{processed}</ReactMarkdown>;
-}
+	return <ReactMarkdown
+		remarkPlugins={streaming ? [remarkGfm] : [[remarkMath, { singleDollarTextMath: true }], remarkGfm]}
+		rehypePlugins={streaming
+			? [[rehypeRaw, { passThrough: ["math", "inlineMath"] }]]
+			: [[rehypeRaw, { passThrough: ["math", "inlineMath"] }], [rehypeKatex, { throwOnError: false, strict: false, output: "html" }]]}
+		components={components}
+	>{processed}</ReactMarkdown>;
+});

package/src/web/client/hooks/useAnalysis.ts

@@ -29,7 +29,7 @@ export function useAnalysis() {
 	});
 	const eventSourceRef = useRef<EventSource | null>(null);
 
-	const start = useCallback(async (prInput: string) => {
+	const start = useCallback(async (prInput: string, options?: { reuseSessionId?: string | null }) => {
 		analytics.analysisStarted(0);
 		setState({
 			phase: "loading",
@@ -43,10 +43,11 @@ export function useAnalysis() {
 		});
 
 		try {
+			const reuseSessionId = options?.reuseSessionId ?? undefined;
 			const res = await fetch("/api/analysis", {
 				method: "POST",
 				headers: { "Content-Type": "application/json" },
-				body: JSON.stringify({ pr: prInput }),
+				body: JSON.stringify({ pr: prInput, reuseSessionId }),
 			});
 			const body = await res.json();
 			if (!res.ok) throw new Error(body.error ?? "Failed to start analysis");

package/src/web/server/routes.ts

@@ -332,7 +332,7 @@ Before posting an inline comment, ALWAYS call \`get_file_diff\` first to find th
 			const body = await req.json() as { pr: string; reuseSessionId?: string };
 			if (!body.pr) return json({ error: "Missing 'pr' field" }, 400);
 
-			const result = startAnalysis(body.pr, token, config);
+			const result = startAnalysis(body.pr, token, config, body.reuseSessionId);
 			if ("error" in result) return json({ error: result.error }, result.status);
 
 			return json({
@@ -953,9 +953,22 @@ Before posting an inline comment, ALWAYS call \`get_file_diff\` first to find th
 			const encoder = new TextEncoder();
 			const stream = new ReadableStream({
 				async start(controller) {
+					let closed = false;
 					const send = (eventType: string, data: string) => {
+						if (closed) return;
 						controller.enqueue(encoder.encode(`event: ${eventType}\ndata: ${data}\n\n`));
 					};
+					const safeClose = () => {
+						if (closed) return;
+						closed = true;
+						clearInterval(heartbeat);
+						setTimeout(() => { try { controller.close(); } catch {} }, 50);
+					};
+					const heartbeat = setInterval(() => {
+						if (closed) return;
+						try { controller.enqueue(encoder.encode(":keepalive\n\n")); } catch { safeClose(); }
+					}, 15_000);
+
 					try {
 						if (config.openrouter_api_key) {
 							await chatWithTools(
@@ -999,7 +1012,7 @@ Before posting an inline comment, ALWAYS call \`get_file_diff\` first to find th
 					} catch (err) {
 						send("chat_error", JSON.stringify({ message: err instanceof Error ? err.message : String(err) }));
 					} finally {
-						controller.close();
+						safeClose();
 					}
 				},
 			});
@@ -1424,9 +1437,21 @@ Before posting an inline comment, ALWAYS call \`get_file_diff\` first to find th
 			const encoder = new TextEncoder();
 			const stream = new ReadableStream({
 				async start(controller) {
+					let closed = false;
 					const send = (eventType: string, data: string) => {
+						if (closed) return;
 						controller.enqueue(encoder.encode(`event: ${eventType}\ndata: ${data}\n\n`));
 					};
+					const safeClose = () => {
+						if (closed) return;
+						closed = true;
+						clearInterval(heartbeat);
+						setTimeout(() => { try { controller.close(); } catch {} }, 50);
+					};
+					const heartbeat = setInterval(() => {
+						if (closed) return;
+						try { controller.enqueue(encoder.encode(":keepalive\n\n")); } catch { safeClose(); }
+					}, 15_000);
 
 					let fullText = "";
 					const collectedToolCalls: ChatToolCall[] = [];
@@ -1526,7 +1551,7 @@ Before posting an inline comment, ALWAYS call \`get_file_diff\` first to find th
 					} catch (err) {
 						send("chat_error", JSON.stringify({ message: err instanceof Error ? err.message : String(err) }));
 					} finally {
-						controller.close();
+						safeClose();
 					}
 				},
 			});

package/src/web/server/session-manager.ts

@@ -3,7 +3,7 @@ import type { NewprOutput } from "../../types/output.ts";
 import type { ProgressEvent } from "../../analyzer/progress.ts";
 import { analyzePr } from "../../analyzer/pipeline.ts";
 import { parsePrInput } from "../../github/parse-pr.ts";
-import { saveSession, savePatchesSidecar } from "../../history/store.ts";
+import { saveSession, savePatchesSidecar, loadSession, loadChatSidecar, saveChatSidecar } from "../../history/store.ts";
 import { telemetry } from "../../telemetry/index.ts";
 
 type SessionStatus = "running" | "done" | "error" | "canceled";
@@ -15,6 +15,7 @@ interface AnalysisSession {
 	events: ProgressEvent[];
 	result?: NewprOutput;
 	historyId?: string;
+	reuseSessionId?: string;
 	error?: string;
 	startedAt: number;
 	finishedAt?: number;
@@ -47,6 +48,7 @@ export function startAnalysis(
 	prInput: string,
 	token: string,
 	config: NewprConfig,
+	reuseSessionId?: string,
 ): { sessionId: string } | { error: string; status: number } {
 	if (runningCount() >= MAX_CONCURRENT) {
 		return { error: "Too many concurrent analyses. Try again later.", status: 429 };
@@ -58,6 +60,7 @@ export function startAnalysis(
 	const session: AnalysisSession = {
 		id,
 		prInput,
+		reuseSessionId,
 		status: "running",
 		events: [],
 		startedAt: Date.now(),
@@ -116,6 +119,15 @@ async function runPipeline(
 			if (Object.keys(capturedPatches).length > 0) {
 				await savePatchesSidecar(record.id, capturedPatches).catch(() => {});
 			}
+			if (session.reuseSessionId) {
+				const prior = await loadSession(session.reuseSessionId).catch(() => null);
+				if (prior?.meta.pr_url === result.meta.pr_url) {
+					const priorChat = await loadChatSidecar(session.reuseSessionId).catch(() => null);
+					if (priorChat && priorChat.length > 0) {
+						await saveChatSidecar(record.id, priorChat).catch(() => {});
+					}
+				}
+			}
 		}
 	} catch (err) {
 		const msg = err instanceof Error ? err.message : String(err);