training.import_all

Instructions

Implementation Reference

• src/tools/training_extractor.ts:242-295 (handler)

  Handler function that imports specified training data sources (CSRF, XSS, SQLi, registration, dorking) from pre-loaded arrays into the database using saveTrainingData, returning import statistics.

    async ({ sources = ['all'] }: any): Promise<ToolResult> => {
      try {
        const allData: any[] = [];
        const importSources = sources.includes('all')
          ? ['csrf', 'xss', 'sqli', 'registration', 'dorking']
          : sources;
  
        if (importSources.includes('csrf')) {
          allData.push(...CSRF_TRAINING_DATA);
        }
        if (importSources.includes('xss')) {
          allData.push(...XSS_TRAINING_DATA);
        }
        if (importSources.includes('sqli')) {
          allData.push(...SQLI_TRAINING_DATA);
        }
        if (importSources.includes('registration')) {
          allData.push(...REGISTRATION_TRAINING_DATA);
        }
        if (importSources.includes('dorking')) {
          allData.push(...GOOGLE_DORKING_PATTERNS);
        }
  
        const imported: number[] = [];
        for (const data of allData) {
          try {
            const id = await saveTrainingData(
              data.source,
              data.sourceId,
              data.vulnerabilityType,
              data.targetPattern,
              data.payloadPattern,
              data.successPattern,
              data.failurePattern,
              data.contextData,
              data.score
            );
            imported.push(id);
          } catch (error: any) {
            console.error(`Error importing ${data.sourceId}:`, error.message);
          }
        }
  
        return formatToolResult(true, {
          imported: imported.length,
          total: allData.length,
          ids: imported,
          sources: importSources,
        });
      } catch (error: any) {
        return formatToolResult(false, null, error.message);
      }
    }
  );

• src/tools/training_extractor.ts:228-241 (schema)

  Input schema defining optional 'sources' array parameter for selecting which training data sets to import.

  {
    description: 'Import all pre-loaded training data from Intigriti, PortSwigger, and other sources',
    inputSchema: {
      type: 'object',
      properties: {
        sources: {
          type: 'array',
          items: { type: 'string' },
          description: 'Sources to import (csrf, xss, sqli, registration, dorking, all)',
          default: ['all'],
        },
      },
    },
  },

• src/tools/training_extractor.ts:225-295 (registration)

  Registration of the 'training.import_all' tool using server.tool(), including schema and handler.

  // Import all pre-loaded training data
  server.tool(
    'training.import_all',
    {
      description: 'Import all pre-loaded training data from Intigriti, PortSwigger, and other sources',
      inputSchema: {
        type: 'object',
        properties: {
          sources: {
            type: 'array',
            items: { type: 'string' },
            description: 'Sources to import (csrf, xss, sqli, registration, dorking, all)',
            default: ['all'],
          },
        },
      },
    },
    async ({ sources = ['all'] }: any): Promise<ToolResult> => {
      try {
        const allData: any[] = [];
        const importSources = sources.includes('all')
          ? ['csrf', 'xss', 'sqli', 'registration', 'dorking']
          : sources;
  
        if (importSources.includes('csrf')) {
          allData.push(...CSRF_TRAINING_DATA);
        }
        if (importSources.includes('xss')) {
          allData.push(...XSS_TRAINING_DATA);
        }
        if (importSources.includes('sqli')) {
          allData.push(...SQLI_TRAINING_DATA);
        }
        if (importSources.includes('registration')) {
          allData.push(...REGISTRATION_TRAINING_DATA);
        }
        if (importSources.includes('dorking')) {
          allData.push(...GOOGLE_DORKING_PATTERNS);
        }
  
        const imported: number[] = [];
        for (const data of allData) {
          try {
            const id = await saveTrainingData(
              data.source,
              data.sourceId,
              data.vulnerabilityType,
              data.targetPattern,
              data.payloadPattern,
              data.successPattern,
              data.failurePattern,
              data.contextData,
              data.score
            );
            imported.push(id);
          } catch (error: any) {
            console.error(`Error importing ${data.sourceId}:`, error.message);
          }
        }
  
        return formatToolResult(true, {
          imported: imported.length,
          total: allData.length,
          ids: imported,
          sources: importSources,
        });
      } catch (error: any) {
        return formatToolResult(false, null, error.message);
      }
    }
  );