raylib-wasm-transpiler/tools/js-class-c-struct-transpiler/transpile.js

217 lines
4.9 KiB
JavaScript
Raw Normal View History

2024-05-27 18:12:34 -04:00
import { parseArgs } from "util";
let types = {
u8: {
c: "unsigned char",
2024-05-27 18:12:34 -04:00
js: "Uint8",
2024-05-27 20:19:37 -04:00
sql: "INTEGER",
2024-05-27 18:12:34 -04:00
size: 1,
},
i8: {
c: "char",
2024-05-27 18:12:34 -04:00
js: "Int8",
2024-05-27 20:19:37 -04:00
sql: "INTEGER",
2024-05-27 18:12:34 -04:00
size: 1,
},
u16: {
c: "unsigned short",
2024-05-27 18:12:34 -04:00
js: "Uint16",
2024-05-27 20:19:37 -04:00
sql: "INTEGER",
2024-05-27 18:12:34 -04:00
size: 2,
},
i16: {
c: "short",
2024-05-27 18:12:34 -04:00
js: "Int16",
2024-05-27 20:19:37 -04:00
sql: "INTEGER",
2024-05-27 18:12:34 -04:00
size: 2,
},
u32: {
c: "unisgned int",
2024-05-27 18:12:34 -04:00
js: "Uint32",
2024-05-27 20:19:37 -04:00
sql: "INTEGER",
2024-05-27 18:12:34 -04:00
size: 4,
},
i32: {
c: "int",
2024-05-27 18:12:34 -04:00
js: "Int32",
2024-05-27 20:19:37 -04:00
sql: "INTEGER",
2024-05-27 18:12:34 -04:00
size: 4,
},
2024-05-27 20:19:37 -04:00
i64: {
c: "long",
js: "BigInt64",
sql: "INTEGER"
},
u64: {
c: "unsigned long",
js: "BigUint64",
sql: "INTEGER"
},
2024-05-27 18:12:34 -04:00
f32: {
c: "float",
js: "Float32",
2024-05-27 20:19:37 -04:00
sql: "REAL",
2024-05-27 18:12:34 -04:00
size: 4,
},
f64: {
c: "double",
js: "Float64",
2024-05-27 20:19:37 -04:00
sql: "REAL",
2024-05-27 18:12:34 -04:00
size: 8,
},
};
const { values } = parseArgs({
args: Bun.argv,
options: {
schema: {
type: "string",
short: "S",
},
2024-05-27 20:19:37 -04:00
javascript_out: {
2024-05-27 19:13:13 -04:00
type: "string",
short: "j",
},
2024-05-27 20:19:37 -04:00
c_out: {
2024-05-27 19:13:13 -04:00
type: "string",
short: "c",
},
2024-05-27 20:19:37 -04:00
sqlite_out: {
type: "string",
short: "s"
}
2024-05-27 18:12:34 -04:00
},
strict: true,
allowPositionals: true,
});
function jsStructConstructor(size, containsString) {
return `
get bytes() {
return new Uint8Array(this._ptr);
}
constructor(init = {}, ptr = undefined) {
this._size = ${size};
this._ptr = ptr.buffer || new ArrayBuffer(this._size);
this._data = new DataView(this._ptr);
${
containsString
? `
this._encoder = new TextEncoder();
this._decoder = new TextDecoder();`
: ""
}
for (const key of Object.keys(init)) {
this[key] = init[key];
}
2024-05-27 18:12:34 -04:00
}`;
}
const sFile = Bun.file(values.schema);
const schema = await sFile.json();
let cData = "";
2024-05-27 20:19:37 -04:00
let sqlData = "";
2024-05-27 18:12:34 -04:00
for (const type of Object.keys(schema)) {
let containsString = false;
let offset = 0;
let size = 0;
2024-05-27 19:08:52 -04:00
let importStatements = "";
2024-05-27 18:12:34 -04:00
let jsData = "";
2024-05-27 20:19:37 -04:00
let foreignKeys = "";
2024-05-27 18:12:34 -04:00
const props = schema[type];
2024-05-27 20:19:37 -04:00
sqlData += `CREATE TABLE ${type} (${type.toLowerCase()}_id INTEGER PRIMARY KEY AUTOINCREMENT`;
2024-05-27 18:12:34 -04:00
cData += `typedef struct ${type} {`;
jsData += `class ${type} {`;
for (const prop of Object.keys(props)) {
const propType = props[prop].type;
const kind = props[prop].kind;
let typeSize = parseInt(types[propType]?.size);
switch (kind) {
case "string":
containsString = true;
typeSize = props[prop].size;
const iSize = parseInt(offset) + parseInt(typeSize);
jsData += `
get ${prop}() {
return this._decoder.decode(new Uint8Array(this._ptr.slice(${parseInt(
offset
)}, ${iSize})));
}
set ${prop}(v) {
this._data.set(this._encoder.encode(v), ${parseInt(offset)});
2024-05-27 18:28:27 -04:00
}`;
2024-05-27 20:19:37 -04:00
sqlData += `, ${propType} TEXT`;
2024-05-27 18:12:34 -04:00
cData += `
char ${prop}[${iSize}];`;
break;
case "scalar":
typeSize = types[propType].size;
jsData += `
get ${prop}() {
return this._data.get${types[propType].js}(${parseInt(offset)}, true);
}
set ${prop}(v) {
return this._data.set${types[propType].js}(${parseInt(
offset
)}, v, true);
}`;
2024-05-27 20:19:37 -04:00
sqlData += `, ${prop} ${types[propType].sql}`;
2024-05-27 18:12:34 -04:00
cData += `
${types[propType].c} ${prop};`;
break;
case "struct":
const jsSize = parseInt(offset) + parseInt(types[propType].size);
jsData += `
get ${prop}() {
return new ${propType}({}, new Uint8Array(this._ptr.slice(${offset}, ${jsSize})));
}
set ${prop}(v) {
this._data.set(v.bytes(), ${offset});
}`;
2024-05-27 19:13:13 -04:00
const importS = `import ${propType} from "./${propType}"\n\n`;
if (!importStatements.includes(importS)) {
importStatements += importS;
}
2024-05-27 20:19:37 -04:00
const foreignKey = `${propType.toLowerCase()}_id`;
const localKey = `${prop.toLowerCase()}_${foreignKey}`;
2024-05-27 18:12:34 -04:00
2024-05-27 20:29:39 -04:00
sqlData += `, ${localKey} INTEGER`;
foreignKeys += `\n, FOREIGN KEY(${localKey}) REFERENCES ${propType}(${foreignKey})`
2024-05-27 19:08:52 -04:00
cData += `\n\t\t${types[propType].c} ${prop};`;
2024-05-27 18:12:34 -04:00
break;
case "array":
throw new Error("Not Implemented!");
2024-05-27 18:12:34 -04:00
break;
default:
break;
}
size += parseInt(typeSize);
offset += parseInt(typeSize);
}
/**
* add the new type to the list so we can use it for structs later.
*/
types[type] = {
c: type,
js: type,
size: parseInt(size),
};
jsData += jsStructConstructor(size, containsString);
2024-05-27 20:19:37 -04:00
jsData += `\n}\n\nexport default ${type}`;
cData += `\n} ${type};\n\n`;
2024-05-27 20:29:39 -04:00
sqlData += `${foreignKeys});\n\n`;
2024-05-27 18:12:34 -04:00
2024-05-27 19:13:13 -04:00
await Bun.write(
2024-05-27 20:19:37 -04:00
Bun.file(values.javascript_out + type + ".js"),
2024-05-27 19:13:13 -04:00
importStatements + jsData
);
2024-05-27 18:12:34 -04:00
}
2024-05-27 20:19:37 -04:00
await Bun.write(Bun.file(values.c_out + "types.h"), cData);
await Bun.write(Bun.file(values.sqlite_out + "types.sql"), sqlData);