Skip to content

Commit b28c96e

Browse files
clarify variable name
1 parent 644f902 commit b28c96e

File tree

1 file changed

+11
-9
lines changed

1 file changed

+11
-9
lines changed

src/hooks/useDiagram.ts

Lines changed: 11 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,11 @@ export function useDiagram(username: string, repo: string) {
4848
const [showApiKeyDialog, setShowApiKeyDialog] = useState(false);
4949
// const [tokenCount, setTokenCount] = useState<number>(0);
5050
const [state, setState] = useState<StreamState>({ status: "idle" });
51-
const [hasFreeGeneration, setHasFreeGeneration] = useState<boolean>(() => {
52-
return localStorage.getItem("has_used_free_generation") === "true";
53-
});
51+
const [hasUsedFreeGeneration, setHasUsedFreeGeneration] = useState<boolean>(
52+
() => {
53+
return localStorage.getItem("has_used_free_generation") === "true";
54+
},
55+
);
5456

5557
const generateDiagram = useCallback(
5658
async (instructions = "", githubPat?: string) => {
@@ -188,12 +190,12 @@ export function useDiagram(username: string, repo: string) {
188190
});
189191
const date = await getLastGeneratedDate(username, repo);
190192
setLastGenerated(date ?? undefined);
191-
if (!hasFreeGeneration) {
193+
if (!hasUsedFreeGeneration) {
192194
localStorage.setItem(
193195
"has_used_free_generation",
194196
"true",
195197
);
196-
setHasFreeGeneration(true);
198+
setHasUsedFreeGeneration(true);
197199
}
198200
break;
199201
case "error":
@@ -223,7 +225,7 @@ export function useDiagram(username: string, repo: string) {
223225
setLoading(false);
224226
}
225227
},
226-
[username, repo, hasFreeGeneration],
228+
[username, repo, hasUsedFreeGeneration],
227229
);
228230

229231
useEffect(() => {
@@ -265,7 +267,7 @@ export function useDiagram(username: string, repo: string) {
265267

266268
// Only check for API key if we need to generate a new diagram
267269
const storedApiKey = localStorage.getItem("openrouter_key");
268-
if (hasFreeGeneration && !storedApiKey) {
270+
if (hasUsedFreeGeneration && !storedApiKey) {
269271
setError(
270272
"You've used your one free diagram. Please enter your API key to continue. As a student, I can't afford to keep it totally free and I hope you understand :)",
271273
);
@@ -304,7 +306,7 @@ export function useDiagram(username: string, repo: string) {
304306
} finally {
305307
setLoading(false);
306308
}
307-
}, [username, repo, generateDiagram, hasFreeGeneration]);
309+
}, [username, repo, generateDiagram, hasUsedFreeGeneration]);
308310

309311
useEffect(() => {
310312
void getDiagram();
@@ -350,7 +352,7 @@ export function useDiagram(username: string, repo: string) {
350352
const storedApiKey = localStorage.getItem("openrouter_key");
351353

352354
// Check if user has used their free generation and doesn't have an API key
353-
if (hasFreeGeneration && !storedApiKey) {
355+
if (hasUsedFreeGeneration && !storedApiKey) {
354356
setError(
355357
"You've used your one free diagram. Please enter your API key to continue. As a student, I can't afford to keep it totally free and I hope you understand :)",
356358
);

0 commit comments

Comments
 (0)