fix headers in pgn

src/components/boards/BoardAnalysis.tsx

@@ -90,7 +90,7 @@ function BoardAnalysis({
     defaultValue: initial_tree,
     serialize: (value) => {
       const storedTree = JSON.stringify({
-        pgn: value.getTopVariation().getPGN(),
+        pgn: value.getTopVariation().getPGN({ headers: game }),
         currentMove: value.getPosition(),
       });
       return storedTree;
@@ -249,7 +249,10 @@ function BoardAnalysis({
       ],
     });
     if (filePath)
-      await writeTextFile(filePath, tree.getTopVariation().getPGN());
+      await writeTextFile(
+        filePath,
+        tree.getTopVariation().getPGN({ headers: game })
+      );
   }
 
   useHotkeys([
@@ -279,7 +282,12 @@ function BoardAnalysis({
   return (
     <TreeContext.Provider value={tree}>
       <ReportModal
-        moves={tree.getTopVariation().getPGN(false, false, false)}
+        moves={tree.getTopVariation().getPGN({
+          headers: game,
+          comments: false,
+          specialSymbols: false,
+          symbols: false,
+        })}
         reportingMode={reportingMode}
         toggleReportingMode={toggleReportingMode}
         setLoading={setAnalysisLoading}
@@ -327,7 +335,7 @@ function BoardAnalysis({
               <Stack>
                 <GameInfo game={game} />
                 <FenInput onSubmit={resetToFen} />
-                <PgnInput />
+                <PgnInput game={game} />
               </Stack>
             </Tabs.Panel>
             <Tabs.Panel value="database" pt="xs">
@@ -392,6 +400,7 @@ function BoardAnalysis({
           </Tabs>
           <Stack>
             <GameNotation
+              game={game}
               setTree={setTree}
               deleteVariation={deleteVariation}
               promoteVariation={promoteVariation}

src/components/boards/BoardGame.tsx

@@ -6,7 +6,7 @@ import {
   Select,
   SimpleGrid,
   Stack,
-  Text
+  Text,
 } from "@mantine/core";
 import { useHotkeys, useSessionStorage } from "@mantine/hooks";
 import {
@@ -114,7 +114,7 @@ function BoardGame({
     defaultValue: initial_tree,
     serialize: (value) => {
       const storedTree = JSON.stringify({
-        pgn: value.getTopVariation().getPGN(),
+        pgn: value.getTopVariation().getPGN({ headers: game }),
         currentMove: value.getPosition(),
       });
       return storedTree;
@@ -366,7 +366,7 @@ function BoardGame({
                     setCompleteGame((prev) => ({
                       game: {
                         ...prev.game,
-                        result: Outcome.Unknown
+                        result: Outcome.Unknown,
                       },
                       currentMove: [],
                     }));
@@ -385,6 +385,7 @@ function BoardGame({
                 </Button>
               </Group>
               <GameNotation
+                game={game}
                 setTree={setTree}
                 topVariation={tree.getTopVariation()}
                 result={Outcome.Unknown}

src/components/boards/GameNotation.tsx

@@ -12,7 +12,7 @@ import {
   Text,
   Tooltip,
   TypographyStylesProvider,
-  useMantineTheme
+  useMantineTheme,
 } from "@mantine/core";
 import { useForceUpdate, useToggle } from "@mantine/hooks";
 import {
@@ -27,10 +27,10 @@ import {
   IconEyeOff,
   IconMinus,
   IconPlus,
-  IconX
+  IconX,
 } from "@tabler/icons-react";
 import { VariationTree } from "../../utils/chess";
-import { Outcome } from "../../utils/db";
+import { NormalizedGame, Outcome } from "../../utils/db";
 import CompleteMoveCell from "./CompleteMoveCell";
 import OpeningName from "./OpeningName";
 
@@ -45,13 +45,15 @@ const useStyles = createStyles((theme) => ({
 }));
 
 function GameNotation({
+  game,
   setTree,
   topVariation,
   result,
   boardSize,
   deleteVariation,
   promoteVariation,
 }: {
+  game: NormalizedGame;
   setTree: (tree: VariationTree) => void;
   topVariation: VariationTree;
   deleteVariation?: () => void;
@@ -65,7 +67,7 @@ function GameNotation({
   const [showVariations, toggleVariations] = useToggle([true, false]);
   const [showComments, toggleComments] = useToggle([true, false]);
   const { classes } = useStyles();
-  const pgn = topVariation.getPGN();
+  const pgn = topVariation.getPGN({ headers: game });
 
   const multipleLine =
     topVariation.commentHTML.split("</p>").length - 1 > 1 ||

src/components/panels/info/PgnInput.tsx

@@ -1,9 +1,10 @@
 import { Checkbox, Group, Stack, Text, Textarea } from "@mantine/core";
 import { useToggle } from "@mantine/hooks";
 import { useContext } from "react";
+import { NormalizedGame } from "../../../utils/db";
 import TreeContext from "../../common/TreeContext";
 
-function PgnInput() {
+function PgnInput({ game }: { game: NormalizedGame }) {
   const tree = useContext(TreeContext);
   const [comments, toggleComments] = useToggle([true, false]);
   const [symbols, toggleSymbols] = useToggle([true, false]);
@@ -42,7 +43,13 @@ function PgnInput() {
         </Group>
         <Textarea
           readOnly
-          value={root.getPGN(symbols, comments, variations, specialSymbols)}
+          value={root.getPGN({
+            headers: game,
+            symbols,
+            comments,
+            variations,
+            specialSymbols,
+          })}
         />
       </Stack>
     </>

src/utils/chess.ts

@@ -7,11 +7,11 @@ import {
     PieceSymbol,
     ROOK,
     Square,
-    SQUARES
+    SQUARES,
 } from "chess.js";
 import { DrawShape } from "chessground/draw";
 import { Key } from "chessground/types";
-import { CompleteGame, Outcome } from "./db";
+import { CompleteGame, getHeaders, NormalizedGame, Outcome } from "./db";
 import { formatScore } from "./format";
 
 export type Score = {
@@ -229,43 +229,45 @@ export class VariationTree {
         return moveText;
     }
 
-    getPGN(
-        symbols: boolean = true,
-        comments: boolean = true,
-        variations: boolean = true,
-        specialSymbols: boolean = true
-    ): string {
+    getPGN({
+        headers,
+        symbols = true,
+        comments = true,
+        variations = true,
+        specialSymbols = true,
+    }: {
+        headers: NormalizedGame | null;
+        symbols?: boolean;
+        comments?: boolean;
+        variations?: boolean;
+        specialSymbols?: boolean;
+    }): string {
         let pgn = "";
+        if (headers) {
+            pgn += getHeaders(headers);
+        }
         if (this.parent === null) {
-            pgn += '[Event "?"]\n';
-            pgn += '[Site "?"]\n';
-            pgn += '[Date "?"]\n';
-            pgn += '[Round "?"]\n';
-            pgn += '[White "?"]\n';
-            pgn += '[Black "?"]\n';
-            pgn += '[Result "*"]\n';
             if (this.fen !== DEFAULT_POSITION) {
                 pgn += '[FEN "' + this.fen + '"]\n\n';
             } else {
                 pgn += "\n";
             }
         }
         const variationsPGN = variations
-            ? this.children
-                  .slice(1)
-                  .map(
-                      (variation) =>
-                          `${variation.getMoveText(
-                              symbols,
-                              comments,
-                              true
-                          )} ${variation.getPGN(
-                              symbols,
-                              comments,
-                              variations,
-                              specialSymbols
-                          )}`
-                  )
+            ? this.children.slice(1).map(
+                  (variation) =>
+                      `${variation.getMoveText(
+                          symbols,
+                          comments,
+                          true
+                      )} ${variation.getPGN({
+                          headers: null,
+                          symbols,
+                          comments,
+                          variations,
+                          specialSymbols,
+                      })}`
+              )
             : [];
         if (this.children.length > 0) {
             const child = this.children[0];
@@ -278,12 +280,13 @@ export class VariationTree {
         }
 
         if (this.children.length > 0) {
-            pgn += this.children[0].getPGN(
+            pgn += this.children[0].getPGN({
+                headers: null,
                 symbols,
                 comments,
                 variations,
-                specialSymbols
-            );
+                specialSymbols,
+            });
         }
         return pgn.trim();
     }

src/utils/db.ts

@@ -222,6 +222,18 @@ export interface NormalizedGame {
     moves: string;
 }
 
+export function getHeaders(game: NormalizedGame): string {
+    let headers = `[Event "${game.event.name}"]
+[Site "${game.site.name}"]
+[Date "${game.date}"]
+[Round "${game.round}"]
+[White "${game.white.name}"]
+[Black "${game.black.name}"]
+[Result "${game.result}"]
+`;
+    return headers;
+}
+
 export function defaultGame(): NormalizedGame {
     return {
         id: 0,