mirror of
https://github.com/Hessenuk/DiscordTickets.git
synced 2025-02-23 18:51:29 +02:00
03f3944d31
* wip: data import/export scripts * feat: data exporter * feat: importer * fix: that was stupid * fix: importer * style: clean up comments
113 lines
3.3 KiB
JavaScript
113 lines
3.3 KiB
JavaScript
import { config } from 'dotenv';
|
|
import { program } from 'commander';
|
|
import fse from 'fs-extra';
|
|
import { join } from 'path';
|
|
import ora from 'ora';
|
|
import { PrismaClient } from '@prisma/client';
|
|
import { createHash } from 'crypto';
|
|
import Cryptr from 'cryptr';
|
|
|
|
config();
|
|
|
|
program
|
|
.requiredOption('-g, --guild <id>', 'the ID of the guild to export');
|
|
|
|
program.parse();
|
|
|
|
const options = program.opts();
|
|
|
|
const hash = createHash('sha256').update(options.guild).digest('hex');
|
|
const file_path = join(process.cwd(), './user/dumps', `${hash}.dump`);
|
|
const file_cryptr = new Cryptr(options.guild);
|
|
const db_cryptr = new Cryptr(process.env.ENCRYPTION_KEY);
|
|
|
|
fse.ensureDirSync(join(process.cwd(), './user/dumps'));
|
|
|
|
let spinner = ora('Connecting').start();
|
|
|
|
const prisma_options = {};
|
|
|
|
if (process.env.DB_PROVIDER === 'sqlite' && !process.env.DB_CONNECTION_URL) {
|
|
prisma_options.datasources = { db: { url: 'file:' + join(process.cwd(), './user/database.db') } };
|
|
}
|
|
|
|
const prisma = new PrismaClient(prisma_options);
|
|
|
|
if (process.env.DB_PROVIDER === 'sqlite') {
|
|
const { default: sqliteMiddleware } = await import('../src/lib/middleware/prisma-sqlite.js');
|
|
prisma.$use(sqliteMiddleware);
|
|
await prisma.$queryRaw`PRAGMA journal_mode=WAL;`;
|
|
await prisma.$queryRaw`PRAGMA synchronous=normal;`;
|
|
}
|
|
|
|
spinner.succeed('Connected');
|
|
|
|
|
|
const dump = {};
|
|
|
|
spinner = ora('Exporting settings').start();
|
|
dump.settings = await prisma.guild.findFirst({ where: { id: options.guild } });
|
|
spinner.succeed('Exported settings');
|
|
|
|
spinner = ora('Exporting categories').start();
|
|
dump.categories = await prisma.category.findMany({
|
|
include: { questions: true },
|
|
where: { guildId: options.guild },
|
|
});
|
|
spinner.succeed(`Exported ${dump.categories.length} categories`);
|
|
|
|
spinner = ora('Exporting tags').start();
|
|
dump.tags = await prisma.tag.findMany({ where: { guildId: options.guild } });
|
|
spinner.succeed(`Exported ${dump.tags.length} tags`);
|
|
|
|
spinner = ora('Exporting tickets').start();
|
|
dump.tickets = await prisma.ticket.findMany({
|
|
include: {
|
|
archivedChannels: true,
|
|
archivedMessages: true,
|
|
archivedRoles: true,
|
|
archivedUsers: true,
|
|
feedback: true,
|
|
questionAnswers: true,
|
|
},
|
|
where: { guildId: options.guild },
|
|
});
|
|
dump.tickets = dump.tickets.map(ticket => {
|
|
if (ticket.topic) ticket.topic = db_cryptr.decrypt(ticket.topic);
|
|
|
|
ticket.archivedChannels = ticket.archivedChannels.map(channel => {
|
|
channel.name = db_cryptr.decrypt(channel.name);
|
|
return channel;
|
|
});
|
|
|
|
ticket.archivedMessages = ticket.archivedMessages.map(message => {
|
|
message.content = db_cryptr.decrypt(message.content);
|
|
return message;
|
|
});
|
|
|
|
ticket.archivedUsers = ticket.archivedUsers.map(user => {
|
|
user.displayName = db_cryptr.decrypt(user.displayName);
|
|
user.username = db_cryptr.decrypt(user.username);
|
|
return user;
|
|
});
|
|
|
|
if (ticket.feedback?.comment) {
|
|
ticket.feedback.comment = db_cryptr.decrypt(ticket.feedback.comment);
|
|
}
|
|
|
|
ticket.questionAnswers = ticket.questionAnswers.map(answer => {
|
|
if (answer.value) answer.value = db_cryptr.decrypt(answer.value);
|
|
return answer;
|
|
});
|
|
|
|
return ticket;
|
|
});
|
|
spinner.succeed(`Exported ${dump.tickets.length} tickets`);
|
|
|
|
spinner = ora(`Writing to "${file_path}"`).start();
|
|
|
|
// async to not freeze the spinner
|
|
await fse.promises.writeFile(file_path, file_cryptr.encrypt(JSON.stringify(dump)));
|
|
|
|
spinner.succeed(`Written to "${file_path}"`);
|