import { z } from 'zod';
import { stringToNumberTransform, parseJsonStringArray } from '../utils/transforms.js';
import {
  severitySchema,
  severitiesSchema,
  statusSchema,
  resolutionSchema,
  typeSchema,
  cleanCodeAttributeCategoriesSchema,
  impactSeveritiesSchema,
  impactSoftwareQualitiesSchema,
  pullRequestNullableSchema,
} from './common.js';

/**
 * Schema for mark issue false positive tool
 */
export const markIssueFalsePositiveToolSchema = {
  issue_key: z.string().describe('The key of the issue to mark as false positive'),
  comment: z
    .string()
    .optional()
    .describe('Optional comment explaining why this is a false positive'),
};

/**
 * Schema for mark issue won\'t fix tool
 */
export const markIssueWontFixToolSchema = {
  issue_key: z.string().describe("The key of the issue to mark as won't fix"),
  comment: z.string().optional().describe("Optional comment explaining why this won't be fixed"),
};

/**
 * Schema for mark issues false positive (bulk) tool
 */
export const markIssuesFalsePositiveToolSchema = {
  issue_keys: z
    .array(z.string())
    .min(1, 'At least one issue key is required')
    .describe('Array of issue keys to mark as false positive'),
  comment: z
    .string()
    .optional()
    .describe('Optional comment explaining why these are false positives'),
};

/**
 * Schema for mark issues won\'t fix (bulk) tool
 */
export const markIssuesWontFixToolSchema = {
  issue_keys: z
    .array(z.string())
    .min(1, 'At least one issue key is required')
    .describe("Array of issue keys to mark as won't fix"),
  comment: z.string().optional().describe("Optional comment explaining why these won't be fixed"),
};

/**
 * Schema for add comment to issue tool
 */
export const addCommentToIssueToolSchema = {
  issue_key: z
    .string()
    .min(1, 'Issue key is required')
    .describe('The key of the issue to add a comment to'),
  text: z
    .string()
    .min(1, 'Comment text is required')
    .describe('The comment text to add. Supports markdown formatting for rich text content'),
};

/**
 * Schema for assign issue tool
 */
export const assignIssueToolSchema = {
  issueKey: z.string().min(1, 'Issue key is required').describe('The key of the issue to assign'),
  assignee: z
    .string()
    .optional()
    .describe('The username of the assignee. Leave empty to unassign the issue'),
};

/**
 * Schema for confirm issue tool
 */
export const confirmIssueToolSchema = {
  issue_key: z.string().describe('The key of the issue to confirm'),
  comment: z
    .string()
    .optional()
    .describe('Optional comment explaining why this issue is confirmed'),
};

/**
 * Schema for unconfirm issue tool
 */
export const unconfirmIssueToolSchema = {
  issue_key: z.string().describe('The key of the issue to unconfirm'),
  comment: z
    .string()
    .optional()
    .describe('Optional comment explaining why this issue needs further investigation'),
};

/**
 * Schema for resolve issue tool
 */
export const resolveIssueToolSchema = {
  issue_key: z.string().describe('The key of the issue to resolve'),
  comment: z.string().optional().describe('Optional comment explaining how the issue was resolved'),
};

/**
 * Schema for reopen issue tool
 */
export const reopenIssueToolSchema = {
  issue_key: z.string().describe('The key of the issue to reopen'),
  comment: z
    .string()
    .optional()
    .describe('Optional comment explaining why the issue is being reopened'),
};

/**
 * Schema for issues tool
 */
export const issuesToolSchema = {
  // Component filters (backward compatible)
  project_key: z.string().optional().describe('Single project key for backward compatibility'), // Made optional to support projects array
  projects: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe('Filter by project keys'),
  component_keys: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe(
      'Filter by component keys (file paths, directories, or modules). Use this to filter issues by specific files or folders'
    ),
  components: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe('Alias for component_keys - filter by file paths, directories, or modules'),
  on_component_only: z
    .union([z.boolean(), z.string().transform((val) => val === 'true')])
    .nullable()
    .optional()
    .describe('Return only issues on the specified components, not on their sub-components'),
  directories: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe('Filter by directory paths'),
  files: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe('Filter by specific file paths'),
  scopes: z
    .union([z.array(z.enum(['MAIN', 'TEST', 'OVERALL'])), z.string()])
    .transform((val) => {
      const parsed = parseJsonStringArray(val);
      // Validate that all values are valid scopes
      if (parsed && Array.isArray(parsed)) {
        return parsed.filter((v) => ['MAIN', 'TEST', 'OVERALL'].includes(v));
      }
      return parsed;
    })
    .nullable()
    .optional()
    .describe('Filter by issue scopes (MAIN, TEST, OVERALL)'),

  // Branch and PR support
  branch: z.string().nullable().optional(),
  pull_request: pullRequestNullableSchema,

  // Issue filters
  issues: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),
  severity: severitySchema, // Deprecated single value
  severities: severitiesSchema, // New array support
  statuses: statusSchema,
  resolutions: resolutionSchema,
  resolved: z
    .union([z.boolean(), z.string().transform((val) => val === 'true')])
    .nullable()
    .optional(),
  types: typeSchema,

  // Clean Code taxonomy (SonarQube 10.x+)
  clean_code_attribute_categories: cleanCodeAttributeCategoriesSchema,
  impact_severities: impactSeveritiesSchema,
  impact_software_qualities: impactSoftwareQualitiesSchema,
  issue_statuses: statusSchema, // New issue status values

  // Rules and tags
  rules: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe('Filter by rule keys'),
  tags: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe(
      'Filter by issue tags. Essential for security audits, regression testing, and categorized analysis'
    ),

  // Date filters
  created_after: z.string().nullable().optional(),
  created_before: z.string().nullable().optional(),
  created_at: z.string().nullable().optional(),
  created_in_last: z.string().nullable().optional(),

  // Assignment
  assigned: z
    .union([z.boolean(), z.string().transform((val) => val === 'true')])
    .nullable()
    .optional()
    .describe('Filter to only assigned (true) or unassigned (false) issues'),
  assignees: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe(
      'Filter by assignee logins. Critical for targeted clean-up sprints and workload analysis'
    ),
  author: z.string().nullable().optional().describe('Filter by single issue author'), // Single author
  authors: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe('Filter by multiple issue authors'), // Multiple authors

  // Security standards
  cwe: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),
  owasp_top10: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),
  owasp_top10_v2021: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(), // New 2021 version
  sans_top25: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),
  sonarsource_security: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),
  sonarsource_security_category: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),

  // Languages
  languages: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),

  // Facets
  facets: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional()
    .describe(
      'Enable faceted search for aggregations. Critical for dashboards. Available facets: severities, statuses, resolutions, rules, tags, types, authors, assignees, languages, etc.'
    ),
  facet_mode: z
    .enum(['effort', 'count'])
    .nullable()
    .optional()
    .describe(
      'Mode for facet computation: count (number of issues) or effort (remediation effort)'
    ),

  // New code
  since_leak_period: z
    .union([z.boolean(), z.string().transform((val) => val === 'true')])
    .nullable()
    .optional(),
  in_new_code_period: z
    .union([z.boolean(), z.string().transform((val) => val === 'true')])
    .nullable()
    .optional(),

  // Sorting
  s: z.string().nullable().optional(), // Sort field
  asc: z
    .union([z.boolean(), z.string().transform((val) => val === 'true')])
    .nullable()
    .optional(), // Sort direction

  // Response optimization
  additional_fields: z
    .union([z.array(z.string()), z.string()])
    .transform(parseJsonStringArray)
    .nullable()
    .optional(),

  // Pagination
  page: z.string().optional().transform(stringToNumberTransform),
  page_size: z.string().optional().transform(stringToNumberTransform),
};