Skip to content

Commit

Permalink
♻️ Update return types in database query functions to use undefined i…
Browse files Browse the repository at this point in the history 
…nstead of null for improved type safety
  • Loading branch information
@pitzzahh
pitzzahh authored Dec 13, 2024
1 parent ede778f commit e0870ad
Showing 1 changed file with 96 additions and 138 deletions.
234 changes: 96 additions & 138 deletions src/lib/db/queries/index.ts
View file
Original file line number Diff line number Diff line change
Expand Up @@ -3,38 +3,26 @@ import { databaseInstance } from '..';
import type { LoadType } from '@/types/load';
import type { Node, Project } from '@/db/schema';

export async function getCurrentProject(project_id?: string): Promise<Project | null> {
export async function getCurrentProject(project_id?: string): Promise<Project | undefined> {
const db = await databaseInstance();

try {
const query = db.projects.find({
selector: project_id
? {
id: project_id
}
: undefined
});
return (await query.exec()).at(0)?._data as Project | null;
} catch (error) {
console.log(error);
throw error;
}
const query = db.projects.find({
selector: project_id
? {
id: project_id
}
: undefined
});
return (await query.exec()).at(0)?._data as Project | undefined;
}

export async function getRootNode(): Promise<Node | null> {
export async function getRootNode(): Promise<Node | undefined> {
const db = await databaseInstance();

try {
const query = db.nodes.find({
selector: {
node_type: 'root'
}
});
return (await query.exec()).at(0)?._data as Node | null;
} catch (error) {
console.log(error);
throw error;
}
const query = db.nodes.find({
selector: {
node_type: 'root'
}
});
return (await query.exec()).at(0)?._data as Node | undefined;
}

export async function checkNodeExists({
Expand Down Expand Up @@ -71,140 +59,110 @@ export async function checkNodeExists({

export async function getNodeById(target_id: string) {
const db = await databaseInstance();

try {
const node = await db.nodes
.findOne({
selector: {
id: target_id
}
})
.exec();

if (node) {
return node._data;
}

return null;
} catch (error) {
console.log(error);
throw error;
}
return (await db.nodes
.findOne({
selector: {
id: target_id
}
})
.exec())?._data;
}

export async function getChildNodesByParentId(parent_id: string): Promise<Node[]> {
const db = await databaseInstance();
const query = db.nodes.find({
selector: {
parent_id
}
});
const children = (await query.exec()).map((doc) => doc._data);

try {
const query = db.nodes.find({
selector: {
parent_id
}
});
const children = (await query.exec()).map((doc) => doc._data);

// sorts the circuit number of every load node
const sortedChildren = children.sort((a, b) => {
return (a.circuit_number || 0) - (b.circuit_number || 0);
});
// sorts the circuit number of every load node
const sortedChildren = children.sort((a, b) => {
return (a.circuit_number || 0) - (b.circuit_number || 0);
});

return sortedChildren as Node[];
} catch (error) {
console.log(error);
throw error;
}
return sortedChildren as Node[];
}

export async function getParentNodes(excluded_id?: string) {
const db = await databaseInstance();
const query = db.nodes.find({
selector: {
node_type: { $in: ['panel', 'root'] },
...(excluded_id && { id: { $ne: excluded_id } })
}
});

try {
const query = db.nodes.find({
selector: {
node_type: { $in: ['panel', 'root'] },
...(excluded_id && { id: { $ne: excluded_id } })
const parent_nodes = (await query.exec())
.map((doc) => {
const data = doc._data;
if (data.panel_data) {
return { name: data.panel_data.name || '', id: data.id };
}
});

const parent_nodes = (await query.exec())
.map((doc) => {
const data = doc._data;
if (data.panel_data) {
return { name: data.panel_data.name || '', id: data.id };
}

if (data.highest_unit_form) {
return { name: data.highest_unit_form.distribution_unit || '', id: data.id };
}
})
.filter(Boolean);
if (data.highest_unit_form) {
return { name: data.highest_unit_form.distribution_unit || '', id: data.id };
}
})
.filter(Boolean);

return parent_nodes;
} catch (error) {
console.log(error);
throw error;
}
return parent_nodes;
}

export async function getComputedLoads(
parent_id: string
): Promise<(Node & { va: number; current: number; voltage: number; load_description: string })[]> {
const db = await databaseInstance();

try {
const childNodes = await db.nodes.find({ selector: { parent_id } }).exec();

const loadsWithComputedFields = await Promise.all(
childNodes.map(async (doc) => {
const data = doc._data;
const voltage = 230; // this may change depending on phase

const va =
computeVoltAmphere({
load_type: data.load_data?.load_type as LoadType,
quantity: data.load_data?.quantity ?? 0,
varies: data.load_data?.varies ?? 0
}) || 0;

const current = va / voltage;

if (data.panel_data) {
// Recursive fetch for panel's computed loads
const panelLoads = await getComputedLoads(data.id);

const totalLoads = panelLoads.reduce(
(totals, load) => ({
va: totals.va + (load.va || 0),
current: totals.current + (load.current || 0)
}),
{ va: 0, current: 0 } // Default initial values
);

return {
...data,
load_description: data.panel_data.name || '',
voltage,
va: totalLoads.va,
current: parseFloat(totalLoads.current.toFixed(2))
};
}
const childNodes = await db.nodes.find({ selector: { parent_id } }).exec();

const loadsWithComputedFields = await Promise.all(
childNodes.map(async (doc) => {
const data = doc._data;
const voltage = 230; // this may change depending on phase

const va =
computeVoltAmphere({
load_type: data.load_data?.load_type as LoadType,
quantity: data.load_data?.quantity ?? 0,
varies: data.load_data?.varies ?? 0
}) || 0;

const current = va / voltage;

if (data.panel_data) {
// Recursive fetch for panel's computed loads
const panelLoads = await getComputedLoads(data.id);

const totalLoads = panelLoads.reduce(
(totals, load) => ({
va: totals.va + (load.va || 0),
current: totals.current + (load.current || 0)
}),
{ va: 0, current: 0 } // Default initial values
);

return {
...data,
load_description: data.load_data?.load_description || '',
load_description: data.panel_data.name || '',
voltage,
va,
current: parseFloat(current.toFixed(2))
va: totalLoads.va,
current: parseFloat(totalLoads.current.toFixed(2))
};
})
);
}

// sorted loads by circuit_number
return loadsWithComputedFields.sort(
(a, b) => (a.circuit_number || 0) - (b.circuit_number || 0)
) as (Node & { va: number; current: number; voltage: number; load_description: string })[];
} catch (error) {
console.error('Error in getComputedLoads:', error);
throw error;
}
return {
...data,
load_description: data.load_data?.load_description || '',
voltage,
va,
current: parseFloat(current.toFixed(2))
};
})
);

// sorted loads by circuit_number
return loadsWithComputedFields.sort(
(a, b) => (a.circuit_number || 0) - (b.circuit_number || 0)
) as (Node & { va: number; current: number; voltage: number; load_description: string })[];
}

0 comments on commit e0870ad

Please sign in to comment.