Something went wrong. Try again.
Starship Simulator Controls thoriumsim.com
Something went wrong. Try again.
16 kB · 511 lines
TSX
at develop
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512import { pubsub } from "@thorium/.server/init/pubsub";import { t } from "@thorium/.server/init/t";import { getClassification } from "@thorium/cards/Navigation/getObjectClassification.server";import type { position } from "@thorium/ecs-components/position";import { type ECS, Entity } from "@thorium/utils/ecs";import { getOrbitPosition } from "@thorium/utils/starmap/getOrbitPosition";import { getCompletePositionFromOrbit, getObjectOffsetPosition, getObjectSystem,} from "@thorium/utils/starmap/position";import { matchSorter } from "match-sorter";import z from "zod";
function isWaypointLocked(ecs: ECS, waypointId: number, shipId: number): boolean { const ship = ecs.getEntityById(shipId); return ship?.components.autopilot?.destinationWaypointId === waypointId;}
type Waypoint = { id: number; name: string; objectId?: number; permanent: boolean; isActive: boolean; lastInteractedAt: number; position: z.infer<typeof position>; systemPosition: z.infer<typeof position> | null;};
export const navigation = t.router({ ship: t.procedure .input(z.object({ shipId: z.number() })) .filter((publish: { shipId: number }, { input }) => { if (publish && "shipId" in publish && publish.shipId !== input.shipId) return false;
return true; }) .autoPublish( ["isShip", "size", "identity", "position"], (entity) => entity.components.isShip && { shipId: entity.id }, ) .request(({ ctx, input }) => { const ship = ctx.ecs.getEntityById(input.shipId); if (!ship) throw new Error("No ship assigned"); return { id: ship.id, name: ship.components.identity?.name, position: ship.components.position, icon: ship.components.isShip?.assets.logo, size: ship.components.size?.length || 350, }; }), object: t.procedure .input(z.object({ shipId: z.number(), objectId: z.number().optional() })) .filter((publish: { shipId: number }, { input }) => { if (publish && publish.shipId !== input.shipId) return false; return true; }) // No need to auto-publish this one. .autoPublish([], () => null)
.request(({ ctx, input }) => { const ship = ctx.ecs.getEntityById(input.shipId); if (!ship) throw new Error("Ship not found");
const shipSolarSystem = getObjectSystem(ship); const object = ctx.ecs.getEntityById(input?.objectId || -1);
if (!object) return { object: null, objectSystem: null, shipSystem: shipSolarSystem?.components.position ? { id: shipSolarSystem?.id, ...shipSolarSystem?.components.position, } : null, }; const objectSystem = getObjectSystem(object); const position = object.components.position || (object.components.satellite ? getOrbitPosition(object.components.satellite) : undefined); return { object: { position, name: object.components.identity?.name, classification: getClassification(object), type: object.components.isShip ? "ship" : object.components.isPlanet ? "planet" : object.components.isStar ? "star" : object.components.isSolarSystem ? "solarSystem" : "unknown", vanity: object.components.isShip?.assets.vanity, hue: object.components.isStar?.hue, isWhite: object.components.isStar?.isWhite, cloudMapAsset: object.components.isPlanet?.cloudMapAsset, ringMapAsset: object.components.isPlanet?.ringMapAsset, textureMapAsset: object.components.isPlanet?.textureMapAsset, }, objectSystem: objectSystem?.components.position ? { id: objectSystem?.id, ...objectSystem.components.position } : null, shipSystem: shipSolarSystem?.components.position ? { id: shipSolarSystem?.id, ...shipSolarSystem.components.position } : null, }; }),
search: t.procedure .input(z.object({ query: z.string() })) // No need to auto-publish this one. .autoPublish([], () => null) .request(({ ctx, input }) => { const { query } = input;
const matchedEntities = [ ...(ctx.ecs.componentCache.get("isStar") || []), ...(ctx.ecs.componentCache.get("isPlanet") || []), ...(ctx.ecs.componentCache.get("isSolarSystem") || []), ]; // Get all of the planet, star, and solar system entities that match the query. const matchItems = matchSorter( matchedEntities.map((m) => { let position = m.components.position; if (!position) { const { x, y, z } = getCompletePositionFromOrbit(m); const parentId = getObjectSystem(m)?.id || null; position = { x, y, z, type: m.components.isSolarSystem ? "interstellar" : "solar", parentId: m.components.isSolarSystem ? null : parentId, }; } return { ...m, type: m.components.isSolarSystem ? "solar" : m.components.isPlanet ? "planet" : m.components.isShip ? "ship" : "star", name: m.components.identity!.name, description: m.components.identity?.description, temperature: m.components.temperature?.temperature, spectralType: m.components.isStar?.spectralType, classification: m.components.isPlanet?.classification, mass: m.components.isStar?.solarMass || m.components.isPlanet?.terranMass, population: m.components.population?.count, position, } as const; }) || [], query, { keys: [ "name", "description", "temperature", "spectralType", "classification", "mass", "population", ], }, ).map((m) => ({ // TODO Aug 1 2022 - Add in a distance calculation. id: m.id, name: m.name, position: m.position, type: m.type, }));
return matchItems; }), stream: t.procedure.input(z.object({ shipId: z.number() })).dataStream(({ ctx, input }) => { const set = new Set<Entity>(); const entity = ctx.ecs.getEntityById(input.shipId); if (entity) set.add(entity); return set; }),});
export const waypoints = t.router({ all: t.procedure .input( z.object({ shipId: z.number(), active: z.boolean(), systemId: z.union([z.literal("all"), z.number(), z.null()]), }), ) .filter((publish: { shipId: number } | null, { input }) => { if (publish && input.shipId !== publish.shipId) return false; return true; }) .autoPublish( ["isWaypoint", "identity", "position"], (entity) => entity.components.isWaypoint && { shipId: entity.components.isWaypoint.assignedShipId, }, ) .request(({ ctx, input: { shipId, systemId, active } }) => { const waypoints: Waypoint[] = []; for (const waypoint of ctx.ecs.componentCache.get("isWaypoint") || []) { if ( waypoint.components.isWaypoint?.assignedShipId === shipId && (systemId === "all" || waypoint.components.position?.parentId === systemId) ) { if (active && !waypoint.components.isWaypoint.isActive) continue; if (waypoint.components.position) { const systemPosition = ctx.flight?.ecs.getEntityById(waypoint.components.position.parentId || -1)?.components .position || null; waypoints.push({ id: waypoint.id, name: waypoint.components.identity?.name || "", objectId: waypoint.components.isWaypoint?.attachedObjectId, position: waypoint.components.position, permanent: waypoint.components.isWaypoint.permanent, isActive: waypoint.components.isWaypoint.isActive, lastInteractedAt: waypoint.components.isWaypoint.lastInteractedAt, systemPosition, }); } } }
// Sort by most recently interacted first waypoints.sort((a, b) => b.lastInteractedAt - a.lastInteractedAt);
return waypoints; }), spawn: t.procedure
.input( z.object({ shipId: z.number(),
entityId: z.number().optional(), position: z .object({ parentId: z .union([z.number(), z.object({ name: z.string(), pluginId: z.string() })]) .nullable(), x: z.number(), y: z.number(), z: z.number(), }) .optional(), permanent: z.boolean().optional(), active: z.boolean().optional(), tags: z.array(z.string()).optional(), }), ) .meta({ action: () => { return { position: { name: "Position", type: "starmapCoordinates", helper: "A specific point in space to place the waypoint. Use as an alternative to Waypoint Entity.", }, entityId: { name: "Waypoint Entity", helper: "The entity to attach the waypoint to. This option is preferred.", }, }; }, event: true, }) .output( z.object({ waypointId: z.number(), shipId: z.number(), }), ) .send(({ ctx, input }) => { if (!ctx.flight) throw new Error("No flight in progress"); const ship = ctx.flight.ecs.getEntityById(input.shipId); if (!ship) throw new Error("No ship selected."); const shipId = ship.id; let position = { x: 0, y: 0, z: 0 }; let systemId: number | null = null; let object: Entity | null = null; if ("entityId" in input) { // This waypoint is being attached to a specific object in space. object = ctx.ecs.getEntityById(input.entityId || -1); if (!object) throw new Error("No object found.");
const targetParentId = ship.components.position ? ship.components.position.parentId : ship.components.satellite?.parentId || null; const targetPosition = { parentId: targetParentId, type: targetParentId ? "solar" : "interstellar", ...(ship.components.position || getCompletePositionFromOrbit(ship)), } as const; position = getObjectOffsetPosition( object, targetPosition, (ship.components.size?.length || 1 / 1000) * 2, ); const sys = getObjectSystem(object); systemId = sys?.id ?? null; if (sys?.id === object.id) systemId = null; for (const maybeWaypoint of ctx.ecs.componentCache.get("isWaypoint") || []) { if ( maybeWaypoint.components.isWaypoint?.assignedShipId === shipId && maybeWaypoint.components.isWaypoint?.attachedObjectId === object?.id ) { maybeWaypoint.updateComponent("position", { ...position, type: systemId ? "solar" : "interstellar", parentId: systemId, }); maybeWaypoint.updateComponent("isWaypoint", { lastInteractedAt: Date.now(), }); if (input.tags && input.tags.length > 0) { maybeWaypoint.updateComponent("tags", { tags: [...(maybeWaypoint.components.tags?.tags || []), ...input.tags], }); } pubsub.publish.waypoints.all({ shipId, }); return { waypointId: maybeWaypoint.id, shipId }; } } } else if ("position" in input && input.position) { // This waypoint is just being plopped at some random point in space. position = input.position; const parentId = input.position.parentId; if (parentId && typeof parentId === "object") { // This waypoint is probably defined in a timeline action, so we need // to find which system matches the name. const solarSystems = ctx.flight.ecs.componentCache.get("isSolarSystem") || []; for (const entity of solarSystems) { if (entity.components.identity?.name === parentId.name) { systemId = entity.id; break; } } } else { systemId = parentId; } } else { throw new Error("Either position or entityId are required"); }
const newWaypoint = new Entity(); newWaypoint.addComponent("isWaypoint", { assignedShipId: shipId, attachedObjectId: object?.id, permanent: input.permanent, isActive: input.active, lastInteractedAt: Date.now(), }); // If we have an object, set the name to the name of that object if (object?.components.identity?.name) { // TODO: INTL in the server here. newWaypoint.addComponent("identity", { name: `${object.components.identity.name} Waypoint`, }); } else { // Count up the highest waypoint count and use that. let waypointNum = 1; for (const waypoint of ctx.ecs.componentCache.get("isWaypoint") || []) { if (waypoint.components.isWaypoint?.assignedShipId === shipId) { const nameWords = waypoint.components.identity?.name.split(" ") || []; const num = Number.parseInt(nameWords[nameWords.length - 1], 10); if (num && num >= waypointNum) waypointNum = num + 1; } } newWaypoint.addComponent("identity", { name: `Waypoint ${waypointNum}`, }); } newWaypoint.addComponent("position", { ...position, parentId: systemId, type: systemId ? "solar" : "interstellar", });
if (input.tags && input.tags.length > 0) { newWaypoint.addComponent("tags", { tags: input.tags, }); }
ctx.flight.ecs.addEntity(newWaypoint);
pubsub.publish.waypoints.all({ shipId, });
return { waypointId: newWaypoint.id, shipId }; }), activate: t.procedure .meta({ action: true, event: true }) .input(z.object({ waypointId: z.number() })) .output(z.object({ shipId: z.number(), waypointId: z.number() })) .send(({ ctx, input }) => { const waypoint = ctx.ecs.getEntityById(input.waypointId); if (!waypoint?.components.isWaypoint) throw new Error("Waypoint not found."); waypoint.updateComponent("isWaypoint", { isActive: true }); pubsub.publish.waypoints.all({ shipId: waypoint.components.isWaypoint.assignedShipId, }); return { waypointId: input.waypointId, shipId: waypoint.components.isWaypoint.assignedShipId, }; }), deactivate: t.procedure .meta({ action: true, event: true }) .input(z.object({ waypointId: z.number() })) .output(z.object({ shipId: z.number(), waypointId: z.number() })) .send(({ ctx, input }) => { const waypoint = ctx.ecs.getEntityById(input.waypointId); if (!waypoint?.components.isWaypoint) throw new Error("Waypoint not found."); const shipId = waypoint.components.isWaypoint.assignedShipId; if (isWaypointLocked(ctx.ecs, input.waypointId, shipId)) { return { waypointId: input.waypointId, shipId }; } waypoint.updateComponent("isWaypoint", { isActive: false }); pubsub.publish.waypoints.all({ shipId: waypoint.components.isWaypoint.assignedShipId, }); return { waypointId: input.waypointId, shipId: waypoint.components.isWaypoint.assignedShipId, }; }), delete: t.procedure .meta({ action: true, event: true }) .input( z.object({ shipId: z.number(), waypointId: z.number(), overridePermanent: z.boolean().optional(), }), ) .send(({ ctx, input: { waypointId, shipId, overridePermanent } }) => { const waypoint = ctx.ecs.getEntityById(waypointId); if (!waypoint) throw new Error("No waypoint found."); if (waypoint.components.isWaypoint?.assignedShipId !== shipId) throw new Error("Waypoint is not assigned to this ship."); if (isWaypointLocked(ctx.ecs, waypointId, shipId)) { throw new Error("Cannot delete a locked waypoint."); } if (waypoint.components.isWaypoint.permanent && !overridePermanent) { throw new Error("Waypoint cannot be deleted."); } ctx.ecs.removeEntity(waypoint); pubsub.publish.waypoints.all({ shipId, }); return null; }), deleteAll: t.procedure .meta({ action: true, event: true }) .input(z.object({ shipId: z.number() })) .output(z.null()) .send(({ ctx, input: { shipId } }) => { const toRemove: Entity[] = []; for (const waypoint of ctx.ecs.componentCache.get("isWaypoint") || []) { if (waypoint.components.isWaypoint?.assignedShipId === shipId) { if (isWaypointLocked(ctx.ecs, waypoint.id, shipId)) continue; toRemove.push(waypoint); } } for (const waypoint of toRemove) { ctx.ecs.removeEntity(waypoint); } pubsub.publish.waypoints.all({ shipId }); return null; }), deactivateAll: t.procedure .meta({ action: true, event: true }) .input(z.object({ shipId: z.number() })) .output(z.null()) .send(({ ctx, input: { shipId } }) => { deactivateWaypoints(ctx.ecs, shipId); return null; }),});
function deactivateWaypoints(ecs: ECS, shipId: number) { for (const waypoint of ecs.componentCache.get("isWaypoint") || []) { if (waypoint.components.isWaypoint?.assignedShipId === shipId) { if (isWaypointLocked(ecs, waypoint.id, shipId)) continue; waypoint.updateComponent("isWaypoint", { isActive: false, }); } } pubsub.publish.waypoints.all({ shipId, });}