removed document file generation

Author: Daniel1464

src/codegen/genConstsFile.ts

@@ -2,14 +2,13 @@ import { Project } from "../document/2025/DocumentTypes";
 import { round, writeConst } from "./internals";
 
 export function genVarsFile(project: Project, packageName: string): string {
-  const expressions = project.variables.expressions;
-  const poses = project.variables.poses;
   const out: string[] = [];
-  out.push(`package ${packageName};\n`);
+  out.push(`package ${packageName};`);
   out.push(`
 import edu.wpi.first.math.geometry.Pose2d;
 import edu.wpi.first.math.geometry.Rotation2d;
 import edu.wpi.first.units.measure.*;
+
 import static edu.wpi.first.units.Units.*;
 
 /**
@@ -18,22 +17,20 @@ import static edu.wpi.first.units.Units.*;
  * in the choreo GUI.
  */
 public final class ChoreoVars {`);
-  for (const varName in project.variables.expressions) {
-    const data = expressions[varName];
+  for (const [varName, data] of Object.entries(project.variables.expressions)) {
     out.push(writeConst(data.var, varName, data.dimension));
   }
   out.push("");
   out.push("    public static final class Poses {");
-  for (const poseName in poses) {
-    const pose = poses[poseName];
+  for (const [varName, pose] of Object.entries(project.variables.poses)) {
     const heading =
       Math.abs(pose.heading.val) < 1e-5
         ? "Rotation2d.kZero"
         : `Rotation2d.fromRadians(${round(pose.heading.val, 3)})`;
     const x = round(pose.x.val, 3);
     const y = round(pose.y.val, 3);
     out.push(
-      `        public static final Pose2d ${poseName} = new Pose2d(${x}, ${y}, ${heading});`
+      `        public static final Pose2d ${varName} = new Pose2d(${x}, ${y}, ${heading});`
     );
   }
   out.push("");

src/codegen/genVarsFile.ts

@@ -46,6 +46,7 @@ export function writeConst(
   variableName: string,
   dimension: DimensionName
 ): string {
+  variableName = variableName.replace(/\b(\w)/g, (char) => char.toLowerCase());
   const unitData = unitDataFrom(dimension);
   let val = typeof expr === "number" ? expr : expr.val;
   val = round(val, dimension === "MoI" ? 5 : 3);
@@ -55,10 +56,6 @@ export function writeConst(
   return `    public static final ${unitData.type} ${variableName} = ${unitData.baseUnit}.of(${val});`;
 }
 
-export function writeVar(variable: Variable, name: string): string {
-  return writeConst(variable.var, name, variable.dimension);
-}
-
 export function round(val: number, digits: number) {
   const roundingFactor = Math.pow(10, digits);
   return Math.round(val * roundingFactor) / roundingFactor;

src/codegen/internals.ts

@@ -64,7 +64,6 @@ import { SavingState, UIStateStore } from "./UIStateStore";
 import { findUUIDIndex } from "./path/utils";
 import { ChoreoError, Commands } from "./tauriCommands";
 import { tracing } from "./tauriTracing";
-import { genConstsFile } from "../codegen/genConstsFile";
 import { genVarsFile } from "../codegen/genVarsFile";
 
 export type OpenFilePayload = {
@@ -716,8 +715,7 @@ export async function newProject(promptForCodegen: boolean = true) {
     (Do this if you're switching to a new codebase)
     `
     : `
-    Choreo can now output Java files
-    containing variables and constants defined in the GUI.
+    Choreo can generate a java file with variables defined in the GUI.
     Select a folder to enable this feature?
     `;
   if (await ask(msg, { title: "Choreo", kind: "warning" })) {
@@ -808,16 +806,13 @@ export async function saveProject() {
       );
       let tasks = [Commands.writeProject(data)];
       if (javaRoot && packageName) {
-        const constsFile = genConstsFile(data, packageName);
-        const varsFile = genVarsFile(data, packageName);
         tasks = [
           ...tasks,
           Commands.writeRawFile(
-            constsFile,
-            javaRoot + packageName + "/ChoreoConsts.java"
-          ),
-          Commands.writeRawFile(
-            varsFile,
+            genVarsFile(
+              data,
+              packageName.replaceAll("/", ".").replaceAll("\\", ".")
+            ),
             javaRoot + packageName + "/ChoreoVars.java"
           )
         ];
@@ -851,6 +846,7 @@ export async function codeGenDialog() {
   }
   localStorage.setItem(LocalStorageKeys.JAVA_ROOT, splitPath[0] + "/java/");
   localStorage.setItem(LocalStorageKeys.CODE_GEN_PACKAGE, splitPath[1]);
+  await saveProject();
 }
 
 export async function disableCodegen() {