kubiya_scheduled_task

locals {
  environments = {
    dev = {
      schedule = "0 6 * * *"  # 6 AM daily
      prompt   = "Check development environment health"
      enabled  = false  # Disable for development
    }
    staging = {
      schedule = "0 7 * * *"  # 7 AM daily
      prompt   = "Verify staging environment readiness"
      enabled  = true
    }
    prod = {
      schedule = "*/10 * * * *"  # Every 10 minutes
      prompt   = "Monitor production environment health and performance"
      enabled  = true
    }
  }
}

resource "kubiya_agent" "env_monitors" {
  for_each = local.environments
  
  name         = "${each.key}-monitor"
  runner       = "kubiya-hosted"
  description  = "Monitor for ${each.key} environment"
  instructions = "You monitor the ${each.key} environment health and performance."
  
  environment_variables = {
    ENVIRONMENT = each.key
    LOG_LEVEL   = each.key == "prod" ? "INFO" : "DEBUG"
  }
}

resource "kubiya_scheduled_task" "env_tasks" {
  for_each = local.environments
  
  name        = "${each.key}-health-check"
  description = "Health check for ${each.key} environment"
  agent       = kubiya_agent.env_monitors[each.key].name
  schedule    = each.value.schedule
  prompt      = each.value.prompt
  enabled     = each.value.enabled
  
  notification {
    method      = "Slack"
    destination = "#${each.key}-alerts"
  }
}

resource "kubiya_scheduled_task" "conditional_task" {
  name        = "conditional-deployment-check"
  description = "Check deployments based on conditions"
  agent       = "deployment-agent"
  schedule    = "0 */4 * * *"  # Every 4 hours
  prompt      = <<-EOT
    Check if there are pending deployments:
    1. If pending deployments exist, validate them
    2. If validation passes, approve for next window
    3. If no pending deployments, check system health
    4. Generate appropriate report based on findings
  EOT
  enabled     = true
  
  notification {
    method      = "Slack"
    destination = "#deployments"
  }
}

resource "kubiya_scheduled_task" "smart_scaling" {
  name        = "smart-resource-scaling"
  description = "Intelligent resource scaling based on usage patterns"
  agent       = "scaling-agent"
  schedule    = "*/30 * * * *"  # Every 30 minutes
  prompt      = <<-EOT
    Analyze current resource usage and demand patterns:
    1. Check CPU, memory, and network utilization
    2. Review historical usage patterns for this time of day
    3. Predict upcoming demand based on trends
    4. Scale resources up or down as needed
    5. Report scaling actions taken
  EOT
  enabled     = true
  
  notification {
    method      = "Slack"
    destination = "#infrastructure"
  }
}

resource "kubiya_agent" "compliance_agent" {
  name         = "compliance-auditor"
  runner       = "kubiya-hosted"
  description  = "Compliance and security audit agent"
  instructions = <<-EOT
    You perform compliance checks and security audits:
    - Verify access controls and permissions
    - Check for policy violations
    - Generate compliance reports
    - Monitor for security incidents
  EOT
  
  integrations = ["aws", "github", "slack_integration"]
  groups       = ["Security", "Compliance"]
}

# Daily security checks
resource "kubiya_scheduled_task" "daily_security_scan" {
  name        = "daily-security-compliance-check"
  description = "Daily security and compliance verification"
  agent       = kubiya_agent.compliance_agent.name
  schedule    = "0 8 * * *"  # 8 AM daily
  prompt      = "Perform daily security compliance check: access controls, policy adherence, and security posture"
  enabled     = true
  
  notification {
    method      = "Slack"
    destination = "#security-alerts"
  }
}

# Weekly compliance report
resource "kubiya_scheduled_task" "weekly_compliance_report" {
  name        = "weekly-compliance-report"
  description = "Generate weekly compliance report"
  agent       = kubiya_agent.compliance_agent.name
  schedule    = "0 9 * * FRI"  # 9 AM every Friday
  prompt      = "Generate comprehensive weekly compliance report with findings and recommendations"
  enabled     = true
  
  notification {
    method      = "Email"
    destination = "compliance@example.com"
  }
}

# Monthly audit
resource "kubiya_scheduled_task" "monthly_audit" {
  name        = "monthly-security-audit"
  description = "Monthly comprehensive security audit"
  agent       = kubiya_agent.compliance_agent.name
  schedule    = "0 10 1 * *"  # 10 AM on 1st of each month
  prompt      = "Conduct comprehensive monthly security audit including penetration testing and vulnerability assessment"
  enabled     = true
  
  notification {
    method      = "Teams"
    team_name   = "Security Team"
    destination = "Audit Reports"
  }
}