Working on API side.

This commit is contained in:
charles-gauthereau
2024-11-29 12:54:34 +01:00
parent 7e98c4b2a1
commit e49a953852
4 changed files with 91 additions and 88 deletions
+80
View File
@@ -0,0 +1,80 @@
import {Agent, Database} from "@prisma/client";
import {prisma} from "@/prisma";
import {NextResponse} from "next/server";
import {Body} from "./route";
// Regular expression for UUIDv4
const uuidv4Regex = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
// Type guard to check if a string is a valid UUIDv4
function isUuidv4(value: string): value is string {
console.log(value)
console.log(uuidv4Regex.test(value))
return uuidv4Regex.test(value);
}
export async function handleDatabases(body: Body, agent: Agent, lastContact: Date) {
const databasesResponse = [];
const formatDatabase = (database: Database) => ({
generatedId: database.generatedId,
dbms: database.dbms,
data: {
backup: {
action: true,
cron: "",
},
restore: {
action: false,
file: "",
},
},
});
for (const db of body.databases) {
const existingDatabase = await prisma.database.findFirst({
where: {
generatedId: db.generatedId,
},
});
if (!existingDatabase) {
if (!isUuidv4(db.generatedId)) {
return NextResponse.json(
{ error: "generatedId is not a valid uuid" },
{ status: 500 }
);
}
const databaseCreated = await prisma.database.create({
data: {
agentId: agent.id,
name: db.name,
dbms: db.dbms,
generatedId: db.generatedId,
lastContact: lastContact,
},
});
if (databaseCreated) {
databasesResponse.push(formatDatabase(databaseCreated));
}
} else {
const databaseUpdated = await prisma.database.update({
where: {
id: existingDatabase.id,
},
data: {
lastContact: lastContact,
},
});
databasesResponse.push(formatDatabase(databaseUpdated));
}
}
return databasesResponse;
}
+2 -87
View File
@@ -2,17 +2,7 @@ import {prisma} from "@/prisma";
import {NextRequest, NextResponse} from "next/server";
import {Agent, Database, Dbms} from "@prisma/client";
import {getFileUrlPresignedLocal} from "@/features/upload/private/upload.action";
import {tree} from "next/dist/build/templates/app-page";
// Regular expression for UUIDv4
const uuidv4Regex = /^[0-9a-f]{8}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{4}-[0-9a-f]{12}$/i;
// Type guard to check if a string is a valid UUIDv4
function isUuidv4(value: string): value is string {
console.log(value)
console.log(uuidv4Regex.test(value))
return uuidv4Regex.test(value);
}
import {handleDatabases} from "./helpers";
export type databaseAgent = {
@@ -33,14 +23,11 @@ export async function GET(request: Request) {
})
}
export async function POST(
request: Request,
{params}: { params: Promise<{ agentId: string }> }
) {
try {
const agentId = (await params).agentId
const body: Body = await request.json();
const lastContact = new Date()
@@ -50,14 +37,11 @@ export async function POST(
id: agentId
}
})
if (!agent) {
return NextResponse.json({error: "Agent not found"}, {status: 404})
}
const databasesResponse = await handleDatabases(body, agent, lastContact)
const response = {
agent: {
id: agentId,
@@ -66,10 +50,7 @@ export async function POST(
databases: databasesResponse
}
return Response.json({
message: response
})
return Response.json(response)
} catch (error) {
console.error('Error in POST handler:', error);
return NextResponse.json(
@@ -79,69 +60,3 @@ export async function POST(
}
}
async function handleDatabases(body: Body, agent: Agent, lastContact: Date) {
const databasesResponse = [];
// Helper function to format the database response
const formatDatabase = (database) => ({
generatedId: database.generatedId,
dbms: database.dbms,
data: {
backup: {
action: true,
cron: "",
},
restore: {
action: false,
file: "",
},
},
});
for (const db of body.databases) {
const existingDatabase = await prisma.database.findFirst({
where: {
generatedId: db.generatedId,
},
});
if (!existingDatabase) {
if (!isUuidv4(db.generatedId)) {
return NextResponse.json(
{ error: "generatedId is not a valid uuid" },
{ status: 500 }
);
}
// Create a new database
const databaseCreated = await prisma.database.create({
data: {
agentId: agent.id,
name: db.name,
dbms: db.dbms,
generatedId: db.generatedId,
lastContact: lastContact,
},
});
if (databaseCreated) {
databasesResponse.push(formatDatabase(databaseCreated));
}
} else {
// Update the existing database
const databaseUpdated = await prisma.database.update({
where: {
id: existingDatabase.id,
},
data: {
lastContact: lastContact,
},
});
databasesResponse.push(formatDatabase(databaseUpdated));
}
}
return databasesResponse;
}
@@ -0,0 +1,8 @@
/*
Warnings:
- A unique constraint covering the columns `[generatedId]` on the table `databases` will be added. If there are existing duplicate values, this will fail.
*/
-- CreateIndex
CREATE UNIQUE INDEX "databases_generatedId_key" ON "databases"("generatedId");
+1 -1
View File
@@ -62,7 +62,7 @@ function consoleAscii(){
console.log("\n" +
" ____ __ __ _____ \n" +
" / __ \\ ____ _____ / /_ ____ _ / /_ ____ _ _____ ___ / ___/ ___ _____ _ __ ___ _____\n" +
" / /_/ // __ \\ / ___// __// __ `// __ \\ / __ `// ___// _ \\ \\__ \\ / _ \\ / ___/| | / // _ \\ / ___/\n" +
" / /_/ // __ \\ / ___// __// __ // __ \\ / __ // ___// _ \\ \\__ \\ / _ \\ / ___/| | / // _ \\ / ___/\n" +
" / ____// /_/ // / / /_ / /_/ // /_/ // /_/ /(__ )/ __/ ___/ // __// / | |/ // __// / \n" +
"/_/ \\____//_/ \\__/ \\__,_//_.___/ \\__,_//____/ \\___/ /____/ \\___//_/ |___/ \\___//_/ \n" +
" \n")