mirror/fosrl.pangolin
1
0
Fork 0
mirror of https://github.com/fosrl/pangolin.git synced 2025-07-10 14:04:51 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Add new schema

Browse source
This commit is contained in:
Owen Schwartz 2024-09-28 13:31:22 -04:00
parent f8ed090a83
commit 06ee87ac2c
No known key found for this signature in database
GPG key ID: 8271FDFFD9E0CCBD
7 changed files with 202 additions and 106 deletions
Download patch file Download diff file Expand all files Collapse all files

2
package.json
View file

@ -5,7 +5,7 @@
"scripts": { "scripts": {
"dev": "dotenvx run -- tsx watch server/index.ts", "dev": "dotenvx run -- tsx watch server/index.ts",
"db:generate": "drizzle-kit generate", "db:generate": "drizzle-kit generate",
"db:push": "drizzle-kit push", "db:push": "npx tsx scripts/runMigrations.ts",
"db:studio": "drizzle-kit studio" "db:studio": "drizzle-kit studio"
}, },
"dependencies": { "dependencies": {

27
scripts/runMigrations.ts Normal file
View file

@ -0,0 +1,27 @@
import { drizzle } from 'drizzle-orm/better-sqlite3';
import { migrate } from 'drizzle-orm/better-sqlite3/migrator';
import Database from 'better-sqlite3';
const runMigrations = async () => {
// Create a new SQLite database connection
const sqlite = new Database('./config/db/db.sqlite');
// Create a Drizzle instance
const db = drizzle(sqlite);
console.log('Running migrations...');
try {
// Run the migrations
await migrate(db, { migrationsFolder: './server/migrations' });
console.log('Migrations completed successfully.');
} catch (error) {
console.error('Error running migrations:', error);
process.exit(1);
} finally {
// Close the database connection
sqlite.close();
}
};
runMigrations();

49
server/db/schema.ts
View file

@ -1,61 +1,70 @@
import { sqliteTable, text, integer } from "drizzle-orm/sqlite-core"; import { sqliteTable, text, integer } from "drizzle-orm/sqlite-core";
import { InferSelectModel } from "drizzle-orm"; import { InferSelectModel } from "drizzle-orm";
// Org table // Orgs table
export const org = sqliteTable("org", { export const orgs = sqliteTable("orgs", {
orgId: integer("orgId").primaryKey({ autoIncrement: true }), orgId: integer("orgId").primaryKey({ autoIncrement: true }),
name: text("name").notNull(), name: text("name").notNull(),
domain: text("domain").notNull(),
}); });
// Users table // Users table
export const users = sqliteTable("users", { export const users = sqliteTable("users", {
uid: integer("uid").primaryKey({ autoIncrement: true }), userId: integer("userId").primaryKey({ autoIncrement: true }),
orgId: integer("orgId").references(() => org.orgId), orgId: integer("orgId").references(() => orgs.orgId),
name: text("name"), name: text("name").notNull(),
email: text("email"), email: text("email").notNull(),
groups: text("groups"), groups: text("groups"),
}); });
// Sites table // Sites table
export const sites = sqliteTable("sites", { export const sites = sqliteTable("sites", {
siteId: integer("siteId").primaryKey({ autoIncrement: true }), siteId: integer("siteId").primaryKey({ autoIncrement: true }),
orgId: integer("orgId").references(() => org.orgId), orgId: integer("orgId").references(() => orgs.orgId),
name: text("name"),
autoSubdomain: text("autoSubdomain"),
pubKey: integer("pubKey"),
subnet: text("subnet"),
exitNode: integer("exitNode").references(() => exitNodes.exitNodeId), exitNode: integer("exitNode").references(() => exitNodes.exitNodeId),
name: text("name").notNull(),
subdomain: text("subdomain"),
pubKey: text("pubKey"),
subnet: text("subnet"),
}); });
// Resources table // Resources table
export const resources = sqliteTable("resources", { export const resources = sqliteTable("resources", {
resourceId: integer("resourceId").primaryKey({ autoIncrement: true }), resourceId: integer("resourceId").primaryKey({ autoIncrement: true }),
siteId: integer("siteId").references(() => sites.siteId), siteId: integer("siteId").references(() => sites.siteId),
name: text("name"), name: text("name").notNull(),
targetIp: text("targetIp"), subdomain: text("subdomain"),
method: text("method"),
port: integer("port"),
proto: text("proto"),
}); });
// Exit Nodes table // Exit Nodes table
export const exitNodes = sqliteTable("exitNodes", { export const exitNodes = sqliteTable("exitNodes", {
exitNodeId: integer("exitNodeId").primaryKey({ autoIncrement: true }), exitNodeId: integer("exitNodeId").primaryKey({ autoIncrement: true }),
name: text("name"), name: text("name").notNull(),
address: integer("address"), address: text("address").notNull(),
}); });
// Routes table // Routes table
export const routes = sqliteTable("routes", { export const routes = sqliteTable("routes", {
routeId: integer("routeId").primaryKey({ autoIncrement: true }), routeId: integer("routeId").primaryKey({ autoIncrement: true }),
subnet: integer("subnet"),
exitNodeId: integer("exitNodeId").references(() => exitNodes.exitNodeId), exitNodeId: integer("exitNodeId").references(() => exitNodes.exitNodeId),
subnet: text("subnet").notNull(),
});
// Targets table
export const targets = sqliteTable("targets", {
targetId: integer("targetId").primaryKey({ autoIncrement: true }),
resourceId: integer("resourceId").references(() => resources.resourceId),
ip: text("ip").notNull(),
method: text("method"),
port: integer("port"),
protocol: text("protocol"),
}); });
// Define the model types for type inference // Define the model types for type inference
export type Org = InferSelectModel<typeof org>; export type Org = InferSelectModel<typeof orgs>;
export type User = InferSelectModel<typeof users>; export type User = InferSelectModel<typeof users>;
export type Site = InferSelectModel<typeof sites>; export type Site = InferSelectModel<typeof sites>;
export type Resource = InferSelectModel<typeof resources>; export type Resource = InferSelectModel<typeof resources>;
export type ExitNode = InferSelectModel<typeof exitNodes>; export type ExitNode = InferSelectModel<typeof exitNodes>;
export type Route = InferSelectModel<typeof routes>; export type Route = InferSelectModel<typeof routes>;
export type Target = InferSelectModel<typeof targets>;

3
server/index.ts
View file

@ -19,7 +19,8 @@ let db: Database.Database;
app.prepare().then(() => { app.prepare().then(() => {
// Open the SQLite database connection // Open the SQLite database connection
const sqlite = new Database(`${environment.CONFIG_PATH}/db/db.sqlite`, { verbose: console.log }); // const sqlite = new Database(`${environment.CONFIG_PATH}/db/db.sqlite`, { verbose: console.log });
const sqlite = new Database(`${environment.CONFIG_PATH}/db/db.sqlite`);
const db = drizzle(sqlite); const db = drizzle(sqlite);
// Main server // Main server

52
server/migrations/0000_lowly_robin_chapel.sql → server/migrations/0000_short_ulik.sql
View file

@ -1,49 +1,57 @@
CREATE TABLE `exitNodes` ( CREATE TABLE `exitNodes` (
`exitNodeId` integer PRIMARY KEY AUTOINCREMENT NOT NULL, `exitNodeId` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
`name` text, `name` text NOT NULL,
`address` integer `address` text NOT NULL
); );
--> statement-breakpoint --> statement-breakpoint
CREATE TABLE `org` ( CREATE TABLE `orgs` (
`orgId` integer PRIMARY KEY AUTOINCREMENT NOT NULL, `orgId` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
`name` text NOT NULL `name` text NOT NULL,
`domain` text NOT NULL
); );
--> statement-breakpoint --> statement-breakpoint
CREATE TABLE `resources` ( CREATE TABLE `resources` (
`resourceId` integer PRIMARY KEY AUTOINCREMENT NOT NULL, `resourceId` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
`siteId` integer, `siteId` integer,
`name` text, `name` text NOT NULL,
`targetIp` text, `subdomain` text,
`method` text,
`port` integer,
`proto` text,
FOREIGN KEY (`siteId`) REFERENCES `sites`(`siteId`) ON UPDATE no action ON DELETE no action FOREIGN KEY (`siteId`) REFERENCES `sites`(`siteId`) ON UPDATE no action ON DELETE no action
); );
--> statement-breakpoint --> statement-breakpoint
CREATE TABLE `routes` ( CREATE TABLE `routes` (
`routeId` integer PRIMARY KEY AUTOINCREMENT NOT NULL, `routeId` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
`subnet` integer,
`exitNodeId` integer, `exitNodeId` integer,
`subnet` text NOT NULL,
FOREIGN KEY (`exitNodeId`) REFERENCES `exitNodes`(`exitNodeId`) ON UPDATE no action ON DELETE no action FOREIGN KEY (`exitNodeId`) REFERENCES `exitNodes`(`exitNodeId`) ON UPDATE no action ON DELETE no action
); );
--> statement-breakpoint --> statement-breakpoint
CREATE TABLE `sites` ( CREATE TABLE `sites` (
`siteId` integer PRIMARY KEY AUTOINCREMENT NOT NULL, `siteId` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
`orgId` integer, `orgId` integer,
`name` text,
`autoSubdomain` text,
`pubKey` integer,
`subnet` text,
`exitNode` integer, `exitNode` integer,
FOREIGN KEY (`orgId`) REFERENCES `org`(`orgId`) ON UPDATE no action ON DELETE no action, `name` text NOT NULL,
`subdomain` text,
`pubKey` text,
`subnet` text,
FOREIGN KEY (`orgId`) REFERENCES `orgs`(`orgId`) ON UPDATE no action ON DELETE no action,
FOREIGN KEY (`exitNode`) REFERENCES `exitNodes`(`exitNodeId`) ON UPDATE no action ON DELETE no action FOREIGN KEY (`exitNode`) REFERENCES `exitNodes`(`exitNodeId`) ON UPDATE no action ON DELETE no action
); );
--> statement-breakpoint --> statement-breakpoint
CREATE TABLE `users` ( CREATE TABLE `targets` (
`uid` integer PRIMARY KEY AUTOINCREMENT NOT NULL, `targetId` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
`orgId` integer, `resourceId` integer,
`name` text, `ip` text NOT NULL,
`email` text, `method` text,
`groups` text, `port` integer,
FOREIGN KEY (`orgId`) REFERENCES `org`(`orgId`) ON UPDATE no action ON DELETE no action `protocol` text,
FOREIGN KEY (`resourceId`) REFERENCES `resources`(`resourceId`) ON UPDATE no action ON DELETE no action
);
--> statement-breakpoint
CREATE TABLE `users` (
`userId` integer PRIMARY KEY AUTOINCREMENT NOT NULL,
`orgId` integer,
`name` text NOT NULL,
`email` text NOT NULL,
`groups` text,
FOREIGN KEY (`orgId`) REFERENCES `orgs`(`orgId`) ON UPDATE no action ON DELETE no action
); );

171
server/migrations/meta/0000_snapshot.json
View file

@ -1,7 +1,7 @@
{ {
"version": "6", "version": "6",
"dialect": "sqlite", "dialect": "sqlite",
"id": "a06283ff-bd4b-4994-988c-4db3279f73dd", "id": "9b039f4c-6867-4b08-8aa9-bc184c37b910",
"prevId": "00000000-0000-0000-0000-000000000000", "prevId": "00000000-0000-0000-0000-000000000000",
"tables": { "tables": {
"exitNodes": { "exitNodes": {
@ -18,14 +18,14 @@
"name": "name", "name": "name",
"type": "text", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": true,
"autoincrement": false "autoincrement": false
}, },
"address": { "address": {
"name": "address", "name": "address",
"type": "integer", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": true,
"autoincrement": false "autoincrement": false
} }
}, },
@ -34,8 +34,8 @@
"compositePrimaryKeys": {}, "compositePrimaryKeys": {},
"uniqueConstraints": {} "uniqueConstraints": {}
}, },
"org": { "orgs": {
"name": "org", "name": "orgs",
"columns": { "columns": {
"orgId": { "orgId": {
"name": "orgId", "name": "orgId",
@ -50,6 +50,13 @@
"primaryKey": false, "primaryKey": false,
"notNull": true, "notNull": true,
"autoincrement": false "autoincrement": false
},
"domain": {
"name": "domain",
"type": "text",
"primaryKey": false,
"notNull": true,
"autoincrement": false
} }
}, },
"indexes": {}, "indexes": {},
@ -78,32 +85,11 @@
"name": "name", "name": "name",
"type": "text", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": true,
"autoincrement": false "autoincrement": false
}, },
"targetIp": { "subdomain": {
"name": "targetIp", "name": "subdomain",
"type": "text",
"primaryKey": false,
"notNull": false,
"autoincrement": false
},
"method": {
"name": "method",
"type": "text",
"primaryKey": false,
"notNull": false,
"autoincrement": false
},
"port": {
"name": "port",
"type": "integer",
"primaryKey": false,
"notNull": false,
"autoincrement": false
},
"proto": {
"name": "proto",
"type": "text", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": false,
@ -139,19 +125,19 @@
"notNull": true, "notNull": true,
"autoincrement": true "autoincrement": true
}, },
"subnet": {
"name": "subnet",
"type": "integer",
"primaryKey": false,
"notNull": false,
"autoincrement": false
},
"exitNodeId": { "exitNodeId": {
"name": "exitNodeId", "name": "exitNodeId",
"type": "integer", "type": "integer",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": false,
"autoincrement": false "autoincrement": false
},
"subnet": {
"name": "subnet",
"type": "text",
"primaryKey": false,
"notNull": true,
"autoincrement": false
} }
}, },
"indexes": {}, "indexes": {},
@ -190,15 +176,22 @@
"notNull": false, "notNull": false,
"autoincrement": false "autoincrement": false
}, },
"name": { "exitNode": {
"name": "name", "name": "exitNode",
"type": "text", "type": "integer",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": false,
"autoincrement": false "autoincrement": false
}, },
"autoSubdomain": { "name": {
"name": "autoSubdomain", "name": "name",
"type": "text",
"primaryKey": false,
"notNull": true,
"autoincrement": false
},
"subdomain": {
"name": "subdomain",
"type": "text", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": false,
@ -206,7 +199,7 @@
}, },
"pubKey": { "pubKey": {
"name": "pubKey", "name": "pubKey",
"type": "integer", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": false,
"autoincrement": false "autoincrement": false
@ -217,21 +210,14 @@
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": false,
"autoincrement": false "autoincrement": false
},
"exitNode": {
"name": "exitNode",
"type": "integer",
"primaryKey": false,
"notNull": false,
"autoincrement": false
} }
}, },
"indexes": {}, "indexes": {},
"foreignKeys": { "foreignKeys": {
"sites_orgId_org_orgId_fk": { "sites_orgId_orgs_orgId_fk": {
"name": "sites_orgId_org_orgId_fk", "name": "sites_orgId_orgs_orgId_fk",
"tableFrom": "sites", "tableFrom": "sites",
"tableTo": "org", "tableTo": "orgs",
"columnsFrom": [ "columnsFrom": [
"orgId" "orgId"
], ],
@ -258,11 +244,76 @@
"compositePrimaryKeys": {}, "compositePrimaryKeys": {},
"uniqueConstraints": {} "uniqueConstraints": {}
}, },
"targets": {
"name": "targets",
"columns": {
"targetId": {
"name": "targetId",
"type": "integer",
"primaryKey": true,
"notNull": true,
"autoincrement": true
},
"resourceId": {
"name": "resourceId",
"type": "integer",
"primaryKey": false,
"notNull": false,
"autoincrement": false
},
"ip": {
"name": "ip",
"type": "text",
"primaryKey": false,
"notNull": true,
"autoincrement": false
},
"method": {
"name": "method",
"type": "text",
"primaryKey": false,
"notNull": false,
"autoincrement": false
},
"port": {
"name": "port",
"type": "integer",
"primaryKey": false,
"notNull": false,
"autoincrement": false
},
"protocol": {
"name": "protocol",
"type": "text",
"primaryKey": false,
"notNull": false,
"autoincrement": false
}
},
"indexes": {},
"foreignKeys": {
"targets_resourceId_resources_resourceId_fk": {
"name": "targets_resourceId_resources_resourceId_fk",
"tableFrom": "targets",
"tableTo": "resources",
"columnsFrom": [
"resourceId"
],
"columnsTo": [
"resourceId"
],
"onDelete": "no action",
"onUpdate": "no action"
}
},
"compositePrimaryKeys": {},
"uniqueConstraints": {}
},
"users": { "users": {
"name": "users", "name": "users",
"columns": { "columns": {
"uid": { "userId": {
"name": "uid", "name": "userId",
"type": "integer", "type": "integer",
"primaryKey": true, "primaryKey": true,
"notNull": true, "notNull": true,
@ -279,14 +330,14 @@
"name": "name", "name": "name",
"type": "text", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": true,
"autoincrement": false "autoincrement": false
}, },
"email": { "email": {
"name": "email", "name": "email",
"type": "text", "type": "text",
"primaryKey": false, "primaryKey": false,
"notNull": false, "notNull": true,
"autoincrement": false "autoincrement": false
}, },
"groups": { "groups": {
@ -299,10 +350,10 @@
}, },
"indexes": {}, "indexes": {},
"foreignKeys": { "foreignKeys": {
"users_orgId_org_orgId_fk": { "users_orgId_orgs_orgId_fk": {
"name": "users_orgId_org_orgId_fk", "name": "users_orgId_orgs_orgId_fk",
"tableFrom": "users", "tableFrom": "users",
"tableTo": "org", "tableTo": "orgs",
"columnsFrom": [ "columnsFrom": [
"orgId" "orgId"
], ],

4
server/migrations/meta/_journal.json
View file

@ -5,8 +5,8 @@
{ {
"idx": 0, "idx": 0,
"version": "6", "version": "6",
"when": 1727538753952, "when": 1727544220529,
"tag": "0000_lowly_robin_chapel", "tag": "0000_short_ulik",
"breakpoints": true "breakpoints": true
} }
] ]