mirror of
https://github.com/fosrl/pangolin.git
synced 2025-08-30 14:39:29 +02:00
add sqlite 1.9.0 migration
This commit is contained in:
miloschwartz
parent
d207318494
commit
ac8b546393
2 changed files with 159 additions and 10 deletions
Binary file not shown.
|
|
@ -10,26 +10,175 @@ export default async function migration() {
|
||||||
const location = path.join(APP_PATH, "db", "db.sqlite");
|
const location = path.join(APP_PATH, "db", "db.sqlite");
|
||||||
const db = new Database(location);
|
const db = new Database(location);
|
||||||
|
|
||||||
|
const resourceSiteMap = new Map<number, number>();
|
||||||
|
|
||||||
|
try {
|
||||||
|
const resources = db
|
||||||
|
.prepare(
|
||||||
|
"SELECT resourceId, siteId FROM resources WHERE siteId IS NOT NULL"
|
||||||
|
)
|
||||||
|
.all() as Array<{ resourceId: number; siteId: number }>;
|
||||||
|
for (const resource of resources) {
|
||||||
|
resourceSiteMap.set(resource.resourceId, resource.siteId);
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
console.log("Error getting resources:", e);
|
||||||
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
db.pragma("foreign_keys = OFF");
|
db.pragma("foreign_keys = OFF");
|
||||||
|
|
||||||
db.transaction(() => {
|
db.transaction(() => {
|
||||||
db.exec(`
|
db.exec(`CREATE TABLE 'setupTokens' (
|
||||||
CREATE TABLE 'setupTokens' (
|
|
||||||
'tokenId' text PRIMARY KEY NOT NULL,
|
'tokenId' text PRIMARY KEY NOT NULL,
|
||||||
'token' text NOT NULL,
|
'token' text NOT NULL,
|
||||||
'used' integer DEFAULT 0 NOT NULL,
|
'used' integer DEFAULT false NOT NULL,
|
||||||
'dateCreated' text NOT NULL,
|
'dateCreated' text NOT NULL,
|
||||||
'dateUsed' text
|
'dateUsed' text
|
||||||
);
|
);
|
||||||
|
--> statement-breakpoint
|
||||||
|
CREATE TABLE 'siteResources' (
|
||||||
|
'siteResourceId' integer PRIMARY KEY AUTOINCREMENT NOT NULL,
|
||||||
|
'siteId' integer NOT NULL,
|
||||||
|
'orgId' text NOT NULL,
|
||||||
|
'name' text NOT NULL,
|
||||||
|
'protocol' text NOT NULL,
|
||||||
|
'proxyPort' integer NOT NULL,
|
||||||
|
'destinationPort' integer NOT NULL,
|
||||||
|
'destinationIp' text NOT NULL,
|
||||||
|
'enabled' integer DEFAULT true NOT NULL,
|
||||||
|
FOREIGN KEY ('siteId') REFERENCES 'sites'('siteId') ON UPDATE no action ON DELETE cascade,
|
||||||
|
FOREIGN KEY ('orgId') REFERENCES 'orgs'('orgId') ON UPDATE no action ON DELETE cascade
|
||||||
|
);
|
||||||
|
--> statement-breakpoint
|
||||||
|
PRAGMA foreign_keys=OFF;--> statement-breakpoint
|
||||||
|
CREATE TABLE '__new_resources' (
|
||||||
|
'resourceId' integer PRIMARY KEY AUTOINCREMENT NOT NULL,
|
||||||
|
'orgId' text NOT NULL,
|
||||||
|
'name' text NOT NULL,
|
||||||
|
'subdomain' text,
|
||||||
|
'fullDomain' text,
|
||||||
|
'domainId' text,
|
||||||
|
'ssl' integer DEFAULT false NOT NULL,
|
||||||
|
'blockAccess' integer DEFAULT false NOT NULL,
|
||||||
|
'sso' integer DEFAULT true NOT NULL,
|
||||||
|
'http' integer DEFAULT true NOT NULL,
|
||||||
|
'protocol' text NOT NULL,
|
||||||
|
'proxyPort' integer,
|
||||||
|
'emailWhitelistEnabled' integer DEFAULT false NOT NULL,
|
||||||
|
'applyRules' integer DEFAULT false NOT NULL,
|
||||||
|
'enabled' integer DEFAULT true NOT NULL,
|
||||||
|
'stickySession' integer DEFAULT false NOT NULL,
|
||||||
|
'tlsServerName' text,
|
||||||
|
'setHostHeader' text,
|
||||||
|
'enableProxy' integer DEFAULT true,
|
||||||
|
'skipToIdpId' integer,
|
||||||
|
FOREIGN KEY ('orgId') REFERENCES 'orgs'('orgId') ON UPDATE no action ON DELETE cascade,
|
||||||
|
FOREIGN KEY ('domainId') REFERENCES 'domains'('domainId') ON UPDATE no action ON DELETE set null,
|
||||||
|
FOREIGN KEY ('skipToIdpId') REFERENCES 'idp'('idpId') ON UPDATE no action ON DELETE cascade
|
||||||
|
);
|
||||||
|
--> statement-breakpoint
|
||||||
|
INSERT INTO '__new_resources'("resourceId", "orgId", "name", "subdomain", "fullDomain", "domainId", "ssl", "blockAccess", "sso", "http", "protocol", "proxyPort", "emailWhitelistEnabled", "applyRules", "enabled", "stickySession", "tlsServerName", "setHostHeader", "enableProxy", "skipToIdpId") SELECT "resourceId", "orgId", "name", "subdomain", "fullDomain", "domainId", "ssl", "blockAccess", "sso", "http", "protocol", "proxyPort", "emailWhitelistEnabled", "applyRules", "enabled", "stickySession", "tlsServerName", "setHostHeader", "enableProxy", null FROM 'resources';--> statement-breakpoint
|
||||||
|
DROP TABLE 'resources';--> statement-breakpoint
|
||||||
|
ALTER TABLE '__new_resources' RENAME TO 'resources';--> statement-breakpoint
|
||||||
|
PRAGMA foreign_keys=ON;--> statement-breakpoint
|
||||||
|
CREATE TABLE '__new_clients' (
|
||||||
|
'id' integer PRIMARY KEY AUTOINCREMENT NOT NULL,
|
||||||
|
'orgId' text NOT NULL,
|
||||||
|
'exitNode' integer,
|
||||||
|
'name' text NOT NULL,
|
||||||
|
'pubKey' text,
|
||||||
|
'subnet' text NOT NULL,
|
||||||
|
'bytesIn' integer,
|
||||||
|
'bytesOut' integer,
|
||||||
|
'lastBandwidthUpdate' text,
|
||||||
|
'lastPing' integer,
|
||||||
|
'type' text NOT NULL,
|
||||||
|
'online' integer DEFAULT false NOT NULL,
|
||||||
|
'lastHolePunch' integer,
|
||||||
|
FOREIGN KEY ('orgId') REFERENCES 'orgs'('orgId') ON UPDATE no action ON DELETE cascade,
|
||||||
|
FOREIGN KEY ('exitNode') REFERENCES 'exitNodes'('exitNodeId') ON UPDATE no action ON DELETE set null
|
||||||
|
);
|
||||||
|
--> statement-breakpoint
|
||||||
|
INSERT INTO '__new_clients'("id", "orgId", "exitNode", "name", "pubKey", "subnet", "bytesIn", "bytesOut", "lastBandwidthUpdate", "lastPing", "type", "online", "lastHolePunch") SELECT "id", "orgId", "exitNode", "name", "pubKey", "subnet", "bytesIn", "bytesOut", "lastBandwidthUpdate", "lastPing", "type", "online", "lastHolePunch" FROM 'clients';--> statement-breakpoint
|
||||||
|
DROP TABLE 'clients';--> statement-breakpoint
|
||||||
|
ALTER TABLE '__new_clients' RENAME TO 'clients';--> statement-breakpoint
|
||||||
|
ALTER TABLE 'clientSites' ADD 'endpoint' text;--> statement-breakpoint
|
||||||
|
ALTER TABLE 'exitNodes' ADD 'online' integer DEFAULT false NOT NULL;--> statement-breakpoint
|
||||||
|
ALTER TABLE 'exitNodes' ADD 'lastPing' integer;--> statement-breakpoint
|
||||||
|
ALTER TABLE 'exitNodes' ADD 'type' text DEFAULT 'gerbil';--> statement-breakpoint
|
||||||
|
ALTER TABLE 'olms' ADD 'version' text;--> statement-breakpoint
|
||||||
|
ALTER TABLE 'orgs' ADD 'createdAt' text;--> statement-breakpoint
|
||||||
|
ALTER TABLE 'targets' ADD 'siteId' integer NOT NULL DEFAULT 1 REFERENCES sites(siteId);`);
|
||||||
|
|
||||||
|
// for each resource, get all of its targets, and update the siteId to be the previously stored siteId
|
||||||
|
for (const [resourceId, siteId] of resourceSiteMap) {
|
||||||
|
const targets = db
|
||||||
|
.prepare(
|
||||||
|
"SELECT targetId FROM targets WHERE resourceId = ?"
|
||||||
|
)
|
||||||
|
.all(resourceId) as Array<{ targetId: number }>;
|
||||||
|
for (const target of targets) {
|
||||||
|
db.prepare(
|
||||||
|
"UPDATE targets SET siteId = ? WHERE targetId = ?"
|
||||||
|
).run(siteId, target.targetId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// list resources that have enableProxy false
|
||||||
|
// move them to the siteResources table
|
||||||
|
// remove them from the resources table
|
||||||
|
const proxyFalseResources = db
|
||||||
|
.prepare("SELECT * FROM resources WHERE enableProxy = 0")
|
||||||
|
.all() as Array<any>;
|
||||||
|
|
||||||
|
for (const resource of proxyFalseResources) {
|
||||||
|
// Get the first target to derive destination IP and port
|
||||||
|
const firstTarget = db
|
||||||
|
.prepare(
|
||||||
|
"SELECT ip, port FROM targets WHERE resourceId = ? LIMIT 1"
|
||||||
|
)
|
||||||
|
.get(resource.resourceId) as
|
||||||
|
| { ip: string; port: number }
|
||||||
|
| undefined;
|
||||||
|
|
||||||
|
if (!firstTarget) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Insert into siteResources table
|
||||||
|
const stmt = db.prepare(`
|
||||||
|
INSERT INTO siteResources (siteId, orgId, name, protocol, proxyPort, destinationPort, destinationIp, enabled)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?)
|
||||||
`);
|
`);
|
||||||
|
stmt.run(
|
||||||
|
resourceSiteMap.get(resource.resourceId),
|
||||||
|
resource.orgId,
|
||||||
|
resource.name,
|
||||||
|
resource.protocol,
|
||||||
|
resource.proxyPort,
|
||||||
|
firstTarget.port,
|
||||||
|
firstTarget.ip,
|
||||||
|
resource.enabled
|
||||||
|
);
|
||||||
|
|
||||||
|
// Delete from resources table
|
||||||
|
db.prepare("DELETE FROM resources WHERE resourceId = ?").run(
|
||||||
|
resource.resourceId
|
||||||
|
);
|
||||||
|
|
||||||
|
// Delete the targets for this resource
|
||||||
|
db.prepare("DELETE FROM targets WHERE resourceId = ?").run(
|
||||||
|
resource.resourceId
|
||||||
|
);
|
||||||
|
}
|
||||||
})();
|
})();
|
||||||
|
|
||||||
db.pragma("foreign_keys = ON");
|
db.pragma("foreign_keys = ON");
|
||||||
|
|
||||||
console.log(`Added setupTokens table`);
|
console.log(`Migrated database`);
|
||||||
} catch (e) {
|
} catch (e) {
|
||||||
console.log("Unable to add setupTokens table:", e);
|
console.log("Failed to migrate db:", e);
|
||||||
throw e;
|
throw e;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue