/*
 * 989_defdim_slice_run (#21) — slicing an array whose dimension is a `def`
 * constant (`[MAX]T`) must resolve the length AND the capacity from the type
 * table, not the AST dimension node.
 *
 * THE BUG (wwstage only): cgslice's default-hi and cgbasecap both read the
 * array dimension straight off the AST tnode and only handled an N_INTLIT
 * dim. A `def MAX` dim arrives as a non-literal node, so default-hi fell to
 * `MOVQ $0` (len 0 → underflow, exit 255) and cgbasecap returned false (cap
 * fell back to len). cstage reads the resolved bu->alen. THE FIX: when the
 * dim is not an N_INTLIT node, read the length from the stamped array tinfo
 * (rule-13), in both the local and global arms of both helpers.
 *
 *   row           | shape                              | exit (cs==ww)
 *   --------------+------------------------------------+--------------
 *   deflen_local  | def MAX=4; buf[1:].len (local)     | 3   (4-1)
 *   deflen_global | def MAX=4; g[1:].len (global)      | 3   (4-1)
 *   defcap        | def MAX=6; s=buf[2:4]; len*100+cap | 204 (len2,cap4)
 *   defcap_global | def MAX=6; g[2:4]; len*100+cap     | 204 (cgbasecap global)
 * Pre-fix: deflen_* ww=255 (len 0/underflow); defcap cap diverged from cs.
 * All four cgen arms (default-hi local+global, cgbasecap local+global) are
 * pinned: defcap teeth the cap≠len word the global arm could drop.
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/stat.h>
#include <sys/wait.h>

static int
runwait(const char *cmd)
{
	int rc = system(cmd);
	if (rc == -1) return -1;
	if (WIFEXITED(rc)) return WEXITSTATUS(rc);
	return -1;
}

struct row { const char *label; const char *src; int want_exit; };

static const struct row rows[] = {
	{ "deflen_local",
	  "package main;\n"
	  "def MAX: i32 = 4;\n"
	  "export fn main() i32 = {\n"
	  "	let buf: [MAX]u8 = [10, 20, 30, 40];\n"
	  "	let s: []u8 = buf[1:];\n"
	  "	return s.len: i32;\n"
	  "};\n",
	  3 },

	{ "deflen_global",
	  "package main;\n"
	  "def MAX: i32 = 4;\n"
	  "let g: [MAX]u8 = [10, 20, 30, 40];\n"
	  "export fn main() i32 = {\n"
	  "	let s: []u8 = g[1:];\n"
	  "	return s.len: i32;\n"
	  "};\n",
	  3 },

	{ "defcap",
	  "package main;\n"
	  "def MAX: i32 = 6;\n"
	  "export fn main() i32 = {\n"
	  "	let buf: [MAX]u8 = [1, 2, 3, 4, 5, 6];\n"
	  "	let s: []u8 = buf[2:4];\n"
	  "	return (s.len * 100 + s.cap): i32;\n"
	  "};\n",
	  204 },

	{ "defcap_global",
	  "package main;\n"
	  "def MAX: i32 = 6;\n"
	  "let g: [MAX]u8 = [1, 2, 3, 4, 5, 6];\n"
	  "export fn main() i32 = {\n"
	  "	let s: []u8 = g[2:4];\n"
	  "	return (s.len * 100 + s.cap): i32;\n"
	  "};\n",
	  204 },
};

static int
run_build(const char *driver, const struct row *r, int i)
{
	char src[64], tmpdir[64], cmd[1024];
	snprintf(src, sizeof src, "/tmp/dds_%d_%d.ww", getpid(), i);
	snprintf(tmpdir, sizeof tmpdir, "/tmp/dds_%d_d_%d", getpid(), i);

	FILE *f = fopen(src, "wb");
	if (!f) return -2;
	fputs(r->src, f);
	fclose(f);

	mkdir(tmpdir, 0755);
	snprintf(cmd, sizeof cmd, "cd %s && %s build %s 2>/dev/null",
	    tmpdir, driver, src);
	int brc = runwait(cmd);

	const char *base = strrchr(src, '/');
	base = base ? base + 1 : src;
	char outbin[128];
	snprintf(outbin, sizeof outbin, "%s/%s", tmpdir, base);
	char *dot = strrchr(outbin, '.');
	if (dot && strcmp(dot, ".ww") == 0) *dot = '\0';

	int got = -1;
	if (brc == 0) got = runwait(outbin);

	unlink(src); unlink(outbin); rmdir(tmpdir);
	return brc == 0 ? got : -1;
}

int
main(void)
{
	const char *bin = getenv("BIN");
	if (!bin) bin = "out/bin";
	char absbin[1024];
	if (bin[0] != '/') {
		char cwd[1024];
		if (getcwd(cwd, sizeof cwd) == NULL) return 1;
		snprintf(absbin, sizeof absbin, "%s/%s", cwd, bin);
		bin = absbin;
	}

	char cdrv[1024], wdrv[1024];
	snprintf(cdrv, sizeof cdrv, "%s/ww", bin);
	snprintf(wdrv, sizeof wdrv, "%s/ww_ww", bin);
	int have_ww = (access(wdrv, X_OK) == 0);

	int n = (int)(sizeof rows / sizeof rows[0]);
	int total = 0, fail = 0;

	for (int i = 0; i < n; i++) {
		total++;
		int gc = run_build(cdrv, &rows[i], i);
		if (gc < 0) {
			fprintf(stderr, "defdim_slice[cstage][%s]: build/run failed "
			    "(got %d)\n", rows[i].label, gc);
			fail++;
			continue;
		}
		if (gc != rows[i].want_exit) {
			fprintf(stderr, "defdim_slice[cstage][%s]: exit=%d want=%d\n",
			    rows[i].label, gc, rows[i].want_exit);
			fail++;
		}
		if (!have_ww) {
			fprintf(stderr, "defdim_slice: skip wwstage (no %s)\n", wdrv);
			continue;
		}
		int gw = run_build(wdrv, &rows[i], i);
		if (gw != gc) {
			fprintf(stderr, "defdim_slice[%s]: cs=%d != ww=%d "
			    "(def-dim length/cap divergence — #21)\n",
			    rows[i].label, gc, gw);
			fail++;
		}
		if (gw != rows[i].want_exit) {
			fprintf(stderr, "defdim_slice[wwstage][%s]: exit=%d want=%d\n",
			    rows[i].label, gw, rows[i].want_exit);
			fail++;
		}
	}

	if (fail) {
		fprintf(stderr, "defdim_slice_run: %d/%d checks failed\n",
		    fail, total);
		return 1;
	}
	printf("defdim_slice_run: %d/%d ok\n", total, total);
	return 0;
}