gsender

export type Operation = "rectangle" | "drill-grid";

export type Units = "mm" | "in";

type ToolParams = {
	feedrate: number;
	plungeRate: number;
	spindleRpm: number;
	safeZ: number;
	units: Units;
};

export type RectangleParams = ToolParams & {
	operation: "rectangle";
	width: number;
	height: number;
	depth: number;
};

export type DrillGridParams = ToolParams & {
	operation: "drill-grid";
	cols: number;
	rows: number;
	spacingX: number;
	spacingY: number;
	depth: number;
};

export type GcodeParams = RectangleParams | DrillGridParams;

const unitsCode = (units: Units) => (units === "in" ? "G20" : "G21");

const generateRectangleProfile = ({
	width,
	height,
	depth,
	feedrate,
	plungeRate,
	spindleRpm,
	safeZ,
	units,
}: RectangleParams): string =>
	[
		"(Basic CAM - Rectangle profile)",
		"(Generated by gSender Basic CAM plugin)",
		unitsCode(units),
		"G90",
		`M3 S${spindleRpm}`,
		`G0 Z${safeZ}`,
		"G0 X0 Y0",
		`G1 F${plungeRate} Z-${depth}`,
		`G1 F${feedrate} X${width} Y0`,
		`G1 X${width} Y${height}`,
		`G1 X0 Y${height}`,
		"G1 X0 Y0",
		`G0 Z${safeZ}`,
		"M5",
		"M2",
	].join("\n");

const generateDrillGrid = ({
	cols,
	rows,
	spacingX,
	spacingY,
	depth,
	plungeRate,
	spindleRpm,
	safeZ,
	units,
}: DrillGridParams): string => {
	const c = Math.max(1, Math.floor(cols));
	const r = Math.max(1, Math.floor(rows));

	const lines = [
		"(Basic CAM - Drill grid)",
		"(Generated by gSender Basic CAM plugin)",
		unitsCode(units),
		"G90",
		`M3 S${spindleRpm}`,
		`G0 Z${safeZ}`,
		`G1 F${plungeRate}`,
	];

	for (let row = 0; row < r; row += 1) {
		for (let col = 0; col < c; col += 1) {
			const x = col * spacingX;
			const y = row * spacingY;
			lines.push(`G0 X${x} Y${y}`);
			lines.push(`G1 Z-${depth}`);
			lines.push(`G0 Z${safeZ}`);
		}
	}

	lines.push("M5", "M2");
	return lines.join("\n");
};

export const generateGcode = (params: GcodeParams): string =>
	params.operation === "drill-grid"
		? generateDrillGrid(params)
		: generateRectangleProfile(params);