//go:build ignore

package main

import (
	"bytes"
	"encoding/json"
	"fmt"
	"go/format"
	"io"
	"net/http"
	"os"
	"strings"

	"github.com/aws-cloudformation/rain/cft/spec"

	"gopkg.in/yaml.v3"
)

const (
	cfnSpecURL = "https://d1uauaxba7bl26.cloudfront.net/latest/gzip/CloudFormationResourceSpecification.json"
	cfnSpecFn  = "internal/CloudFormationResourceSpecification.json"
	iamSpecFn  = "internal/IamSpecification.json"
	samSpecFn  = "internal/SamSpecification.yaml"
)

func load(r io.Reader, s *spec.Spec) {
	var intermediate map[string]interface{}
	yamlDecoder := yaml.NewDecoder(r)
	err := yamlDecoder.Decode(&intermediate)
	if err != nil {
		panic(err)
	}

	inJSON, err := json.Marshal(intermediate)
	if err != nil {
		panic(err)
	}

	jsonReader := bytes.NewReader(inJSON)

	decoder := json.NewDecoder(jsonReader)
	decoder.DisallowUnknownFields()
	err = decoder.Decode(&s)
	if err != nil {
		panic(err)
	}
}

func loadURL(url string) spec.Spec {
	var s spec.Spec

	resp, err := http.Get(url)
	if err != nil {
		panic(err)
	}

	defer resp.Body.Close()
	load(resp.Body, &s)

	return s
}

func loadFile(fn string) spec.Spec {
	var s spec.Spec

	f, err := os.Open(fn)
	if err != nil {
		panic(err)
	}

	defer f.Close()
	load(f, &s)

	return s
}

func saveSpec(s spec.Spec, name string) {
	// Write out the file
	source := fmt.Sprintf(`package spec

// %s is generated from the specification file
var %s = %s`, name, name, s)

	out, err := format.Source([]byte(source))
	if err != nil {
		panic(err)
	}

	err = os.WriteFile(strings.ToLower(name)+".go", out, 0644)
	if err != nil {
		panic(err)
	}
}

func saveJSON(s spec.Spec, name string) {
	data, err := json.MarshalIndent(s, "", "    ")
	if err != nil {
		panic(err)
	}

	err = os.WriteFile(name, data, 0644)
	if err != nil {
		panic(err)
	}
}

func patchCfnSpec(s spec.Spec) {
	for _, r := range s.ResourceTypes {
		for _, p := range r.Properties {
			if p.Type == "Json" {
				p.Type = ""
				p.PrimitiveType = "Json"
			} else if p.ItemType == "Json" {
				p.PrimitiveItemType = "Json"
			}
		}
	}

	for _, pt := range s.PropertyTypes {
		for _, p := range pt.Properties {
			if p.Type == "Json" {
				p.Type = ""
				p.PrimitiveType = "Json"
			} else if p.ItemType == "Json" {
				p.PrimitiveItemType = "Json"
			}
		}
	}

	s.ResourceTypes["AWS::Rekognition::StreamProcessor"].Properties["PolygonRegionsOfInterest"].Type = spec.TypeEmpty
	s.ResourceTypes["AWS::Rekognition::StreamProcessor"].Properties["PolygonRegionsOfInterest"].PrimitiveType = "Json"
}

func patchSamSpec(s spec.Spec) {
	s.PropertyTypes["AWS::Serverless::Api.ApiUsagePlan"].Properties["Quota"].Type = "AWS::ApiGateway::UsagePlan.QuotaSettings"
	s.PropertyTypes["AWS::Serverless::Api.ApiUsagePlan"].Properties["Throttle"].Type = "AWS::ApiGateway::UsagePlan.ThrottleSettings"
	s.PropertyTypes["AWS::Serverless::Api.DomainConfiguration"].Properties["MutualTlsAuthentication"].Type = "AWS::ApiGateway::DomainName.MutualTlsAuthentication"
	s.PropertyTypes["AWS::Serverless::Function.CloudWatchEvent"].Properties["Pattern"].PrimitiveType = "Json"
	s.PropertyTypes["AWS::Serverless::Function.CloudWatchEvent"].Properties["Pattern"].Type = spec.TypeEmpty
	s.PropertyTypes["AWS::Serverless::Function.DocumentDB"].Properties["Cluster"].ItemType = ""
	s.PropertyTypes["AWS::Serverless::Function.DocumentDB"].Properties["Cluster"].PrimitiveItemType = ""
	s.PropertyTypes["AWS::Serverless::Function.DocumentDB"].Properties["Cluster"].Type = ""
	s.PropertyTypes["AWS::Serverless::Function.DocumentDB"].Properties["Cluster"].PrimitiveType = "String"
	s.PropertyTypes["AWS::Serverless::Function.DocumentDB"].Properties["FilterCriteria"].Type = "AWS::Lambda::EventSourceMapping.FilterCriteria"
	s.PropertyTypes["AWS::Serverless::Function.DynamoDB"].Properties["DestinationConfig"].Type = "AWS::Lambda::EventSourceMapping.DestinationConfig"
	s.PropertyTypes["AWS::Serverless::Function.DynamoDB"].Properties["FilterCriteria"].Type = "AWS::Lambda::EventSourceMapping.FilterCriteria"
	s.PropertyTypes["AWS::Serverless::Function.EventBridgeRule"].Properties["Pattern"].PrimitiveType = "Json"
	s.PropertyTypes["AWS::Serverless::Function.EventBridgeRule"].Properties["Pattern"].Type = spec.TypeEmpty
	s.PropertyTypes["AWS::Serverless::Function.EventBridgeRule"].Properties["RetryPolicy"].Type = "AWS::Events::Rule.RetryPolicy"
	s.PropertyTypes["AWS::Serverless::Function.FunctionUrlConfig"].Properties["Cors"].Type = "AWS::Lambda::Url.Cors"
	s.PropertyTypes["AWS::Serverless::Function.HttpApi"].Properties["RouteSettings"].Type = "AWS::ApiGatewayV2::Stage.RouteSettings"
	s.PropertyTypes["AWS::Serverless::Function.Kinesis"].Properties["DestinationConfig"].Type = "AWS::Lambda::EventSourceMapping.DestinationConfig"
	s.PropertyTypes["AWS::Serverless::Function.Kinesis"].Properties["FilterCriteria"].Type = "AWS::Lambda::EventSourceMapping.FilterCriteria"
	s.PropertyTypes["AWS::Serverless::Function.MQ"].Properties["FilterCriteria"].Type = "AWS::Lambda::EventSourceMapping.FilterCriteria"
	s.PropertyTypes["AWS::Serverless::Function.MSK"].Properties["FilterCriteria"].Type = "AWS::Lambda::EventSourceMapping.FilterCriteria"
	s.PropertyTypes["AWS::Serverless::Function.MSK"].Properties["SourceAccessConfigurations"].ItemType = "AWS::Lambda::EventSourceMapping.SourceAccessConfiguration"
	s.PropertyTypes["AWS::Serverless::Function.MSK"].Properties["SourceAccessConfigurations"].Type = spec.TypeList
	s.PropertyTypes["AWS::Serverless::Function.S3"].Properties["Filter"].Type = "AWS::S3::Bucket.NotificationFilter"
	s.PropertyTypes["AWS::Serverless::Function.SNS"].Properties["FilterPolicy"].PrimitiveType = "Json"
	s.PropertyTypes["AWS::Serverless::Function.SNS"].Properties["FilterPolicy"].Type = spec.TypeEmpty
	s.PropertyTypes["AWS::Serverless::Function.SQS"].Properties["FilterCriteria"].Type = "AWS::Lambda::EventSourceMapping.FilterCriteria"
	s.PropertyTypes["AWS::Serverless::Function.SQS"].Properties["ScalingConfig"].Type = "AWS::Lambda::EventSourceMapping.ScalingConfig"
	s.PropertyTypes["AWS::Serverless::Function.Schedule"].Properties["RetryPolicy"].Type = "AWS::Events::Rule.RetryPolicy"
	s.PropertyTypes["AWS::Serverless::Function.ScheduleV2"].Properties["FlexibleTimeWindow"].Type = "AWS::Scheduler::Schedule.FlexibleTimeWindow"
	s.PropertyTypes["AWS::Serverless::Function.ScheduleV2"].Properties["RetryPolicy"].Type = "AWS::Scheduler::Schedule.RetryPolicy"
	s.PropertyTypes["AWS::Serverless::Function.SelfManagedKafka"].Properties["FilterCriteria"].Type = "AWS::Lambda::EventSourceMapping.FilterCriteria"
	s.PropertyTypes["AWS::Serverless::Function.SelfManagedKafka"].Properties["SourceAccessConfigurations"].ItemType = "AWS::Lambda::EventSourceMapping.SourceAccessConfiguration"
	s.PropertyTypes["AWS::Serverless::Function.SelfManagedKafka"].Properties["SourceAccessConfigurations"].Type = spec.TypeList
	s.PropertyTypes["AWS::Serverless::HttpApi.HttpApiDomainConfiguration"].Properties["MutualTlsAuthentication"].Type = "AWS::ApiGateway::DomainName.MutualTlsAuthentication"
	s.PropertyTypes["AWS::Serverless::StateMachine.CloudWatchEvent"].Properties["Pattern"].PrimitiveType = "Json"
	s.PropertyTypes["AWS::Serverless::StateMachine.CloudWatchEvent"].Properties["Pattern"].Type = spec.TypeEmpty
	s.PropertyTypes["AWS::Serverless::StateMachine.EventBridgeRule"].Properties["Pattern"].PrimitiveType = "Json"
	s.PropertyTypes["AWS::Serverless::StateMachine.EventBridgeRule"].Properties["Pattern"].Type = spec.TypeEmpty
	s.PropertyTypes["AWS::Serverless::StateMachine.EventBridgeRule"].Properties["RetryPolicy"].Type = "AWS::Events::Rule.RetryPolicy"
	s.PropertyTypes["AWS::Serverless::StateMachine.Schedule"].Properties["RetryPolicy"].Type = "AWS::Events::Rule.RetryPolicy"
	s.PropertyTypes["AWS::Serverless::StateMachine.ScheduleV2"].Properties["FlexibleTimeWindow"].Type = "AWS::Scheduler::Schedule.FlexibleTimeWindow"
	s.PropertyTypes["AWS::Serverless::StateMachine.ScheduleV2"].Properties["RetryPolicy"].Type = "AWS::Scheduler::Schedule.RetryPolicy"
	s.ResourceTypes["AWS::Serverless::Api"].Properties["AccessLogSetting"].Type = "AWS::ApiGateway::Stage.AccessLogSetting"
	s.ResourceTypes["AWS::Serverless::Api"].Properties["CanarySetting"].Type = "AWS::ApiGateway::Stage.CanarySetting"
	s.ResourceTypes["AWS::Serverless::Api"].Properties["DefinitionBody"].PrimitiveType = "Json"
	s.ResourceTypes["AWS::Serverless::Api"].Properties["DefinitionBody"].Type = spec.TypeEmpty
	s.ResourceTypes["AWS::Serverless::Api"].Properties["MethodSettings"].ItemType = "AWS::ApiGateway::Stage.MethodSetting"
	s.ResourceTypes["AWS::Serverless::Api"].Properties["MethodSettings"].Type = "List"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["AssumeRolePolicyDocument"].PrimitiveType = "Json"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["AssumeRolePolicyDocument"].Type = spec.TypeEmpty
	s.ResourceTypes["AWS::Serverless::Function"].Properties["Environment"].Type = "AWS::Lambda::Function.Environment"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["EphemeralStorage"].PrimitiveType = "Integer"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["EphemeralStorage"].Type = spec.TypeEmpty
	s.ResourceTypes["AWS::Serverless::Function"].Properties["ImageConfig"].Type = "AWS::Lambda::Function.ImageConfig"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["ProvisionedConcurrencyConfig"].Type = "AWS::Lambda::Alias.ProvisionedConcurrencyConfiguration"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["RuntimeManagementConfig"].Type = "AWS::Lambda::Function.RuntimeManagementConfig"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["SnapStart"].Type = "AWS::Lambda::Function.SnapStart"
	s.ResourceTypes["AWS::Serverless::Function"].Properties["VpcConfig"].Type = "AWS::Lambda::Function.VpcConfig"
	s.ResourceTypes["AWS::Serverless::HttpApi"].Properties["AccessLogSettings"].Type = "AWS::ApiGatewayV2::Stage.AccessLogSettings"
	s.ResourceTypes["AWS::Serverless::HttpApi"].Properties["DefaultRouteSettings"].Type = "AWS::ApiGatewayV2::Stage.RouteSettings"
	s.ResourceTypes["AWS::Serverless::HttpApi"].Properties["DefinitionBody"].PrimitiveType = "Json"
	s.ResourceTypes["AWS::Serverless::HttpApi"].Properties["DefinitionBody"].Type = spec.TypeEmpty
	s.ResourceTypes["AWS::Serverless::HttpApi"].Properties["RouteSettings"].Type = "AWS::ApiGatewayV2::Stage.RouteSettings"
	s.ResourceTypes["AWS::Serverless::SimpleTable"].Properties["ProvisionedThroughput"].Type = "AWS::DynamoDB::Table.ProvisionedThroughput"
	s.ResourceTypes["AWS::Serverless::SimpleTable"].Properties["SSESpecification"].Type = "AWS::DynamoDB::Table.SSESpecification"
	s.ResourceTypes["AWS::Serverless::SimpleTable"].Properties["PointInTimeRecoverySpecification"].Type = "AWS::DynamoDB::Table.PointInTimeRecoverySpecification"
	s.ResourceTypes["AWS::Serverless::StateMachine"].Properties["DefinitionUri"].Type = "AWS::StepFunctions::StateMachine.S3Location"
	s.ResourceTypes["AWS::Serverless::StateMachine"].Properties["Logging"].Type = "AWS::StepFunctions::StateMachine.LoggingConfiguration"
	s.ResourceTypes["AWS::Serverless::StateMachine"].Properties["Tracing"].Type = "AWS::StepFunctions::StateMachine.TracingConfiguration"
}

func checkIntegrity(s spec.Spec) bool {
	passed := true

	for rName, r := range s.ResourceTypes {
		for pName, p := range r.Properties {
			if p.PrimitiveType != spec.TypeEmpty {
				continue
			}

			t := p.Type
			if p.Type == spec.TypeList || p.Type == spec.TypeMap {
				if p.PrimitiveItemType != spec.TypeEmpty {
					continue
				}

				t = p.ItemType
			}

			if t != "" {
				if _, ok := s.PropertyTypes[t]; !ok {
					if _, ok := s.PropertyTypes[rName+"."+t]; !ok {
						fmt.Fprintf(os.Stderr, "s.ResourceTypes[\"%s\"].Properties[\"%s\"].Type = \"NOT %s\"\n", rName, pName, t)
						passed = false
					}
				}
			}
		}
	}

	for ptName, r := range s.PropertyTypes {
		parts := strings.Split(ptName, ".")
		rName := parts[0]

		for pName, p := range r.Properties {
			t := p.Type
			if p.Type == spec.TypeList || p.Type == spec.TypeMap {
				t = p.ItemType
			}

			if t != "" {
				if _, ok := s.PropertyTypes[t]; !ok {
					if _, ok := s.PropertyTypes[rName+"."+t]; !ok {
						fmt.Fprintf(os.Stderr, "s.PropertyTypes[\"%s\"].Properties[\"%s\"].Type = \"NOT %s\"\n", ptName, pName, t)
						passed = false
					}
				}
			}
		}
	}

	return passed
}

func main() {
	// Merge cfn and sam specs
	cfnSpec := loadURL(cfnSpecURL)
	patchCfnSpec(cfnSpec)
	saveJSON(cfnSpec, cfnSpecFn)

	samSpec := loadFile(samSpecFn)
	patchSamSpec(samSpec)
	saveJSON(samSpec, samSpecFn)

	for name, res := range samSpec.ResourceTypes {
		cfnSpec.ResourceTypes[name] = res
	}

	for name, prop := range samSpec.PropertyTypes {
		cfnSpec.PropertyTypes[name] = prop
	}

	if !checkIntegrity(cfnSpec) {
		panic("Failed integrity check")
	}

	// Save specs
	saveSpec(cfnSpec, "Cfn")
	saveSpec(loadFile(iamSpecFn), "Iam")

	// Clean up
	os.Remove(cfnSpecFn)
	os.Remove(samSpecFn)
}