ml

type AdaptiveResult struct {
	Ecosystem            string                 `json:"ecosystem"`
	OriginalScore        float64                `json:"original_score"`
	AdjustedScore        float64                `json:"adjusted_score"`
	ThresholdUsed        float64                `json:"threshold_used"`
	ModelVersion         string                 `json:"model_version"`
	ConfidenceLevel      float64                `json:"confidence_level"`
	FeatureContributions map[string]float64     `json:"feature_contributions"`
	AdaptationApplied    bool                   `json:"adaptation_applied"`
	Details              map[string]interface{} `json:"details"`
}

type AdaptiveThresholdManager struct {
	// contains filtered or unexported fields
}

type AdvancedAnomalyDetector struct {
	// contains filtered or unexported fields
}

type AdvancedFeatureExtractor struct {
	// contains filtered or unexported fields
}

type AdvancedScorer struct {
	// contains filtered or unexported fields
}

type AggregatedFeatureStoreStats struct {
	TotalEntries     int                           `json:"total_entries"`
	TotalCacheHits   int64                         `json:"total_cache_hits"`
	TotalCacheMisses int64                         `json:"total_cache_misses"`
	OverallHitRatio  float64                       `json:"overall_hit_ratio"`
	TotalMemoryUsage int64                         `json:"total_memory_usage_bytes"`
	RegistryStats    map[string]*FeatureStoreStats `json:"registry_stats"`
}

type AlertSeverity string

type AnalysisMetadata struct {
	AnalysisTime    time.Time `json:"analysis_time"`
	Duration        string    `json:"duration"`
	ModelsUsed      []string  `json:"models_used"`
	FeaturesUsed    []string  `json:"features_used"`
	DatasetVersion  string    `json:"dataset_version"`
	ModelVersion    string    `json:"model_version"`
	AnalysisVersion string    `json:"analysis_version"`
}

type AnalysisOptions struct {
	CheckSimilarity     bool    `json:"check_similarity"`
	CheckMalicious      bool    `json:"check_malicious"`
	SimilarityThreshold float64 `json:"similarity_threshold"`
	MaliciousThreshold  float64 `json:"malicious_threshold"`
	TopK                int     `json:"top_k"`
}

type AnalysisResult struct {
	SimilarityScore    float64            `json:"similarity_score"`
	MaliciousScore     float64            `json:"malicious_score"`
	ReputationScore    float64            `json:"reputation_score"`
	TyposquattingScore float64            `json:"typosquatting_score"`
	Features           map[string]float64 `json:"features"`
	Predictions        []Prediction       `json:"predictions"`
	SimilarPackages    []SimilarPackage   `json:"similar_packages"`
	AnomalyDetection   AnomalyDetection   `json:"anomaly_detection"`
	ReputationAnalysis ReputationAnalysis `json:"reputation_analysis"`
	BehavioralAnalysis BehavioralAnalysis `json:"behavioral_analysis"`
	RiskAssessment     RiskAssessment     `json:"risk_assessment"`
	Findings           []Finding          `json:"findings"`
	Recommendations    []string           `json:"recommendations"`
	Metadata           AnalysisMetadata   `json:"metadata"`
}

type AnomalyDetection struct {
	IsAnomaly       bool            `json:"is_anomaly"`
	AnomalyScore    float64         `json:"anomaly_score"`
	AnomalyType     string          `json:"anomaly_type"`
	AnomalyFeatures []string        `json:"anomaly_features"`
	OutlierAnalysis OutlierAnalysis `json:"outlier_analysis"`
	PatternAnalysis PatternAnalysis `json:"pattern_analysis"`
}

type AnomalyDetector struct{}

type AnomalyModel struct {
	// contains filtered or unexported fields
}

type AuthorFeatures struct {
	AuthorName         string    `json:"author_name"`
	AuthorEmail        string    `json:"author_email"`
	AccountAge         int       `json:"account_age_days"`
	PublishedPackages  int       `json:"published_packages"`
	TotalDownloads     int64     `json:"total_downloads"`
	AverageRating      float64   `json:"average_rating"`
	VerifiedAccount    bool      `json:"verified_account"`
	HasGitHubProfile   bool      `json:"has_github_profile"`
	GitHubFollowers    int       `json:"github_followers"`
	GitHubRepos        int       `json:"github_repos"`
	LastActivity       time.Time `json:"last_activity"`
	SuspiciousPatterns []string  `json:"suspicious_patterns"`
}

type BasicFeatureExtractor struct {
	// contains filtered or unexported fields
}

type BasicMLScorer struct {
	// contains filtered or unexported fields
}

type BasicPackageFeatures struct {
	DownloadCount           float64
	MaintainerReputation    float64
	PackageAge              float64
	VersionCount            float64
	DescriptionLength       float64
	DependencyCount         float64
	TyposquattingSimilarity float64
	NameEntropy             float64
	UpdateFrequency         float64
	LicensePresent          float64
	ReadmePresent           float64
	HomepagePresent         float64
	RepositoryPresent       float64
	KeywordCount            float64
	MaintainerCount         float64
}

type BasicScorerConfig struct {
	MaliciousThreshold   float64
	SuspiciousThreshold  float64
	MinConfidence        float64
	FeatureWeights       map[string]float64
	NormalizationEnabled bool
}

type BatchAnalysisRequest struct {
	Packages []PackageToAnalyze `json:"packages"`
	Options  AnalysisOptions    `json:"options"`
}

type BatchAnalysisResponse struct {
	Results []PackageAnalysisResult `json:"results"`
	Time    float64                 `json:"time_ms"`
}

type BehavioralAnalysis struct {
	InstallBehavior    InstallBehavior    `json:"install_behavior"`
	RuntimeBehavior    RuntimeBehavior    `json:"runtime_behavior"`
	NetworkBehavior    NetworkBehavior    `json:"network_behavior"`
	FileSystemBehavior FileSystemBehavior `json:"filesystem_behavior"`
}

type BehavioralAnalysisResult struct {
	BehaviorScore       float64  `json:"behavior_score"`
	SuspiciousBehaviors []string `json:"suspicious_behaviors"`
	RiskFactors         []string `json:"risk_factors"`
	BehaviorPatterns    []string `json:"behavior_patterns"`
	AnomalousActivities []string `json:"anomalous_activities"`
	Confidence          float64  `json:"confidence"`
}

type BehavioralAnalyzer struct {
	Config BehavioralConfig
}

type BehavioralConfig struct {
	InstallBehaviorWeight    float64
	RuntimeBehaviorWeight    float64
	NetworkBehaviorWeight    float64
	FileSystemBehaviorWeight float64
	SuspiciousScoreThreshold float64
}

type BehavioralMetrics struct {
	InstallationBehavior EnhancedInstallBehavior    `json:"installation_behavior"`
	RuntimeBehavior      EnhancedRuntimeBehavior    `json:"runtime_behavior"`
	NetworkBehavior      EnhancedNetworkBehavior    `json:"network_behavior"`
	FileSystemBehavior   EnhancedFileSystemBehavior `json:"file_system_behavior"`
	ProcessBehavior      EnhancedProcessBehavior    `json:"process_behavior"`
	AnomalyScore         float64                    `json:"anomaly_score"`
}

type CachedFeatures struct {
	Features  *PackageFeatures `json:"features"`
	Timestamp time.Time        `json:"timestamp"`
	TTL       time.Duration    `json:"ttl"`
	HitCount  int              `json:"hit_count"`
	Version   string           `json:"version"`
}

type CachedPrediction struct {
	Result    *ThreatPrediction `json:"result"`
	Timestamp time.Time         `json:"timestamp"`
	HitCount  int               `json:"hit_count"`
}

type Client struct {
	// contains filtered or unexported fields
}

type CodeMetrics struct {
	LinesOfCode          int     `json:"lines_of_code"`
	CyclomaticComplexity float64 `json:"cyclomatic_complexity"`
	CodeDuplication      float64 `json:"code_duplication"`
	TestCoverage         float64 `json:"test_coverage"`
	DocumentationRatio   float64 `json:"documentation_ratio"`
	ObfuscationScore     float64 `json:"obfuscation_score"`
	MinificationScore    float64 `json:"minification_score"`
	CommentRatio         float64 `json:"comment_ratio"`
}

type Config struct {
	Enabled             bool    `mapstructure:"enabled"`
	SimilarityThreshold float64 `mapstructure:"similarity_threshold"`
	MaliciousThreshold  float64 `mapstructure:"malicious_threshold"`
	ReputationThreshold float64 `mapstructure:"reputation_threshold"`
	ModelPath           string  `mapstructure:"model_path"`
	BatchSize           int     `mapstructure:"batch_size"`
	MaxFeatures         int     `mapstructure:"max_features"`
	CacheEmbeddings     bool    `mapstructure:"cache_embeddings"`
	ParallelProcessing  bool    `mapstructure:"parallel_processing"`
	GPUAcceleration     bool    `mapstructure:"gpu_acceleration"`
}

type DecisionNode struct {
	Feature   string  `json:"feature"`
	Threshold float64 `json:"threshold"`
	Direction string  `json:"direction"`
	Impact    float64 `json:"impact"`
}

type Dependency struct {
	Name       string  `json:"name"`
	Version    string  `json:"version"`
	Type       string  `json:"type"`
	Optional   bool    `json:"optional"`
	Dev        bool    `json:"dev"`
	Suspicious bool    `json:"suspicious"`
	RiskScore  float64 `json:"risk_score"`
}

type DetectionMetrics struct {
	TotalDetections   int64     `json:"total_detections"`
	TruePositives     int64     `json:"true_positives"`
	FalsePositives    int64     `json:"false_positives"`
	TrueNegatives     int64     `json:"true_negatives"`
	FalseNegatives    int64     `json:"false_negatives"`
	AverageConfidence float64   `json:"average_confidence"`
	LastUpdated       time.Time `json:"last_updated"`
}

type EcosystemModel struct {
	Ecosystem       string                 `json:"ecosystem"`
	ModelVersion    string                 `json:"model_version"`
	ModelPath       string                 `json:"model_path"`
	FeatureWeights  map[string]float64     `json:"feature_weights"`
	TrainingMetrics TrainingMetrics        `json:"training_metrics"`
	LastTrained     time.Time              `json:"last_trained"`
	IsActive        bool                   `json:"is_active"`
	Metadata        map[string]interface{} `json:"metadata"`
}

type EnhancedAnomalyDetection struct {
	IsAnomalous       bool     `json:"is_anomalous"`
	AnomalyScore      float64  `json:"anomaly_score"`
	AnomalyType       string   `json:"anomaly_type"`
	Confidence        float64  `json:"confidence"`
	AnomalousFeatures []string `json:"anomalous_features"`
	BaselineDeviation float64  `json:"baseline_deviation"`
}

type EnhancedFileSystemBehavior struct {
	FilesCreated        int      `json:"files_created"`
	FilesModified       int      `json:"files_modified"`
	FilesDeleted        int      `json:"files_deleted"`
	SuspiciousLocations []string `json:"suspicious_locations"`
	HiddenFiles         int      `json:"hidden_files"`
	SystemFileAccess    bool     `json:"system_file_access"`
	TempFileUsage       int      `json:"temp_file_usage"`
}

type EnhancedInstallBehavior struct {
	PostInstallScript  bool    `json:"post_install_script"`
	PreInstallScript   bool    `json:"pre_install_script"`
	NetworkActivity    bool    `json:"network_activity"`
	FileModifications  int     `json:"file_modifications"`
	PermissionChanges  int     `json:"permission_changes"`
	SuspiciousCommands int     `json:"suspicious_commands"`
	InstallationTime   float64 `json:"installation_time"`
}

type EnhancedMLConfig struct {
	Enabled             bool               `yaml:"enabled"`
	SimilarityThreshold float64            `yaml:"similarity_threshold"`
	MalwareThreshold    float64            `yaml:"malware_threshold"`
	AnomalyThreshold    float64            `yaml:"anomaly_threshold"`
	TypoThreshold       float64            `yaml:"typo_threshold"`
	ReputationThreshold float64            `yaml:"reputation_threshold"`
	EnsembleWeights     map[string]float64 `yaml:"ensemble_weights"`
	ConfidenceThreshold float64            `yaml:"confidence_threshold"`
	ModelUpdateInterval time.Duration      `yaml:"model_update_interval"`
	BatchSize           int                `yaml:"batch_size"`
	MaxFeatures         int                `yaml:"max_features"`
	ParallelProcessing  bool               `yaml:"parallel_processing"`
	Verbose             bool               `yaml:"verbose"`
}

type EnhancedMLDetector struct {
	// contains filtered or unexported fields
}

type EnhancedMLEngine struct {
	// contains filtered or unexported fields
}

type EnhancedNetworkBehavior struct {
	OutboundConnections int      `json:"outbound_connections"`
	InboundConnections  int      `json:"inbound_connections"`
	SuspiciousHosts     []string `json:"suspicious_hosts"`
	UnusualPorts        []int    `json:"unusual_ports"`
	DataExfiltration    bool     `json:"data_exfiltration"`
	C2Communication     bool     `json:"c2_communication"`
	DNSTunneling        bool     `json:"dns_tunneling"`
}

type EnhancedPackageFeatures struct {
	Name              string                 `json:"name"`
	Registry          string                 `json:"registry"`
	Version           string                 `json:"version"`
	Description       string                 `json:"description"`
	Author            string                 `json:"author"`
	Maintainers       []string               `json:"maintainers"`
	Keywords          []string               `json:"keywords"`
	License           string                 `json:"license"`
	Homepage          string                 `json:"homepage"`
	Repository        string                 `json:"repository"`
	Downloads         int64                  `json:"downloads"`
	Stars             int                    `json:"stars"`
	Forks             int                    `json:"forks"`
	Issues            int                    `json:"issues"`
	CreationDate      time.Time              `json:"creation_date"`
	LastUpdated       time.Time              `json:"last_updated"`
	Dependencies      []Dependency           `json:"dependencies"`
	DevDependencies   []Dependency           `json:"dev_dependencies"`
	Scripts           map[string]string      `json:"scripts"`
	FileStructure     FileStructure          `json:"file_structure"`
	CodeMetrics       CodeMetrics            `json:"code_metrics"`
	SecurityMetrics   SecurityMetrics        `json:"security_metrics"`
	BehavioralMetrics BehavioralMetrics      `json:"behavioral_metrics"`
	Metadata          map[string]interface{} `json:"metadata"`
}

type EnhancedProcessBehavior struct {
	ChildProcesses      int      `json:"child_processes"`
	PrivilegeEscalation bool     `json:"privilege_escalation"`
	CodeInjection       bool     `json:"code_injection"`
	Hollowing           bool     `json:"hollowing"`
	DLLInjection        bool     `json:"dll_injection"`
	SuspiciousCommands  []string `json:"suspicious_commands"`
}

type EnhancedReputationAnalysis struct {
	ReputationScore      float64  `json:"reputation_score"`
	TrustLevel           string   `json:"trust_level"`
	AuthorReputation     float64  `json:"author_reputation"`
	MaintainerReputation float64  `json:"maintainer_reputation"`
	CommunityTrust       float64  `json:"community_trust"`
	HistoricalIssues     []string `json:"historical_issues"`
	VerificationStatus   string   `json:"verification_status"`
}

type EnhancedRuntimeBehavior struct {
	CPUUsage               float64 `json:"cpu_usage"`
	MemoryUsage            float64 `json:"memory_usage"`
	NetworkConnections     int     `json:"network_connections"`
	FileOperations         int     `json:"file_operations"`
	ProcessSpawning        int     `json:"process_spawning"`
	AntiAnalysisTechniques bool    `json:"anti_analysis_techniques"`
	PersistenceMechanisms  bool    `json:"persistence_mechanisms"`
}

type EnhancedSimilarityResult struct {
	SimilarPackage  string  `json:"similar_package"`
	SimilarityScore float64 `json:"similarity_score"`
	SimilarityType  string  `json:"similarity_type"`
	Confidence      float64 `json:"confidence"`
	Reason          string  `json:"reason"`
}

type EnsembleManager struct {
	// contains filtered or unexported fields
}

type EnsembleModel struct{}

type EnsembleResults struct {
	FinalScore          float64            `json:"final_score"`
	ModelScores         map[string]float64 `json:"model_scores"`
	ModelWeights        map[string]float64 `json:"model_weights"`
	ConsensusLevel      float64            `json:"consensus_level"`
	DisagreementFactors []string           `json:"disagreement_factors"`
	Confidence          float64            `json:"confidence"`
}

type FeatureContribution struct {
	FeatureName  string  `json:"feature_name"`
	Contribution float64 `json:"contribution"`
	Value        float64 `json:"value"`
}

type FeatureDefinition struct {
	Name        string      `json:"name"`
	Type        FeatureType `json:"type"`
	Description string      `json:"description"`
	Extractor   func(interface{}) float64
	Weight      float64 `json:"weight"`
	Normalize   bool    `json:"normalize"`
}

type FeatureDriftDetector struct {
	// contains filtered or unexported fields
}

type FeatureExtractor interface {
	ExtractFeatures(pkg *types.Package) (*PackageFeatures, error)
	GetFeatureNames() []string
	NormalizeFeatures(features *PackageFeatures) []float64
}

type FeatureImportance struct {
	Feature    string  `json:"feature"`
	Importance float64 `json:"importance"`
	Value      float64 `json:"value"`
	Impact     string  `json:"impact"` // positive, negative, neutral
}

type FeatureStats struct {
	Mean   float64
	StdDev float64
	Min    float64
	Max    float64
}

type FeatureStore struct {
	// contains filtered or unexported fields
}

type FeatureStoreManager struct {
	// contains filtered or unexported fields
}

type FeatureStoreStats struct {
	TotalEntries int           `json:"total_entries"`
	CacheHits    int64         `json:"cache_hits"`
	CacheMisses  int64         `json:"cache_misses"`
	HitRatio     float64       `json:"hit_ratio"`
	AverageAge   time.Duration `json:"average_age"`
	OldestEntry  time.Time     `json:"oldest_entry"`
	NewestEntry  time.Time     `json:"newest_entry"`
	MemoryUsage  int64         `json:"memory_usage_bytes"`
	CleanupCount int64         `json:"cleanup_count"`
	LastCleanup  time.Time     `json:"last_cleanup"`
}

type FeatureType string

type FeatureVector struct {
	PackageID   string             `json:"package_id"`
	Features    map[string]float64 `json:"features"`
	ExtractedAt time.Time          `json:"extracted_at"`
	Version     string             `json:"version"`
}

type FeedbackData struct {
	FeedbackEntries []FeedbackEntry `json:"feedback_entries"`
	LastUpdated     time.Time       `json:"last_updated"`
	TotalEntries    int             `json:"total_entries"`
	FalsePositives  int             `json:"false_positives"`
	FalseNegatives  int             `json:"false_negatives"`
	TruePositives   int             `json:"true_positives"`
	TrueNegatives   int             `json:"true_negatives"`
}

type FeedbackEntry struct {
	PackageName     string             `json:"package_name"`
	Version         string             `json:"version"`
	PredictedLabel  string             `json:"predicted_label"`
	ActualLabel     string             `json:"actual_label"`
	ConfidenceScore float64            `json:"confidence_score"`
	FeedbackSource  string             `json:"feedback_source"`
	Timestamp       time.Time          `json:"timestamp"`
	Features        map[string]float64 `json:"features,omitempty"`
}

type FeedbackLoop struct {
	FeedbackDir     string
	FeedbackFile    string
	FeedbackData    FeedbackData
	LearningEnabled bool
	// contains filtered or unexported fields
}

type FileStructure struct {
	TotalFiles         int      `json:"total_files"`
	JavaScriptFiles    int      `json:"javascript_files"`
	TypeScriptFiles    int      `json:"typescript_files"`
	ConfigFiles        int      `json:"config_files"`
	TestFiles          int      `json:"test_files"`
	DocumentationFiles int      `json:"documentation_files"`
	BinaryFiles        int      `json:"binary_files"`
	HiddenFiles        int      `json:"hidden_files"`
	SuspiciousFiles    []string `json:"suspicious_files"`
	LargeFiles         []string `json:"large_files"`
	UnusualExtensions  []string `json:"unusual_extensions"`
}

type FileSystemBehavior struct {
	FileCreation     []string `json:"file_creation"`
	FileDeletion     []string `json:"file_deletion"`
	FileModification []string `json:"file_modification"`
	DirectoryAccess  []string `json:"directory_access"`
}

type Finding struct {
	Type        string  `json:"type"`
	Severity    string  `json:"severity"`
	Confidence  float64 `json:"confidence"`
	Description string  `json:"description"`
	Evidence    string  `json:"evidence"`
	Model       string  `json:"model"`
}

type InstallBehavior struct {
	SuspiciousCommands []string `json:"suspicious_commands"`
	NetworkRequests    []string `json:"network_requests"`
	FileModifications  []string `json:"file_modifications"`
	PermissionChanges  []string `json:"permission_changes"`
}

type MLAnalysisResult struct {
	PackageName       string    `json:"package_name"`
	Registry          string    `json:"registry"`
	AnalysisTimestamp time.Time `json:"analysis_timestamp"`

	// Feature analysis
	Features PackageFeatures `json:"features"`

	// Similarity analysis
	SimilarPackages   []SimilarPackage `json:"similar_packages"`
	TyposquattingRisk float64          `json:"typosquatting_risk"`

	// Malicious detection
	MaliciousScore      float64              `json:"malicious_score"`
	MaliciousIndicators []MaliciousIndicator `json:"malicious_indicators"`

	// Reputation scoring
	ReputationScore   float64            `json:"reputation_score"`
	ReputationFactors []ReputationFactor `json:"reputation_factors"`

	// Overall assessment
	RiskLevel      string   `json:"risk_level"`
	Confidence     float64  `json:"confidence"`
	Recommendation string   `json:"recommendation"`
	Warnings       []string `json:"warnings"`

	// Model information
	ModelVersions  map[string]string `json:"model_versions"`
	ProcessingTime time.Duration     `json:"processing_time"`
}

type MLAnalyzer struct {
	Config             config.MLAnalysisConfig
	Client             *Client
	BehavioralAnalyzer *BehavioralAnalyzer
	FeedbackLoop       *FeedbackLoop
}

type MLConfig struct {
	EmbeddingModel      string  `yaml:"embedding_model"`
	MaliciousModel      string  `yaml:"malicious_model"`
	ReputationModel     string  `yaml:"reputation_model"`
	SimilarityThreshold float64 `yaml:"similarity_threshold"`
	MaliciousThreshold  float64 `yaml:"malicious_threshold"`
	ReputationThreshold float64 `yaml:"reputation_threshold"`
	FeatureStoreEnabled bool    `yaml:"feature_store_enabled"`
	FeatureStoreTTL     string  `yaml:"feature_store_ttl"`
	BatchSize           int     `yaml:"batch_size"`
	MaxRetries          int     `yaml:"max_retries"`
	Timeout             string  `yaml:"timeout"`
}

type MLDetectionResult struct {
	PackageName           string                     `json:"package_name"`
	Registry              string                     `json:"registry"`
	AnalysisTimestamp     time.Time                  `json:"analysis_timestamp"`
	OverallRiskScore      float64                    `json:"overall_risk_score"`
	ConfidenceScore       float64                    `json:"confidence_score"`
	ThreatLevel           string                     `json:"threat_level"`
	IsMalicious           bool                       `json:"is_malicious"`
	IsTyposquatting       bool                       `json:"is_typosquatting"`
	IsAnomalous           bool                       `json:"is_anomalous"`
	SimilarityResults     []EnhancedSimilarityResult `json:"similarity_results"`
	MalwareClassification MalwareClassification      `json:"malware_classification"`
	AnomalyDetection      EnhancedAnomalyDetection   `json:"anomaly_detection"`
	TypoDetection         TypoDetection              `json:"typo_detection"`
	ReputationAnalysis    EnhancedReputationAnalysis `json:"reputation_analysis"`
	BehavioralAnalysis    BehavioralAnalysisResult   `json:"behavioral_analysis"`
	EnsembleResults       EnsembleResults            `json:"ensemble_results"`
	Recommendations       []string                   `json:"recommendations"`
	Metadata              map[string]interface{}     `json:"metadata"`
}

type MLModel interface {
	Predict(features []float64) (*Prediction, error)
	Train(data []TrainingData) error
	GetModelInfo() *ModelInfo
	IsReady() bool
}

type MLPipeline struct {
	// contains filtered or unexported fields
}

type MLScore struct {
	MaliciousScore      float64
	Confidence          float64
	RiskLevel           string
	Features            BasicPackageFeatures
	ContributingFactors []string
	Recommendation      string
	Timestamp           time.Time
}

type MaliciousIndicator struct {
	Type        string  `json:"type"`
	Description string  `json:"description"`
	Severity    string  `json:"severity"`
	Confidence  float64 `json:"confidence"`
	Evidence    string  `json:"evidence"`
}

type MaliciousRequest struct {
	PackageName string                 `json:"package_name"`
	Registry    string                 `json:"registry"`
	Version     string                 `json:"version,omitempty"`
	Features    map[string]interface{} `json:"features,omitempty"`
}

type MaliciousResponse struct {
	IsMalicious bool                   `json:"is_malicious"`
	Score       float64                `json:"score"`
	Confidence  float64                `json:"confidence"`
	Reasons     []string               `json:"reasons"`
	Features    map[string]interface{} `json:"features"`
	Model       string                 `json:"model"`
	Time        float64                `json:"time_ms"`
}

type MalwareClassification struct {
	IsMalware            bool               `json:"is_malware"`
	MalwareType          string             `json:"malware_type"`
	MalwareFamily        string             `json:"malware_family"`
	Confidence           float64            `json:"confidence"`
	FeatureImportance    map[string]float64 `json:"feature_importance"`
	ClassificationReason string             `json:"classification_reason"`
}

type MalwareClassifier struct{}

type MetadataFeatures struct {
	HasDescription     bool    `json:"has_description"`
	DescriptionLength  int     `json:"description_length"`
	DescriptionQuality float64 `json:"description_quality"`
	HasHomepage        bool    `json:"has_homepage"`
	HasRepository      bool    `json:"has_repository"`
	HasLicense         bool    `json:"has_license"`
	LicenseType        string  `json:"license_type"`
	HasKeywords        bool    `json:"has_keywords"`
	KeywordCount       int     `json:"keyword_count"`
	VersionCount       int     `json:"version_count"`
	LatestVersion      string  `json:"latest_version"`
	VersionPattern     string  `json:"version_pattern"`
	UnusualVersionJump bool    `json:"unusual_version_jump"`
	PublicationRecency int     `json:"publication_recency_days"`
}

type ModelExplainer struct {
	// contains filtered or unexported fields
}

type ModelInfo struct {
	Name         string                 `json:"name"`
	Version      string                 `json:"version"`
	Description  string                 `json:"description"`
	Type         string                 `json:"type"`
	TrainedAt    time.Time              `json:"trained_at"`
	Accuracy     float64                `json:"accuracy"`
	Precision    float64                `json:"precision"`
	Recall       float64                `json:"recall"`
	F1Score      float64                `json:"f1_score"`
	FeatureCount int                    `json:"feature_count"`
	Metadata     map[string]interface{} `json:"metadata"`
}

type ModelMetrics struct {
	Accuracy  float64     `json:"accuracy"`
	Precision float64     `json:"precision"`
	Recall    float64     `json:"recall"`
	F1Score   float64     `json:"f1_score"`
	Status    ModelStatus `json:"status"`
	TrainedAt time.Time   `json:"trained_at"`
}

type ModelPerformanceMonitor struct {
	// contains filtered or unexported fields
}

type ModelStatus string

type ModelType string

type NetworkBehavior struct {
	OutboundConnections []string `json:"outbound_connections"`
	DNSQueries          []string `json:"dns_queries"`
	DataExfiltration    []string `json:"data_exfiltration"`
	C2Communication     []string `json:"c2_communication"`
}

type NormalizationParams struct {
	Means   []float64 `json:"means"`
	StdDevs []float64 `json:"std_devs"`
	Mins    []float64 `json:"mins"`
	Maxs    []float64 `json:"maxs"`
}

type OutlierAnalysis struct {
	IsOutlier       bool    `json:"is_outlier"`
	OutlierScore    float64 `json:"outlier_score"`
	IsolationForest float64 `json:"isolation_forest"`
	LocalOutlier    float64 `json:"local_outlier"`
	OneClassSVM     float64 `json:"one_class_svm"`
}

type PackageAccessInfo struct {
	PackageName string        `json:"package_name"`
	Registry    string        `json:"registry"`
	HitCount    int           `json:"hit_count"`
	LastAccess  time.Time     `json:"last_access"`
	CacheAge    time.Duration `json:"cache_age"`
}

type PackageAnalysisResult struct {
	Package        PackageToAnalyze   `json:"package"`
	Similarities   []SimilarityResult `json:"similarities,omitempty"`
	MaliciousCheck *MaliciousResponse `json:"malicious_check,omitempty"`
	Threats        []types.Threat     `json:"threats,omitempty"`
	Errors         []string           `json:"errors,omitempty"`
}

type PackageFeatures struct {
	PackageName string `json:"package_name"`
	Registry    string `json:"registry"`
	Version     string `json:"version,omitempty"`
	PackageType string `json:"package_type,omitempty"`

	// Name-based features
	NameEmbedding  []float64 `json:"name_embedding"`
	NameLength     int       `json:"name_length"`
	NameComplexity float64   `json:"name_complexity"`
	NameEntropy    float64   `json:"name_entropy"`

	// Direct access fields for compatibility
	VersionComplexity  float64 `json:"version_complexity"`
	DescriptionLength  int     `json:"description_length"`
	DependencyCount    int     `json:"dependency_count"`
	DownloadCount      int64   `json:"download_count"`
	StarCount          int     `json:"star_count"`
	ForkCount          int     `json:"fork_count"`
	ContributorCount   int     `json:"contributor_count"`
	AgeInDays          int     `json:"age_in_days"`
	TyposquattingScore float64 `json:"typosquatting_score"`
	SuspiciousKeywords int     `json:"suspicious_keywords"`
	VersionSpoofing    float64 `json:"version_spoofing"`
	DomainReputation   float64 `json:"domain_reputation"`
	UpdateFrequency    float64 `json:"update_frequency"`
	MaintainerCount    int     `json:"maintainer_count"`
	IssueCount         int     `json:"issue_count"`
	LicenseScore       float64 `json:"license_score"`

	// Author features
	AuthorFeatures AuthorFeatures `json:"author_features"`

	// Package metadata features
	MetadataFeatures MetadataFeatures `json:"metadata_features"`

	// Repository features
	RepositoryFeatures RepositoryFeatures `json:"repository_features"`

	// Download and popularity features
	PopularityFeatures PopularityFeatures `json:"popularity_features"`

	// Security features
	SecurityFeatures SecurityFeatures `json:"security_features"`

	// Computed scores
	MaliciousScore  float64 `json:"malicious_score"`
	ReputationScore float64 `json:"reputation_score"`
	OverallRisk     float64 `json:"overall_risk"`

	// Metadata
	Timestamp      time.Time         `json:"timestamp"`
	ModelVersions  map[string]string `json:"model_versions"`
	FeatureVersion string            `json:"feature_version"`
}

type PackageToAnalyze struct {
	Name     string `json:"name"`
	Registry string `json:"registry"`
	Version  string `json:"version,omitempty"`
}

type PatternAnalysis struct {
	UnusualPatterns []string `json:"unusual_patterns"`
	PatternScore    float64  `json:"pattern_score"`
	FrequencyScore  float64  `json:"frequency_score"`
	SequenceScore   float64  `json:"sequence_score"`
}

type PerformanceAlert struct {
	ModelID   string        `json:"model_id"`
	Metric    string        `json:"metric"`
	Value     float64       `json:"value"`
	Threshold float64       `json:"threshold"`
	Timestamp time.Time     `json:"timestamp"`
	Severity  AlertSeverity `json:"severity"`
}

type PerformanceMetrics struct {
	Accuracy   float64       `json:"accuracy"`
	Precision  float64       `json:"precision"`
	Recall     float64       `json:"recall"`
	F1Score    float64       `json:"f1_score"`
	Latency    time.Duration `json:"latency"`
	Throughput float64       `json:"throughput"`
	Timestamp  time.Time     `json:"timestamp"`
}

type PerformanceStats struct {
	Ecosystem      string    `json:"ecosystem"`
	TruePositives  int       `json:"true_positives"`
	FalsePositives int       `json:"false_positives"`
	TrueNegatives  int       `json:"true_negatives"`
	FalseNegatives int       `json:"false_negatives"`
	Precision      float64   `json:"precision"`
	Recall         float64   `json:"recall"`
	F1Score        float64   `json:"f1_score"`
	Accuracy       float64   `json:"accuracy"`
	LastCalculated time.Time `json:"last_calculated"`
}

type PopularityFeatures struct {
	TotalDownloads    int64   `json:"total_downloads"`
	WeeklyDownloads   int64   `json:"weekly_downloads"`
	MonthlyDownloads  int64   `json:"monthly_downloads"`
	DownloadTrend     string  `json:"download_trend"`
	DownloadVelocity  float64 `json:"download_velocity"`
	PopularityRank    int     `json:"popularity_rank"`
	DependentPackages int     `json:"dependent_packages"`
	DependencyRank    int     `json:"dependency_rank"`
	CommunityScore    float64 `json:"community_score"`
	MaintenanceScore  float64 `json:"maintenance_score"`
}

type Prediction struct {
	Model       string  `json:"model"`
	Probability float64 `json:"probability"`
	Label       string  `json:"label"`
	Confidence  float64 `json:"confidence"`
}

type PredictionCache struct {
	// contains filtered or unexported fields
}

type PredictionExplanation struct {
	TopFeatures       []FeatureContribution `json:"top_features"`
	ShapValues        map[string]float64    `json:"shap_values"`
	DecisionPath      []DecisionNode        `json:"decision_path"`
	ConfidenceFactors []string              `json:"confidence_factors"`
}

type PredictionMetrics struct {
	// contains filtered or unexported fields
}

type PredictionModel struct {
	ID              string                 `json:"id"`
	Name            string                 `json:"name"`
	Type            ModelType              `json:"type"`
	Version         string                 `json:"version"`
	Accuracy        float64                `json:"accuracy"`
	Precision       float64                `json:"precision"`
	Recall          float64                `json:"recall"`
	F1Score         float64                `json:"f1_score"`
	TrainedAt       time.Time              `json:"trained_at"`
	Features        []string               `json:"features"`
	Weights         map[string]float64     `json:"weights"`
	Thresholds      map[string]float64     `json:"thresholds"`
	Hyperparameters map[string]interface{} `json:"hyperparameters"`
	Status          ModelStatus            `json:"status"`
}

type PredictorConfig struct {
	ModelsEnabled      []ModelType        `json:"models_enabled"`
	RetrainingInterval time.Duration      `json:"retraining_interval"`
	MinTrainingSamples int                `json:"min_training_samples"`
	ValidationSplit    float64            `json:"validation_split"`
	FeatureSelection   bool               `json:"feature_selection"`
	EnsembleWeights    map[string]float64 `json:"ensemble_weights"`
	Thresholds         map[string]float64 `json:"thresholds"`
	CacheEnabled       bool               `json:"cache_enabled"`
	CacheTTL           time.Duration      `json:"cache_ttl"`
	MetricsEnabled     bool               `json:"metrics_enabled"`
}

type RealtimeLearner struct {
	// contains filtered or unexported fields
}

type RepositoryFeatures struct {
	HasRepository        bool               `json:"has_repository"`
	RepositoryURL        string             `json:"repository_url"`
	RepositoryType       string             `json:"repository_type"`
	StarCount            int                `json:"star_count"`
	ForkCount            int                `json:"fork_count"`
	IssueCount           int                `json:"issue_count"`
	CommitCount          int                `json:"commit_count"`
	ContributorCount     int                `json:"contributor_count"`
	LastCommit           time.Time          `json:"last_commit"`
	HasReadme            bool               `json:"has_readme"`
	ReadmeLength         int                `json:"readme_length"`
	HasTests             bool               `json:"has_tests"`
	TestCoverage         float64            `json:"test_coverage"`
	HasCI                bool               `json:"has_ci"`
	LanguageDistribution map[string]float64 `json:"language_distribution"`
	SuspiciousFiles      []string           `json:"suspicious_files"`
}

type ReputationAnalysis struct {
	OverallScore      float64            `json:"overall_score"`
	MaintainerScore   float64            `json:"maintainer_score"`
	DownloadScore     float64            `json:"download_score"`
	AgeScore          float64            `json:"age_score"`
	CommunityScore    float64            `json:"community_score"`
	SecurityScore     float64            `json:"security_score"`
	ReputationSources []ReputationSource `json:"reputation_sources"`
}

type ReputationAnalyzer struct{}

type ReputationFactor struct {
	Factor      string  `json:"factor"`
	Score       float64 `json:"score"`
	Weight      float64 `json:"weight"`
	Description string  `json:"description"`
}

type ReputationModel struct {
	// contains filtered or unexported fields
}

type ReputationSource struct {
	Source string  `json:"source"`
	Score  float64 `json:"score"`
	Weight float64 `json:"weight"`
}

type RiskAssessment struct {
	OverallRisk     string       `json:"overall_risk"`
	RiskScore       float64      `json:"risk_score"`
	RiskFactors     []RiskFactor `json:"risk_factors"`
	Mitigations     []string     `json:"mitigations"`
	ConfidenceLevel float64      `json:"confidence_level"`
}

type RiskFactor struct {
	Factor      string  `json:"factor"`
	Severity    string  `json:"severity"`
	Score       float64 `json:"score"`
	Description string  `json:"description"`
}

type RuntimeBehavior struct {
	ProcessSpawning   []string `json:"process_spawning"`
	SystemCalls       []string `json:"system_calls"`
	ResourceUsage     []string `json:"resource_usage"`
	EnvironmentAccess []string `json:"environment_access"`
}

type Scorer interface {
	// Score calculates a risk score for a package based on extracted features
	Score(ctx context.Context, pkg *types.Package, features map[string]interface{}) (*ScoringResult, error)

	// GetModelInfo returns information about the underlying model
	GetModelInfo() *ModelInfo

	// UpdateModel updates the scorer with new model parameters
	UpdateModel(modelData []byte) error

	// GetThresholds returns the scoring thresholds used by this scorer
	GetThresholds() ScoringThresholds
}

type ScorerConfig struct {
	Thresholds     ScoringThresholds      `json:"thresholds"`
	FeatureWeights map[string]float64     `json:"feature_weights"`
	Normalization  bool                   `json:"normalization"`
	ModelPath      string                 `json:"model_path"`
	UpdateInterval string                 `json:"update_interval"`
	Metadata       map[string]interface{} `json:"metadata,omitempty"`
}

type ScoringResult struct {
	Score          float64                `json:"score"`
	Confidence     float64                `json:"confidence"`
	RiskLevel      string                 `json:"risk_level"`
	FeatureScores  map[string]float64     `json:"feature_scores"`
	Explanation    string                 `json:"explanation"`
	ModelVersion   string                 `json:"model_version"`
	ProcessingTime float64                `json:"processing_time_ms"`
	Metadata       map[string]interface{} `json:"metadata,omitempty"`
}

type ScoringThresholds struct {
	Malicious     float64 `json:"malicious"`
	Suspicious    float64 `json:"suspicious"`
	MinConfidence float64 `json:"min_confidence"`
}

type SecurityFeatures struct {
	KnownVulnerabilities int      `json:"known_vulnerabilities"`
	SecurityScore        float64  `json:"security_score"`
	HasSecurityPolicy    bool     `json:"has_security_policy"`
	SignedReleases       bool     `json:"signed_releases"`
	SigstoreVerified     bool     `json:"sigstore_verified"`
	SLSALevel            int      `json:"slsa_level"`
	HasSBOM              bool     `json:"has_sbom"`
	SupplyChainRisk      float64  `json:"supply_chain_risk"`
	MalwareIndicators    []string `json:"malware_indicators"`
	SuspiciousScripts    []string `json:"suspicious_scripts"`
}

type SecurityMetrics struct {
	VulnerabilityCount    int     `json:"vulnerability_count"`
	HighSeverityVulns     int     `json:"high_severity_vulns"`
	CriticalSeverityVulns int     `json:"critical_severity_vulns"`
	SuspiciousPatterns    int     `json:"suspicious_patterns"`
	ObfuscatedCode        bool    `json:"obfuscated_code"`
	NetworkCalls          int     `json:"network_calls"`
	FileSystemAccess      int     `json:"file_system_access"`
	ProcessExecution      int     `json:"process_execution"`
	CryptographicUsage    int     `json:"cryptographic_usage"`
	DangerousFunctions    int     `json:"dangerous_functions"`
	SecurityScore         float64 `json:"security_score"`
}

type SimilarPackage struct {
	Name           string  `json:"name"`
	Similarity     float64 `json:"similarity"`
	Distance       int     `json:"distance"`
	Algorithm      string  `json:"algorithm"`
	Registry       string  `json:"registry"`
	Downloads      int64   `json:"downloads"`
	LastUpdated    string  `json:"last_updated"`
	Maintainer     string  `json:"maintainer"`
	SuspiciousFlag bool    `json:"suspicious_flag"`
}

type SimilarityModel struct{}

type SimilarityRequest struct {
	PackageName string   `json:"package_name"`
	Registry    string   `json:"registry"`
	TopK        int      `json:"top_k"`
	Threshold   float64  `json:"threshold"`
	Exclude     []string `json:"exclude,omitempty"`
}

type SimilarityResponse struct {
	Results []SimilarityResult `json:"results"`
	Model   string             `json:"model"`
	Time    float64            `json:"time_ms"`
}

type SimilarityResult struct {
	PackageName string  `json:"package_name"`
	Registry    string  `json:"registry"`
	Score       float64 `json:"score"`
	Distance    float64 `json:"distance"`
	Rank        int     `json:"rank"`
}

type ThreatFeatureExtractor struct {
	// contains filtered or unexported fields
}

type ThreatPrediction struct {
	PackageID      string                 `json:"package_id"`
	ThreatScore    float64                `json:"threat_score"`
	Confidence     float64                `json:"confidence"`
	ThreatTypes    []string               `json:"threat_types"`
	Explanation    *PredictionExplanation `json:"explanation"`
	ModelVersions  map[string]string      `json:"model_versions"`
	PredictionTime time.Time              `json:"prediction_time"`
	FeatureVector  *FeatureVector         `json:"feature_vector"`
}

type ThreatPredictor struct {
	// contains filtered or unexported fields
}

type ThresholdChange struct {
	ThreatType      string    `json:"threat_type"`
	OldThreshold    float64   `json:"old_threshold"`
	NewThreshold    float64   `json:"new_threshold"`
	Reason          string    `json:"reason"`
	Timestamp       time.Time `json:"timestamp"`
	PerformanceGain float64   `json:"performance_gain"`
}

type ThresholdSet struct {
	Ecosystem              string            `json:"ecosystem"`
	TyposquattingThreshold float64           `json:"typosquatting_threshold"`
	ConfusionThreshold     float64           `json:"confusion_threshold"`
	SupplyChainThreshold   float64           `json:"supply_chain_threshold"`
	MalwareThreshold       float64           `json:"malware_threshold"`
	ConfidenceLevel        float64           `json:"confidence_level"`
	LastUpdated            time.Time         `json:"last_updated"`
	AdaptationHistory      []ThresholdChange `json:"adaptation_history"`
}

type TrainingData struct {
	Features []float64              `json:"features"`
	Label    float64                `json:"label"`
	Metadata map[string]interface{} `json:"metadata"`
}

type TrainingDataset struct {
	// contains filtered or unexported fields
}

type TrainingExample struct {
	Features  *FeatureVector `json:"features"`
	Label     float64        `json:"label"`
	Weight    float64        `json:"weight"`
	Timestamp time.Time      `json:"timestamp"`
}

type TrainingMetrics struct {
	Accuracy       float64       `json:"accuracy"`
	Precision      float64       `json:"precision"`
	Recall         float64       `json:"recall"`
	F1Score        float64       `json:"f1_score"`
	TrainingLoss   float64       `json:"training_loss"`
	ValidationLoss float64       `json:"validation_loss"`
	TrainingTime   time.Duration `json:"training_time"`
	DatasetSize    int           `json:"dataset_size"`
}

type TrainingSample struct {
	ID         string                 `json:"id"`
	Features   map[string]float64     `json:"features"`
	Label      string                 `json:"label"`
	ThreatType types.ThreatType       `json:"threat_type"`
	Severity   types.Severity         `json:"severity"`
	Confidence float64                `json:"confidence"`
	Timestamp  time.Time              `json:"timestamp"`
	Metadata   map[string]interface{} `json:"metadata"`
	Validated  bool                   `json:"validated"`
}

type TypoDetection struct {
	IsTyposquatting    bool     `json:"is_typosquatting"`
	TargetPackage      string   `json:"target_package"`
	TypoType           string   `json:"typo_type"`
	EditDistance       int      `json:"edit_distance"`
	SimilarityScore    float64  `json:"similarity_score"`
	Confidence         float64  `json:"confidence"`
	SuspiciousPatterns []string `json:"suspicious_patterns"`
}

type TypoDetector struct{}

type TyposquattingModel struct {
	// contains filtered or unexported fields
}

type VotingType string