Brainiac

This JavaScript code implements the core functionality of a Visual Studio Code extension named "Brainiac". Here's a breakdown in bullet points:

Core Functionality

• Chatbot Integration: The code enables communication with an AI chatbot, likely using a service like Microsoft ChatGpt. It handles user prompts, sends them to the AI, receives responses as a stream, and displays them in a dedicated webview panel within VS Code.
• Authentication & Token Handling: It manages user login, likely to an internal authentication service (zpsso.zhaopin.com), retrieves and stores an authentication token (INNER_AUTHENTICATION cookie), and handles token refresh when necessary.
• Configuration & State Management:
  • It uses VS Code's configuration API to store and retrieve extension settings like the AI model (microsoftChatGpt), temperature (default: 0.5), system message, max tokens, request timeout, and the number of chat records to store (default: 10).
  • It utilizes VS Code's global state API to persist chat records and user identity.
• File Handling:
  • The extension allows users to attach files to their prompts (documents, code files, etc.).
  • It handles file selection, reads content (including converting .pdf and .docx to text), checks file size (maximum 500 KB), and sends the relevant text to the AI.
• Record Management:
  • Chat records are stored locally with encryption using the BRAINIAC seed.
  • Users can view, delete, and reset the context (start a new conversation) using commands and UI elements.
• User Interface:
  • The extension creates a webview panel (brainiac.chatView) displaying the chat interface.
  • It uses Tailwind CSS and other libraries for styling and interactive elements.
  • The webview communicates with the extension using VS Code's messaging API.
• Error Handling and Retries:
  • It includes error handling for failed requests, parsing errors, invalid authentication, and network issues.
  • It implements a retry mechanism for failed AI requests (up to 3 retries).
  • It provides user feedback through alerts, warnings, and error messages.

How It Works

• Extension Activation: The activate function is called when the extension is loaded, setting up the webview provider, registering commands, and handling configuration changes.
• Chat Functionality: The chat function initiates the conversation process, sending the prompt to the AI, managing the response stream, and updating the UI.
• Background Worker: The WorkerMessageHandler handles messages from the webview and interacts with the core logic of the extension (document loading, parsing, etc.).
• PDF.js & Mammoth.js: These libraries are used to extract text from PDF and Word documents, respectively.
• Crypto.js: This library handles the AES encryption and decryption for storing chat records securely.

Key Classes

• PDFDocumentLoadingTask: Manages the asynchronous loading of PDF documents.
• PDFDocumentProxy: Provides access to PDF document information and functionality.
• PDFPageProxy: Represents a single page within a PDF document.
• PDFWorker: Handles PDF processing in a separate worker thread (or using a fallback mechanism if worker threads are disabled).
• AnnotationStorage: Stores and manages user annotations on PDF documents.
• AnnotationEditor: Base class for annotation editing tools (FreeTextEditor, InkEditor, StampEditor).
• AnnotationEditorUIManager: Manages the UI and interactions for annotation editors.
• AnnotationLayer: Renders annotations on a PDF page.
• XFAFactory: Handles parsing and rendering XFA forms (XML Forms Architecture).

Overall, the code is complex and demonstrates a well-structured architecture for building a feature-rich VS Code extension. It integrates with various APIs and external libraries to provide a seamless chat experience with AI, enhanced by file handling, persistent record management, and annotation editing capabilities.