cmd

type Constraint struct {
	SHAModule                 *SHAModule                 `yaml:"securityHealthAnalyticsModule,omitempty"`
	CustomSHAModule           *CustomSHAModule           `yaml:"securityHealthAnalyticsCustomModule,omitempty"`
	OrgPolicyConstraint       *OrgPolicyConstraint       `yaml:"orgPolicyConstraint,omitempty"`
	CustomOrgPolicyConstraint *CustomOrgPolicyConstraint `yaml:"orgPolicyConstraintCustom,omitempty"`
}

type CustomOrgPolicyConstraint struct {
	CustomConstraint struct {
		Name          string   `yaml:"name"`
		DisplayName   string   `yaml:"displayName"`
		Description   string   `yaml:"description"`
		ActionType    string   `yaml:"actionType"`
		Condition     string   `yaml:"condition"`
		MethodTypes   []string `yaml:"methodTypes"`
		ResourceTypes []string `yaml:"resourceTypes"`
	} `yaml:"customConstraint"`
	PolicyRules []struct {
		Enforce   bool `yaml:"enforce"`
		Condition *struct {
			Description string `yaml:"description"`
			Expression  string `yaml:"expression"`
			Title       string `yaml:"title"`
		} `yaml:"condition,omitempty"`
	} `yaml:"policyRules"`
}

type CustomSHAModule struct {
	DisplayName string `yaml:"displayName"`
	Config      struct {
		Predicate struct {
			Expression string `yaml:"expression"`
		} `yaml:"predicate"`
		// Corrected (Matches API doc)
		CustomOutput *struct {
			Properties []struct {
				Name            string `yaml:"name"`
				ValueExpression struct {
					Expression string `yaml:"expression"`
				} `yaml:"valueExpression"`
			} `yaml:"properties"` // Note: YAML tag remains the same
		} `yaml:"customOutput,omitempty"`
		ResourceSelector *struct {
			ResourceTypes []string `yaml:"resourceTypes"`
		} `yaml:"resourceSelector"`
		Severity       string `yaml:"severity"`
		Description    string `yaml:"description"`
		Recommendation string `yaml:"recommendation,omitempty"`
	} `yaml:"config"`
	ModuleEnablementState string `yaml:"moduleEnablementState"`
}

type Metadata struct {
	PolicyID            string   `yaml:"policyId"`
	PolicyFileName      string   `yaml:"-"`
	Description         string   `yaml:"description"`
	Postures            []string `yaml:"postures"`
	Location            string   `yaml:"location"`
	Author              string   `yaml:"author"`
	PolicyType          string   `yaml:"policyType"`
	Implements          string   `yaml:"implements"`
	ComplianceStandards []struct {
		Standard string `yaml:"standard"`
		Control  string `yaml:"control"`
	} `yaml:"complianceStandards"`
}

type OrgPolicyConstraint struct {
	CannedConstraintID string `yaml:"cannedConstraintId"`
	PolicyRules        []struct {
		DenyAll    bool `yaml:"denyAll"`
		AllowAll   bool `yaml:"allowAll"`
		ListPolicy *struct {
			AllowedValues     []string `yaml:"allowedValues"`
			DeniedValues      []string `yaml:"deniedValues"`
			InheritFromParent bool     `yaml:"inheritFromParent"`
			SuggestedValue    string   `yaml:"suggestedValue"`
		} `yaml:"listPolicy,omitempty"`
		Condition *struct {
			Description string `yaml:"description"`
			Expression  string `yaml:"expression"`
			Title       string `yaml:"title"`
		} `yaml:"condition,omitempty"`
		Parameters *struct {
			Fields []struct {
				Key   string `yaml:"key"`
				Value *struct {
					BoolValue   bool   `yaml:"bool_value,omitempty"`
					NullValue   string `yaml:"null_value,omitempty"`
					StringValue string `yaml:"string_value,omitempty"`
				} `yaml:"value,omitempty"`
			} `yaml:"fields,omitempty"`
		} `yaml:"parameters,omitempty"`
		ResourceTypes *struct {
			Included string `yaml:"included"`
		} `yaml:"resource_types,omitempty"`
	} `yaml:"policyRules"`
}

type Policy struct {
	PolicyID     string      `yaml:"policy_id"`
	MetadataName string      `yaml:"-"` // Ignore during YAML parsing
	Constraint   *Constraint `yaml:"constraint"`
}

type PolicyJson struct {
	PolicyID    string `json:"policy_id"`
	Description string `json:"description"`
}

type PostureData struct {
	Name     string       `json:"name"`
	Policies []PolicyJson `json:"policies"`
	Posture  string       `json:"posture"`
}

type SHAModule struct {
	ModuleName            string `yaml:"moduleName"`
	ModuleEnablementState string `yaml:"moduleEnablementState"`
}

type TemplateData struct {
	Name                string
	NameWithHyphens     string
	Description         string
	Parent              string
	Target              string
	RandomIdForResource string
}