12826888982

Committed 17 Jan 2025 10:05AM CUT coverage: 68.391% (+1.2%) from 67.241%

Build # 12826888982

Build Type

Pull #99

github

Committed by

web-flow

Commit Message

Merge e9242c8bc into 32def3dae

Pull Request Pull Request #99: fix: fix data fetching bug

Run Details

205 of 379 branches covered (54.09%)

Branch coverage included in aggregate %.

390 of 491 relevant lines covered (79.43%)

27.11 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

70.87

/src/lib/utils.ts

import { AlertConversation, Conversation } from "@/api/generated/types.gen";
import { MaliciousPkgType, TriggerType } from "@/types";
import { isToday, isYesterday } from "date-fns";

const ONE_DAY_MS = 24 * 60 * 60 * 1000;
const SEVEN_DAYS_MS = 7 * ONE_DAY_MS;
const TEEN_DAYS_MS = 14 * ONE_DAY_MS;
const THTY_DAYS_MS = 30 * ONE_DAY_MS;

export function extractTitleFromMessage(message: string) {
  try {
    const regex = /^(.*)```[\s\S]*?```(.*)$/s;
    const match = message.match(regex);

    if (match !== null && match !== undefined) {
      const beforeMarkdown = match[1]?.trim();
      const afterMarkdown = match[2]?.trim();
      const title = beforeMarkdown || afterMarkdown;
      return title;
    }

    return message.trim();
  } catch {
    return message.trim();
  }
}

function getGroup(differenceInMs: number, promptDate: Date): string {
  if (isToday(promptDate)) {
    return "Today";
  }
  if (isYesterday(promptDate)) {
    return "Yesterday";
  }
  if (differenceInMs <= SEVEN_DAYS_MS) {
    return "Previous 7 days";
  }
  if (differenceInMs <= TEEN_DAYS_MS) {
    return "Previous 14 days";
  }
  if (differenceInMs <= THTY_DAYS_MS) {
    return "Previous 30 days";
  }
  return "Beyond 30 days";
}

export function groupPromptsByRelativeDate(prompts: Conversation[]) {
  const promptsSorted = prompts.sort(
    (a, b) =>
      new Date(b.conversation_timestamp).getTime() -
      new Date(a.conversation_timestamp).getTime(),
  );

  const grouped = promptsSorted.reduce(
    (groups, prompt) => {
      const promptDate = new Date(prompt.conversation_timestamp);
      const now = new Date();
      const differenceInMs = now.getTime() - promptDate.getTime();
      const group = getGroup(differenceInMs, promptDate);

      if (!groups[group]) {
        groups[group] = [];
      }

      (groups[group] ?? []).push(prompt);
      return groups;
    },
    {} as Record<string, Conversation[]>,
  );

  return grouped;
}

export function getAllIssues(alerts: AlertConversation[]) {
  const groupedTriggerCounts = alerts.reduce<Record<string, number>>(
    (acc, alert) => {
      const triggerType: TriggerType = alert.trigger_type;
      if (triggerType) {
        acc[triggerType] = (acc[triggerType] || 0) + 1;
      }
      return acc;
    },
    {},
  );

  const maxCount = Math.max(...Object.values(groupedTriggerCounts));

  const sortedTagCounts = Object.entries(groupedTriggerCounts).sort(
    ([, countA], [, countB]) => countB - countA,
  );
  return { maxCount, sortedTagCounts };
}

export function getMaliciousPackages() {
  const packageCounts = ([] as { packages: [] }[]).reduce<
    Record<string, number>
  >((acc, prompt) => {
    (prompt?.packages ?? []).forEach((pkg) => {
      acc[pkg] = (acc[pkg] || 0) + 1;
    });
    return acc;
  }, {});

  const chartData = Object.entries(packageCounts).map(([pkg, count]) => ({
    id: pkg,
    label: pkg,
    value: count,
  }));

  return chartData;
}

export function sanitizeQuestionPrompt({
  question,
  answer,
}: {
  question: string;
  answer: string;
}) {
  try {
    // it shouldn't be possible to receive the prompt answer without a question
    if (!answer) return question;

    // Check if 'answer' is truthy; if so, try to find and return the text after "Query:"
    const index = question.indexOf("Query:");
    if (index !== -1) {
      // Return the substring starting right after the first occurrence of "Query:"
      // Adding the length of "Query:" to the index to start after it
      return question.substring(index + "Query:".length).trim();
    }
    return question;
  } catch (error) {
    // Log the error and return the original question as a fallback
    console.error("Error processing the question:", error);
    return question;
  }
}

export function getMaliciousPackage(
  value: AlertConversation["trigger_string"],
): string | (MaliciousPkgType & { [key: string]: string }) | null {
  if (typeof value === "string") {
    return value;
  }

  if (typeof value === "object" && value !== null) {
    return value as MaliciousPkgType;
  }

  return null;
}

1	import { AlertConversation, Conversation } from "@/api/generated/types.gen";
2	import { MaliciousPkgType, TriggerType } from "@/types";
3	import { isToday, isYesterday } from "date-fns";
4
5	const ONE_DAY_MS = 24 * 60 * 60 * 1000;	5✔
6	const SEVEN_DAYS_MS = 7 * ONE_DAY_MS;	5✔
7	const TEEN_DAYS_MS = 14 * ONE_DAY_MS;	5✔
8	const THTY_DAYS_MS = 30 * ONE_DAY_MS;	5✔
9
10	export function extractTitleFromMessage(message: string) {
11	try {	3✔
12	const regex = /^(.)```[\s\S]?```(.*)$/s;	3✔
13	const match = message.match(regex);	3✔
14
15	if (match !== null && match !== undefined) {	3✔
16	const beforeMarkdown = match[1]?.trim();	1✔
17	const afterMarkdown = match[2]?.trim();	1✔
18	const title = beforeMarkdown \|\| afterMarkdown;	1✔
19	return title;	1✔
20	}
21
22	return message.trim();	2✔
23	} catch {
24	return message.trim();	×
25	}
26	}
27
28	function getGroup(differenceInMs: number, promptDate: Date): string {
29	if (isToday(promptDate)) {	2!
30	return "Today";	×
31	}
32	if (isYesterday(promptDate)) {	2!
33	return "Yesterday";	×
34	}
35	if (differenceInMs <= SEVEN_DAYS_MS) {	2!
36	return "Previous 7 days";	×
37	}
38	if (differenceInMs <= TEEN_DAYS_MS) {	2!
39	return "Previous 14 days";	×
40	}
41	if (differenceInMs <= THTY_DAYS_MS) {	2!
42	return "Previous 30 days";	2✔
43	}
44	return "Beyond 30 days";	×
45	}
46
47	export function groupPromptsByRelativeDate(prompts: Conversation[]) {
48	const promptsSorted = prompts.sort(	2✔
49	(a, b) =>
50	new Date(b.conversation_timestamp).getTime() -	×
51	new Date(a.conversation_timestamp).getTime(),
52	);
53
54	const grouped = promptsSorted.reduce(	2✔
55	(groups, prompt) => {
56	const promptDate = new Date(prompt.conversation_timestamp);	2✔
57	const now = new Date();	2✔
58	const differenceInMs = now.getTime() - promptDate.getTime();	2✔
59	const group = getGroup(differenceInMs, promptDate);	2✔
60
61	if (!groups[group]) {	2!
62	groups[group] = [];	2✔
63	}
64
65	(groups[group] ?? []).push(prompt);	2!
66	return groups;	2✔
67	},
68	{} as Record<string, Conversation[]>,
69	);
70
71	return grouped;	2✔
72	}
73
74	export function getAllIssues(alerts: AlertConversation[]) {
75	const groupedTriggerCounts = alerts.reduce<Record<string, number>>(	65✔
76	(acc, alert) => {
77	const triggerType: TriggerType = alert.trigger_type;	138✔
78	if (triggerType) {	138!
79	acc[triggerType] = (acc[triggerType] \|\| 0) + 1;	138✔
80	}
81	return acc;	138✔
82	},
83	{},
84	);
85
86	const maxCount = Math.max(...Object.values(groupedTriggerCounts));	65✔
87
88	const sortedTagCounts = Object.entries(groupedTriggerCounts).sort(	65✔
89	([, countA], [, countB]) => countB - countA,	56✔
90	);
91	return { maxCount, sortedTagCounts };	65✔
92	}
93
94	export function getMaliciousPackages() {
95	const packageCounts = ([] as { packages: [] }[]).reduce<	×
96	Record<string, number>
97	>((acc, prompt) => {
98	(prompt?.packages ?? []).forEach((pkg) => {	×
99	acc[pkg] = (acc[pkg] \|\| 0) + 1;	×
100	});
101	return acc;	×
102	}, {});
103
104	const chartData = Object.entries(packageCounts).map(([pkg, count]) => ({	×
105	id: pkg,
106	label: pkg,
107	value: count,
108	}));
109
110	return chartData;	×
111	}
112
113	export function sanitizeQuestionPrompt({
114	question,
115	answer,
116	}: {
117	question: string;
118	answer: string;
119	}) {
120	try {	3✔
121	// it shouldn't be possible to receive the prompt answer without a question
122	if (!answer) return question;	3!
123
124	// Check if 'answer' is truthy; if so, try to find and return the text after "Query:"
125	const index = question.indexOf("Query:");	3✔
126	if (index !== -1) {	3!
127	// Return the substring starting right after the first occurrence of "Query:"
128	// Adding the length of "Query:" to the index to start after it
129	return question.substring(index + "Query:".length).trim();	×
130	}
131	return question;	3✔
132	} catch (error) {
133	// Log the error and return the original question as a fallback
134	console.error("Error processing the question:", error);	×
135	return question;	×
136	}
137	}
138
139	export function getMaliciousPackage(
140	value: AlertConversation["trigger_string"],
141	): string \| (MaliciousPkgType & { [key: string]: string }) \| null {
142	if (typeof value === "string") {	310✔
143	return value;	208✔
144	}
145
146	if (typeof value === "object" && value !== null) {	102✔
147	return value as MaliciousPkgType;	94✔
148	}
149
150	return null;	8✔
151	}

stacklok / codegate-ui / 12826888982

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous