module kubernetes

import incubaid.herolib.core.pathlib

// Parse YAML file and return validation result
pub fn yaml_validate(yaml_path string) !K8sValidationResult {
	mut file := pathlib.get(yaml_path)

	if !file.exists() {
		return error('YAML file not found: ${yaml_path}')
	}

	yaml_content := file.read()!

	// Extract kind and apiVersion from YAML
	lines := yaml_content.split('\n')
	mut kind := ''
	mut api_version := ''
	mut metadata_name := ''
	mut metadata_namespace := 'default'

	for line in lines {
		if line.starts_with('kind:') {
			kind = line.replace('kind:', '').trim_space()
		}
		if line.starts_with('apiVersion:') {
			api_version = line.replace('apiVersion:', '').trim_space()
		}
		if line.contains('name:') && line.trim_space().starts_with('name:') {
			metadata_name = line.replace('name:', '').trim_space()
		}
		if line.contains('namespace:') && line.trim_space().starts_with('namespace:') {
			metadata_namespace = line.replace('namespace:', '').trim_space()
		}
	}

	mut errors := []string{}

	if kind.len == 0 {
		errors << 'Missing "kind" field'
	}
	if api_version.len == 0 {
		errors << 'Missing "apiVersion" field'
	}
	if metadata_name.len == 0 {
		errors << 'Missing metadata.name field'
	}

	// Validate kind values for standard Kubernetes resources
	// Allow custom resources (CRDs) which typically have non-standard apiVersions
	standard_kinds := ['Pod', 'Deployment', 'Service', 'ConfigMap', 'Secret', 'StatefulSet',
		'DaemonSet', 'Job', 'CronJob', 'Ingress', 'PersistentVolume', 'PersistentVolumeClaim',
		'Namespace', 'ServiceAccount', 'Role', 'RoleBinding', 'ClusterRole', 'ClusterRoleBinding']

	// Check if it's a standard Kubernetes resource or a custom resource
	is_standard_api := api_version.starts_with('v1') || api_version.starts_with('apps/')
		|| api_version.starts_with('batch/') || api_version.starts_with('networking.k8s.io/')
		|| api_version.starts_with('rbac.authorization.k8s.io/')

	// Only validate kind for standard Kubernetes resources
	if is_standard_api && kind !in standard_kinds {
		errors << 'Invalid kind: ${kind}. Valid kinds for standard resources: ${standard_kinds.join(', ')}'
	}

	return K8sValidationResult{
		valid:       errors.len == 0
		kind:        kind
		api_version: api_version
		metadata:    K8sMetadata{
			name:      metadata_name
			namespace: metadata_namespace
		}
		errors:      errors
	}
}

// Generate YAML from model
pub fn yaml_from_deployment(spec DeploymentSpec) !string {
	mut yaml := 'apiVersion: apps/v1\nkind: Deployment\nmetadata:\n'
	yaml += '  name: ${spec.metadata.name}\n'
	yaml += '  namespace: ${spec.metadata.namespace}\n'

	if spec.metadata.labels.len > 0 {
		yaml += '  labels:\n'
		for key, value in spec.metadata.labels {
			yaml += '    ${key}: ${value}\n'
		}
	}

	yaml += 'spec:\n'
	yaml += '  replicas: ${spec.replicas}\n'
	yaml += '  selector:\n'
	yaml += '    matchLabels:\n'
	for key, value in spec.selector {
		yaml += '      ${key}: ${value}\n'
	}

	yaml += '  template:\n'
	yaml += '    metadata:\n'
	yaml += '      labels:\n'
	for key, value in spec.selector {
		yaml += '        ${key}: ${value}\n'
	}

	yaml += '    spec:\n'
	yaml += '      containers:\n'
	for container in spec.template.containers {
		yaml += '      - name: ${container.name}\n'
		yaml += '        image: ${container.image}\n'
		yaml += '        imagePullPolicy: ${container.image_pull_policy}\n'

		if container.ports.len > 0 {
			yaml += '        ports:\n'
			for port in container.ports {
				yaml += '        - name: ${port.name}\n'
				yaml += '          containerPort: ${port.container_port}\n'
				yaml += '          protocol: ${port.protocol}\n'
			}
		}

		if container.env.len > 0 {
			yaml += '        env:\n'
			for env in container.env {
				yaml += '        - name: ${env.name}\n'
				yaml += '          value: "${env.value}"\n'
			}
		}
	}

	return yaml
}

pub fn yaml_from_service(spec ServiceSpec) !string {
	mut yaml := 'apiVersion: v1\nkind: Service\nmetadata:\n'
	yaml += '  name: ${spec.metadata.name}\n'
	yaml += '  namespace: ${spec.metadata.namespace}\n'

	yaml += 'spec:\n'
	yaml += '  type: ${spec.service_type}\n'
	yaml += '  selector:\n'
	for key, value in spec.selector {
		yaml += '    ${key}: ${value}\n'
	}

	yaml += '  ports:\n'
	for port in spec.ports {
		yaml += '  - name: ${port.name}\n'
		yaml += '    protocol: ${port.protocol}\n'
		yaml += '    port: ${port.port}\n'
		yaml += '    targetPort: ${port.target_port}\n'
		if port.node_port > 0 {
			yaml += '    nodePort: ${port.node_port}\n'
		}
	}

	return yaml
}

pub fn yaml_from_configmap(spec ConfigMapSpec) !string {
	mut yaml := 'apiVersion: v1\nkind: ConfigMap\nmetadata:\n'
	yaml += '  name: ${spec.metadata.name}\n'
	yaml += '  namespace: ${spec.metadata.namespace}\n'
	yaml += 'data:\n'

	for key, value in spec.data {
		yaml += '  ${key}: |\n'
		// Indent multiline values
		lines := value.split('\n')
		for line in lines {
			yaml += '    ${line}\n'
		}
	}

	return yaml
}