11 |
13 | );
14 | }
15 |
--------------------------------------------------------------------------------
/app/login/password/page.tsx:
--------------------------------------------------------------------------------
1 | 'use client';
2 |
3 | import { ChangePassword } from '@/components/utility/change-password';
4 | import { PentestGPTContext } from '@/context/context';
5 | import { useRouter, useSearchParams } from 'next/navigation';
6 | import { useContext, useEffect } from 'react';
7 |
8 | export default function ChangePasswordPage() {
9 | const { user } = useContext(PentestGPTContext);
10 | const router = useRouter();
11 | const searchParams = useSearchParams();
12 |
13 | useEffect(() => {
14 | (async () => {
15 | // Check for error parameters in URL
16 | const error = searchParams.get('error');
17 |
18 | if (error) {
19 | router.push('/login?message=code_expired');
20 | return;
21 | }
22 |
23 | if (!user) {
24 | router.push('/login');
25 | }
26 | })();
27 | }, [user, searchParams]);
28 |
29 | return
21 |
43 | );
44 | }
45 |
--------------------------------------------------------------------------------
/app/posthog.js:
--------------------------------------------------------------------------------
1 | import { PostHog } from 'posthog-node';
2 | import { isProductionEnvironment } from '@/lib/utils';
3 |
4 | export default function PostHogClient() {
5 | if (!process.env.NEXT_PUBLIC_POSTHOG_KEY || !isProductionEnvironment) {
6 | return null;
7 | }
8 |
9 | const posthogClient = new PostHog(process.env.NEXT_PUBLIC_POSTHOG_KEY, {
10 | host: process.env.NEXT_PUBLIC_POSTHOG_HOST,
11 | flushAt: 1,
12 | flushInterval: 0,
13 | });
14 | return posthogClient;
15 | }
16 |
--------------------------------------------------------------------------------
/app/providers.tsx:
--------------------------------------------------------------------------------
1 | 'use client';
2 |
3 | import posthog from 'posthog-js';
4 | import { PostHogProvider as PHProvider } from 'posthog-js/react';
5 | import { useEffect, useContext } from 'react';
6 | import { PentestGPTContext } from '@/context/context';
7 |
8 | export function PostHogProvider({ children }: { children: React.ReactNode }) {
9 | const { isPremiumSubscription } = useContext(PentestGPTContext);
10 |
11 | useEffect(() => {
12 | if (!process.env.NEXT_PUBLIC_POSTHOG_KEY) return;
13 |
14 | posthog.init(process.env.NEXT_PUBLIC_POSTHOG_KEY, {
15 | api_host: `${process.env.NEXT_PUBLIC_APP_URL}/ingest`,
16 | ui_host: `${process.env.NEXT_PUBLIC_POSTHOG_HOST}`,
17 | capture_pageview: false, // Disable automatic pageview capture, as we capture manually
18 | autocapture: false, // Disable automatic event capture, as we capture manually
19 | disable_session_recording: true, // Disable session recording by default
20 | });
21 |
22 | // Only start session recording for premium users
23 | if (isPremiumSubscription) {
24 | posthog.startSessionRecording();
25 | }
26 | }, [isPremiumSubscription]);
27 |
28 | return
32 |
36 | Privacy Policy
37 |
38 |
39 | Terms of Use
40 |
41 |
42 |
33 |
43 |
44 | );
45 | }
46 |
47 | return null;
48 | };
49 |
--------------------------------------------------------------------------------
/components/chat/chat-helpers/drag.ts:
--------------------------------------------------------------------------------
1 | export function dragHelper(e: {
2 | preventDefault: () => void;
3 | currentTarget: any;
4 | clientX: any;
5 | }) {
6 | e.preventDefault(); // Prevents the click event after dragging
7 | const el = e.currentTarget;
8 | let isDragging = false;
9 | const posX = e.clientX;
10 | const scrollLeft = el.scrollLeft;
11 | function onMouseMove(e: { clientX: number }) {
12 | isDragging = true;
13 | const dx = e.clientX - posX;
14 | el.scrollLeft = scrollLeft - dx;
15 | }
16 | function onMouseUp() {
17 | document.removeEventListener('mousemove', onMouseMove);
18 | el.style.cursor = 'grab';
19 | if (isDragging) {
20 | el.addEventListener('click', preventClick, { once: true });
21 | }
22 | isDragging = false;
23 | }
24 | function preventClick(e: { stopPropagation: () => void }) {
25 | e.stopPropagation(); // Prevents the click event from firing after dragging
26 | }
27 | document.addEventListener('mousemove', onMouseMove);
28 | document.addEventListener('mouseup', onMouseUp, { once: true });
29 | el.style.cursor = 'grabbing';
30 | }
31 |
--------------------------------------------------------------------------------
/components/chat/chat-helpers/file-upload.ts:
--------------------------------------------------------------------------------
1 | import type { MessageImage } from '@/types/images/message-image';
2 | import type { Doc } from '@/convex/_generated/dataModel';
3 | import { toast } from 'sonner';
4 | import {
5 | MAX_TOTAL_FILES,
6 | calculateTotalFileCount,
7 | validateFileUpload,
8 | } from './file-constants';
9 |
10 | export const handleFileUpload = (
11 | files: File[],
12 | handleSelectDeviceFile: (file: File) => void,
13 | newMessageImages: MessageImage[] = [],
14 | newMessageFiles: Doc<'files'>[] = [],
15 | ) => {
16 | let currentTotalFiles = calculateTotalFileCount(
17 | newMessageImages,
18 | newMessageFiles,
19 | );
20 |
21 | for (const file of files) {
22 | if (currentTotalFiles >= MAX_TOTAL_FILES) {
23 | toast.error(
24 | `Maximum of ${MAX_TOTAL_FILES} files (including images) allowed.`,
25 | );
26 | break;
27 | }
28 |
29 | // Use unified validation function
30 | if (!validateFileUpload(file, newMessageImages, newMessageFiles)) {
31 | break;
32 | }
33 |
34 | handleSelectDeviceFile(file);
35 | currentTotalFiles++;
36 | }
37 | };
38 |
--------------------------------------------------------------------------------
/components/chat/chat-helpers/validation.ts:
--------------------------------------------------------------------------------
1 | import type { ChatSettings, LLM } from '@/types';
2 | import type { Doc } from '@/convex/_generated/dataModel';
3 |
4 | export const validateChatSettings = (
5 | chatSettings: ChatSettings | null,
6 | modelData: LLM | undefined,
7 | profile: Doc<'profiles'> | null,
8 | isContinuation: boolean,
9 | messageContent: string | null,
10 | ) => {
11 | if (!chatSettings) throw new Error('Chat settings not found');
12 | if (!modelData) throw new Error('Model not found');
13 | if (!profile) throw new Error('Profile not found');
14 | if (!isContinuation && !messageContent)
15 | throw new Error('Message content not found');
16 | };
17 |
--------------------------------------------------------------------------------
/components/chat/chat-hooks/use-agent-sidebar.tsx:
--------------------------------------------------------------------------------
1 | import React, {
2 | createContext,
3 | useContext,
4 | useState,
5 | type ReactNode,
6 | } from 'react';
7 | import type { AgentSidebarState } from '@/types/agent';
8 |
9 | interface AgentSidebarContextType {
10 | agentSidebar: AgentSidebarState;
11 | setAgentSidebar: React.Dispatch
15 | 
24 |
27 | );
28 | };
29 |
--------------------------------------------------------------------------------
/components/chat/chat-scroll-buttons.tsx:
--------------------------------------------------------------------------------
1 | import { ArrowDown } from 'lucide-react';
2 | import type { FC } from 'react';
3 |
4 | interface ChatScrollButtonsProps {
5 | isAtBottom: boolean;
6 | scrollToBottom: (options?: { force?: boolean }) => Promise{pluginInfo.name}
25 |{pluginInfo.description}
26 | {
19 | void scrollToBottom({ force: true });
20 | }}
21 | >
22 |
24 | )}
25 |
26 | );
27 | };
28 |
--------------------------------------------------------------------------------
/components/chat/chat-secondary-buttons.tsx:
--------------------------------------------------------------------------------
1 | import { useChatHandler } from '@/components/chat/chat-hooks/use-chat-handler';
2 | import { IconMessagePlus } from '@tabler/icons-react';
3 | import type { FC } from 'react';
4 | import { WithTooltip } from '../ui/with-tooltip';
5 |
6 | export const ChatSecondaryButtons: FC = () => {
7 | const { handleNewChat } = useChatHandler();
8 |
9 | return (
10 |
11 | New chat
}
13 | trigger={
14 |
25 |
33 |
34 | );
35 | }
36 |
37 | return (
38 | {children}
,
17 | portalRoot,
18 | );
19 | };
20 |
21 | export default Modal;
22 |
--------------------------------------------------------------------------------
/components/chat/shared-message.tsx:
--------------------------------------------------------------------------------
1 | 'use client';
2 |
3 | import React from 'react';
4 | import { MessageTypeResolver } from '@/components/messages/message-type-solver';
5 | import { ImageIcon } from 'lucide-react';
6 | import type { Doc } from '@/convex/_generated/dataModel';
7 |
8 | interface SharedMessageProps {
9 | message: Doc<'messages'>;
10 | isLastMessage: boolean;
11 | }
12 |
13 | export const SharedMessage: React.FC
19 |
43 | );
44 | };
45 |
--------------------------------------------------------------------------------
/components/chat/temporary-chat-info.tsx:
--------------------------------------------------------------------------------
1 | import React from 'react';
2 | import { BrandLarge } from '@/components/ui/brand';
3 |
4 | export const TemporaryChatInfo: React.FC = () => {
5 | return (
6 |
20 |
42 |
21 |
41 |
24 |
40 |
25 | {message.image_paths.length > 0 && (
26 |
39 |
29 |
32 | )}
33 |
7 |
17 | );
18 | };
19 |
--------------------------------------------------------------------------------
/components/icons/google-icon.tsx:
--------------------------------------------------------------------------------
1 | import React from 'react';
2 |
3 | interface GoogleIconProps extends React.SVGProps
8 |
10 | Temporary Chat
11 |12 | This chat won't appear in your history, and no data from these 13 | conversations will be stored or retained. To clear the chat, simply 14 | reload the page or click the "Clear chat" button. 15 |
16 |
67 | {icon}
68 |
70 | );
71 | };
72 |
--------------------------------------------------------------------------------
/components/messages/message-citations.tsx:
--------------------------------------------------------------------------------
1 | import React, { type FC, useMemo } from 'react';
2 | import { MessageMarkdown } from './message-markdown';
3 | import { ReasoningMarkdown } from './reasoning-markdown';
4 | import { CitationDisplay } from './citation-display';
5 |
6 | interface MessageCitationsProps {
7 | content: string;
8 | citations?: string[];
9 | isAssistant: boolean;
10 | reasoningWithCitations?: boolean;
11 | }
12 |
13 | export const MessageCitations: FC{text}
69 |
32 | {reasoningWithCitations ? (
33 |
39 | );
40 | };
41 |
--------------------------------------------------------------------------------
/components/messages/message-quick-feedback.tsx:
--------------------------------------------------------------------------------
1 | import type { Doc } from '@/convex/_generated/dataModel';
2 | import { Button } from '../ui/button';
3 |
4 | interface QuickFeedbackProps {
5 | handleBadResponseReason: (reason: string) => void;
6 | feedback?: Doc<'feedback'>;
7 | }
8 |
9 | export const MessageQuickFeedback: React.FC
23 |
37 | );
38 | };
39 |
--------------------------------------------------------------------------------
/components/messages/reasoning-markdown.tsx:
--------------------------------------------------------------------------------
1 | import React from 'react';
2 | import ReactMarkdown, { type Components } from 'react-markdown';
3 |
4 | const components: PartialWhat was wrong?
24 |
25 | {feedbackOptions.map((option) => (
26 |
34 | ))}
35 |
36 | {children}
24 | ), 25 | }; 26 | 27 | export const ReasoningMarkdown: React.FC<{ content: string }> = ({ 28 | content, 29 | }) => { 30 | return ( 31 |
32 | {content}
33 |
34 | );
35 | };
36 |
--------------------------------------------------------------------------------
/components/messages/terminal-messages/message-terminal-block.tsx:
--------------------------------------------------------------------------------
1 | import { type FC, memo, useMemo } from 'react';
2 | import chalk from 'chalk';
3 | import AnsiToHtml from 'ansi-to-html';
4 | import DOMPurify from 'isomorphic-dompurify';
5 |
6 | interface MessageTerminalBlockProps {
7 | value: string;
8 | }
9 |
10 | const customColors = {
11 | 0: '#000000',
12 | 1: '#FF5555',
13 | 2: '#50FA7B',
14 | 3: '#F1FA8C',
15 | 4: '#BD93F9',
16 | 5: '#FF79C6',
17 | 6: '#8BE9FD',
18 | 7: '#F8F8F2',
19 | };
20 |
21 | const converter = new AnsiToHtml({
22 | fg: '#F8F8F2',
23 | bg: '#282A36',
24 | colors: customColors,
25 | newline: true,
26 | });
27 |
28 | export const MessageTerminalBlock: FC
55 |
60 |
61 | );
62 | },
63 | );
64 |
65 | MessageTerminalBlock.displayName = 'MessageTerminalBlock';
66 |
--------------------------------------------------------------------------------
/components/messages/terminal-messages/shell-wait-block.tsx:
--------------------------------------------------------------------------------
1 | import React from 'react';
2 | import { Clock } from 'lucide-react';
3 | import { cn } from '@/lib/utils';
4 |
5 | interface ShellWaitBlockProps {
6 | block: {
7 | seconds: string;
8 | };
9 | }
10 |
11 | export const ShellWaitBlockComponent: React.FC
16 |
30 | );
31 | };
32 |
--------------------------------------------------------------------------------
/components/messages/terminal-messages/show-more-button.tsx:
--------------------------------------------------------------------------------
1 | import React from 'react';
2 | import { Button } from '@/components/ui/button';
3 | import { ArrowDown, ArrowUp } from 'lucide-react';
4 |
5 | interface ShowMoreButtonProps {
6 | isExpanded: boolean;
7 | onClick: () => void;
8 | remainingCount: number;
9 | type?: 'lines' | 'results';
10 | icon?: React.ReactNode;
11 | }
12 |
13 | export const ShowMoreButton: React.FC
17 |
29 |
18 |
28 |
19 |
22 |
23 |
27 |
24 | {block.seconds}
25 |
26 |
21 |
39 |
40 | );
41 |
--------------------------------------------------------------------------------
/components/messages/terminal-messages/types.ts:
--------------------------------------------------------------------------------
1 | export interface MessageTerminalProps {
2 | content: string;
3 | isAssistant: boolean;
4 | isLastMessage: boolean;
5 | }
6 |
7 | export interface TerminalBlock {
8 | command: string;
9 | stdout: string;
10 | error?: string;
11 | exec_dir?: string;
12 | }
13 |
14 | export interface ShellWaitBlock {
15 | seconds: string;
16 | }
17 |
18 | export interface FileContentBlock {
19 | path: string;
20 | content: string;
21 | mode?: 'read' | 'create' | 'append' | 'overwrite';
22 | }
23 |
24 | export interface InfoSearchWebBlock {
25 | query: string;
26 | results: Array<{
27 | title: string;
28 | url: string;
29 | description: string;
30 | }>;
31 | }
32 |
33 | export type ContentBlock =
34 | | {
35 | type: 'text';
36 | content: string;
37 | }
38 | | {
39 | type: 'terminal';
40 | content: TerminalBlock;
41 | }
42 | | {
43 | type: 'shell-wait';
44 | content: ShellWaitBlock;
45 | }
46 | | {
47 | type: 'file-content';
48 | content: FileContentBlock;
49 | }
50 | | {
51 | type: 'info-search-web';
52 | content: InfoSearchWebBlock;
53 | };
54 |
55 | export interface ShowMoreButtonProps {
56 | isExpanded: boolean;
57 | onClick: () => void;
58 | remainingLines: number;
59 | icon?: React.ReactNode;
60 | }
61 |
62 | export const MAX_VISIBLE_LINES = 12;
63 | export const COMMAND_LENGTH_THRESHOLD = 40; // Threshold for when to switch to full terminal view
64 |
--------------------------------------------------------------------------------
/components/messages/terminal-messages/use-auto-run-preference.ts:
--------------------------------------------------------------------------------
1 | import { useState, useEffect } from 'react';
2 | import type { AgentMode } from '@/types/llms';
3 |
4 | const agentModeKey = 'agentMode';
5 |
6 | export const useAgentModePreference = () => {
7 | const [agentMode, setAgentMode] = useState
16 |
22 | );
23 | };
24 |
--------------------------------------------------------------------------------
/components/sidebar/sidebar-header.tsx:
--------------------------------------------------------------------------------
1 | import { IconLayoutSidebar } from '@tabler/icons-react';
2 | import type { FC } from 'react';
3 | import type { ContentType } from '@/types';
4 | import { SidebarCreateButtons } from './sidebar-create-buttons';
5 | import { SIDEBAR_ICON_SIZE } from './sidebar-content';
6 | import { Button } from '../ui/button';
7 | import { WithTooltip } from '../ui/with-tooltip';
8 |
9 | interface SidebarHeaderProps {
10 | handleToggleSidebar: () => void;
11 | contentType: ContentType;
12 | handleSidebarVisibility: () => void;
13 | }
14 |
15 | export const SidebarHeader: FC
17 |
21 | {model.shortModelName}
20 |
23 |
31 |
44 |
45 | );
46 | };
47 |
--------------------------------------------------------------------------------
/components/sidebar/sidebar-invite-button.tsx:
--------------------------------------------------------------------------------
1 | import type { FC } from 'react';
2 | import { UserPlus } from 'lucide-react';
3 |
4 | interface SidebarInviteButtonProps {
5 | onInvite: () => void;
6 | title?: string;
7 | subtitle?: string;
8 | }
9 |
10 | export const SidebarInviteButton: FC
38 |
43 |
17 |
30 | );
31 | };
32 |
--------------------------------------------------------------------------------
/components/sidebar/sidebar-switch-item.tsx:
--------------------------------------------------------------------------------
1 | import type { ContentType } from '@/types';
2 | import type { FC } from 'react';
3 | import { TabsTrigger } from '../ui/tabs';
4 | import { WithTooltip } from '../ui/with-tooltip';
5 |
6 | interface SidebarSwitchItemProps {
7 | contentType: ContentType;
8 | icon: React.ReactNode;
9 | onContentTypeChange: (contentType: ContentType) => void;
10 | display?: string;
11 | }
12 |
13 | export const SidebarSwitchItem: FC
21 |
29 |
22 |
28 |
24 |
27 | {title}
25 | {subtitle}
26 |
14 |
29 | );
30 | };
31 |
--------------------------------------------------------------------------------
/components/sidebar/sidebar.tsx:
--------------------------------------------------------------------------------
1 | import { PentestGPTContext } from '@/context/context';
2 | import type { ContentType } from '@/types';
3 | import { type FC, useContext } from 'react';
4 | import { SIDEBAR_WIDTH } from '../ui/dashboard';
5 | import { TabsContent } from '../ui/tabs';
6 | import { SidebarContent } from './sidebar-content';
7 |
8 | interface SidebarProps {
9 | contentType: ContentType;
10 | showSidebar: boolean;
11 | }
12 |
13 | export const Sidebar: FC
18 |
28 |
19 |
27 |
21 |
26 | Upgrade plan
22 |
23 | Upgrade for file upload, smarter AI, and more
24 |
25 |
32 | {(() => {
33 | switch (contentType) {
34 | case 'chats':
35 | return renderSidebarContent('chats', chats);
36 |
37 | default:
38 | return null;
39 | }
40 | })()}
41 |
42 |
52 |
53 |
54 |
55 | );
56 | }
57 |
58 | return {nodes}
;
59 | }
60 |
--------------------------------------------------------------------------------
/components/ui/alert.tsx:
--------------------------------------------------------------------------------
1 | import * as React from 'react';
2 | import { cva, type VariantProps } from 'class-variance-authority';
3 |
4 | import { cn } from '@/lib/utils';
5 |
6 | const alertVariants = cva(
7 | '[&>svg]:text-foreground relative w-full rounded-lg border p-4 [&>svg+div]:translate-y-[-3px] [&>svg]:absolute [&>svg]:left-4 [&>svg]:top-4 [&>svg~*]:pl-7',
8 | {
9 | variants: {
10 | variant: {
11 | default: 'bg-background text-foreground',
12 | destructive:
13 | 'border-destructive/50 text-destructive dark:border-destructive [&>svg]:text-destructive',
14 | },
15 | },
16 | defaultVariants: {
17 | variant: 'default',
18 | },
19 | },
20 | );
21 |
22 | const Alert = React.forwardRef<
23 | HTMLDivElement,
24 | React.HTMLAttributes
29 |
36 | );
37 | });
38 |
39 | BrandBase.displayName = 'BrandBase';
40 |
41 | export const Brand: FC
30 |
32 | {scale === 0.4 && (
33 | PentestGPT
34 | )}
35 |
16 | {used}/{limit}
17 |
18 | );
19 | };
20 |
--------------------------------------------------------------------------------
/components/ui/link-with-tooltip.tsx:
--------------------------------------------------------------------------------
1 | import React from 'react';
2 | import { WithTooltip } from './with-tooltip';
3 | import { WebsiteCard } from '../messages/citation-display';
4 |
5 | interface LinkWithTooltipProps {
6 | href: string;
7 | title?: string;
8 | children: React.ReactNode;
9 | }
10 |
11 | export const LinkWithTooltip: React.FC
19 |
25 |
26 |
27 |
28 |
41 |
42 |
43 |
45 |
46 |
47 |
48 | );
49 | }
50 |
--------------------------------------------------------------------------------
/components/ui/popover.tsx:
--------------------------------------------------------------------------------
1 | 'use client';
2 |
3 | import * as React from 'react';
4 | import * as PopoverPrimitive from '@radix-ui/react-popover';
5 |
6 | import { cn } from '@/lib/utils';
7 |
8 | const Popover = PopoverPrimitive.Root;
9 |
10 | const PopoverTrigger = PopoverPrimitive.Trigger;
11 |
12 | const PopoverContent = React.forwardRef<
13 | React.ElementRef{showPassword ? 'Hide password' : 'Show password'}
44 |
7 |
9 | );
10 | };
11 |
--------------------------------------------------------------------------------
/components/ui/scroll-area.tsx:
--------------------------------------------------------------------------------
1 | 'use client';
2 |
3 | import * as React from 'react';
4 | import * as ScrollAreaPrimitive from '@radix-ui/react-scroll-area';
5 |
6 | import { cn } from '@/lib/utils';
7 |
8 | function ScrollArea({
9 | className,
10 | children,
11 | ...props
12 | }: React.ComponentProps
5 |
6 | {children}
7 |
8 |
9 | );
10 |
11 | export const Th: React.FC
17 |
24 | placeholder
27 |
19 | setIsSharedChatsPopupOpen(false)}
48 | /> */}
49 |
50 | );
51 | };
52 |
--------------------------------------------------------------------------------
/components/utility/profile-tabs/personalization-tab.tsx:
--------------------------------------------------------------------------------
1 | import type { FC } from 'react';
2 | import { Label } from '@/components/ui/label';
3 | import { TextareaAutosize } from '@/components/ui/textarea-autosize';
4 | import { LimitDisplay } from '@/components/ui/limit-display';
5 | import { PROFILE_CONTEXT_MAX } from '@/db/limits';
6 | import { Button } from '@/components/ui/button';
7 |
8 | interface PersonalizationTabProps {
9 | profileInstructions: string;
10 | setProfileInstructions: (value: string) => void;
11 | onSave: () => void;
12 | isMobile: boolean;
13 | }
14 |
15 | export const PersonalizationTab: FC
20 | {/*
45 | {/*
21 |
22 |
29 |
*/}
30 |
31 |
32 |
44 |
33 |
34 |
42 |
43 |
25 |
55 | );
56 | };
57 |
--------------------------------------------------------------------------------
/components/utility/providers.tsx:
--------------------------------------------------------------------------------
1 | 'use client';
2 |
3 | import { TooltipProvider } from '@/components/ui/tooltip';
4 | import { AlertProvider } from '@/context/alert-context';
5 | import {
6 | ThemeProvider as NextThemesProvider,
7 | type ThemeProviderProps,
8 | } from 'next-themes';
9 | import type { FC } from 'react';
10 |
11 | export const Providers: FC
26 |
48 |
49 | {isMobile && (
50 |
27 |
31 |
32 |
33 |
51 |
52 |
53 | )}
54 |