Format zed-local script

This commit is contained in:
Marshall Bowers 2023-12-15 16:27:58 -05:00
parent b1a61ca21e
commit 0ea59d6466

View file

@ -1,95 +1,86 @@
#!/usr/bin/env node #!/usr/bin/env node
const {spawn, execFileSync} = require('child_process') const { spawn, execFileSync } = require("child_process");
const RESOLUTION_REGEX = /(\d+) x (\d+)/ const RESOLUTION_REGEX = /(\d+) x (\d+)/;
const DIGIT_FLAG_REGEX = /^--?(\d+)$/ const DIGIT_FLAG_REGEX = /^--?(\d+)$/;
const ZED_2_MODE = "--zed2" const ZED_2_MODE = "--zed2";
const args = process.argv.slice(2) const args = process.argv.slice(2);
// Parse the number of Zed instances to spawn. // Parse the number of Zed instances to spawn.
let instanceCount = 1 let instanceCount = 1;
const digitMatch = args[0]?.match(DIGIT_FLAG_REGEX) const digitMatch = args[0]?.match(DIGIT_FLAG_REGEX);
if (digitMatch) { if (digitMatch) {
instanceCount = parseInt(digitMatch[1]) instanceCount = parseInt(digitMatch[1]);
args.shift() args.shift();
} }
const isZed2 = args.some(arg => arg === ZED_2_MODE); const isZed2 = args.some((arg) => arg === ZED_2_MODE);
if (instanceCount > 4) { if (instanceCount > 4) {
throw new Error('Cannot spawn more than 4 instances') throw new Error("Cannot spawn more than 4 instances");
} }
// Parse the resolution of the main screen // Parse the resolution of the main screen
const displayInfo = JSON.parse( const displayInfo = JSON.parse(
execFileSync( execFileSync("system_profiler", ["SPDisplaysDataType", "-json"], {
'system_profiler', encoding: "utf8",
['SPDisplaysDataType', '-json'], }),
{encoding: 'utf8'} );
) const mainDisplayResolution =
) displayInfo?.SPDisplaysDataType[0]?.spdisplays_ndrvs
const mainDisplayResolution = displayInfo ?.find((entry) => entry.spdisplays_main === "spdisplays_yes")
?.SPDisplaysDataType[0] ?._spdisplays_resolution?.match(RESOLUTION_REGEX);
?.spdisplays_ndrvs
?.find(entry => entry.spdisplays_main === "spdisplays_yes")
?._spdisplays_resolution
?.match(RESOLUTION_REGEX)
if (!mainDisplayResolution) { if (!mainDisplayResolution) {
throw new Error('Could not parse screen resolution') throw new Error("Could not parse screen resolution");
} }
const screenWidth = parseInt(mainDisplayResolution[1]) const screenWidth = parseInt(mainDisplayResolution[1]);
const screenHeight = parseInt(mainDisplayResolution[2]) const screenHeight = parseInt(mainDisplayResolution[2]);
// Determine the window size for each instance // Determine the window size for each instance
let instanceWidth = screenWidth let instanceWidth = screenWidth;
let instanceHeight = screenHeight let instanceHeight = screenHeight;
if (instanceCount > 1) { if (instanceCount > 1) {
instanceWidth = Math.floor(screenWidth / 2) instanceWidth = Math.floor(screenWidth / 2);
if (instanceCount > 2) { if (instanceCount > 2) {
instanceHeight = Math.floor(screenHeight / 2) instanceHeight = Math.floor(screenHeight / 2);
} }
} }
let users = [ let users = ["nathansobo", "as-cii", "maxbrunsfeld", "iamnbutler"];
'nathansobo',
'as-cii',
'maxbrunsfeld',
'iamnbutler'
]
const RUST_LOG = process.env.RUST_LOG || 'info' const RUST_LOG = process.env.RUST_LOG || "info";
// If a user is specified, make sure it's first in the list // If a user is specified, make sure it's first in the list
const user = process.env.ZED_IMPERSONATE const user = process.env.ZED_IMPERSONATE;
if (user) { if (user) {
users = [user].concat(users.filter(u => u !== user)) users = [user].concat(users.filter((u) => u !== user));
} }
const positions = [ const positions = [
'0,0', "0,0",
`${instanceWidth},0`, `${instanceWidth},0`,
`0,${instanceHeight}`, `0,${instanceHeight}`,
`${instanceWidth},${instanceHeight}` `${instanceWidth},${instanceHeight}`,
] ];
const buildArgs = isZed2 ? ["build", "-p", "zed2"] : ["build"] const buildArgs = isZed2 ? ["build", "-p", "zed2"] : ["build"];
const zedBinary = isZed2 ? "target/debug/Zed2" : "target/debug/Zed" const zedBinary = isZed2 ? "target/debug/Zed2" : "target/debug/Zed";
execFileSync('cargo', buildArgs, { stdio: 'inherit' }) execFileSync("cargo", buildArgs, { stdio: "inherit" });
setTimeout(() => { setTimeout(() => {
for (let i = 0; i < instanceCount; i++) { for (let i = 0; i < instanceCount; i++) {
spawn(zedBinary, i == 0 ? args : [], { spawn(zedBinary, i == 0 ? args : [], {
stdio: 'inherit', stdio: "inherit",
env: { env: {
ZED_IMPERSONATE: users[i], ZED_IMPERSONATE: users[i],
ZED_WINDOW_POSITION: positions[i], ZED_WINDOW_POSITION: positions[i],
ZED_STATELESS: '1', ZED_STATELESS: "1",
ZED_ALWAYS_ACTIVE: '1', ZED_ALWAYS_ACTIVE: "1",
ZED_SERVER_URL: 'http://localhost:8080', ZED_SERVER_URL: "http://localhost:8080",
ZED_ADMIN_API_TOKEN: 'secret', ZED_ADMIN_API_TOKEN: "secret",
ZED_WINDOW_SIZE: `${instanceWidth},${instanceHeight}`, ZED_WINDOW_SIZE: `${instanceWidth},${instanceHeight}`,
PATH: process.env.PATH, PATH: process.env.PATH,
RUST_LOG, RUST_LOG,
},
});
} }
}) }, 0.1);
}
}, 0.1)