preamble.go

package write

import (
	"path"

	"github.com/mavolin/corgi/file"
	"github.com/mavolin/corgi/file/fileutil"
	"github.com/mavolin/corgi/internal/meta"
)

func writePackage(ctx *ctx) {
	ctx.writeln("package " + ctx.destPackage)
}

func writeImports(ctx *ctx) {
	processed := make(map[string]struct{})
	namespaceUsed := make(map[string]struct{})

	files := make([]*file.File, len(ctx._stack))
	copy(files, ctx._stack)

	ctx.writeln("import (")
	defer ctx.writeln(")")

	writeBaseImports(ctx)

	for i := 0; i < len(files); i++ {
		f := files[i]

		if _, ok := processed[f.Module+"/"+f.PathInModule]; ok {
			continue
		}

		processed[f.Module+"/"+f.PathInModule] = struct{}{}

		for _, use := range f.Uses {
			for _, useSpec := range use.Uses {
				files = append(files, useSpec.Library.Files...)

				for _, dep := range useSpec.Library.Dependencies {
					files = append(files, dep.Library.Files...)
				}
			}
		}

		for _, imp := range f.Imports {
			for _, impSpec := range imp.Imports {
				var namespace string
				if impSpec.Alias != nil {
					namespace = impSpec.Alias.Ident
				} else {
					namespace = path.Base(fileutil.Unquote(impSpec.Path))
				}

				if _, ok := namespaceUsed[namespace]; ok {
					continue
				}

				namespaceUsed[namespace] = struct{}{}

				if impSpec.Alias != nil {
					ctx.writeln(impSpec.Alias.Ident + " " + fileutil.Quote(impSpec.Path))
				} else {
					ctx.writeln(fileutil.Quote(impSpec.Path))
				}
			}
		}
	}
}

func writeBaseImports(ctx *ctx) {
	ctx.writeln(ctx.ident("fmt") + ` "fmt"`)
	ctx.writeln(ctx.ident("io") + ` "io"`)
	ctx.writeln(ctx.ident("woof") + ` "github.com/mavolin/corgi/woof"`)
}

func writeCodegenComment(ctx *ctx) {
	ctx.writeln("")
	ctx.writeln("// Code generated by github.com/mavolin/corgi (" + meta.Version + "). DO NOT EDIT.")
	ctx.writeln("")
}

func writeGlobalCode(ctx *ctx) {
	for _, code := range ctx.mainFile().GlobalCode {
		for _, ln := range code.Lines {
			ctx.writeln(ln.Code)
		}
	}
}

func writeFuncCode(ctx *ctx) {
	for _, f := range ctx._stack[:len(ctx._stack)-1] {
		ctx.debug("func code", f.Name)
		for _, itm := range f.Scope {
			c, ok := itm.(file.Code)
			if !ok {
				continue
			}

			code(ctx, c)
		}
	}
}

func writeFunc(ctx *ctx) {
	ctx.write("func " + ctx.mainFile().Func.Name.Ident + "(" + ctx.ident("w") + " " + ctx.ioQual("Writer"))

	for _, param := range ctx.mainFile().Func.Params {
		ctx.write(", ")

		for nameI, name := range param.Names {
			if nameI > 0 {
				ctx.writeln(", ")
			}
			ctx.write(name.Ident)
		}

		ctx.write(" " + param.Type.Type)
	}

	ctx.writeln(") error {")
	defer ctx.writeln("}")

	ctx.writeln(ctx.ident(ctxVar) + " := " + ctx.woofFunc("NewContext", ctx.ident("w")))
	ctx.writeln("var " + ctx.ident("err") + " error")
	ctx.writeln("defer func() { " + ctx.ident("err") + " = " + ctx.contextFunc("Recover") + "}()")

	for _, comm := range ctx.mainFile().TopLevelComments {
		mcom := fileutil.ParseMachineComment(comm)
		if mcom.Namespace == "corgi" && mcom.Directive == "nonce" {
			ctx.debugItem(comm, comm.Lines[0].Comment+" (used to inject script nonce attr)")
			ctx.writeln(ctx.contextFunc("SetScriptNonce", mcom.Args))
			ctx.hasNonce = true
			break
		}
	}

	writeLibMixins(ctx)
	writeFuncCode(ctx)

	scope(ctx, ctx.baseFile().Scope, false)
	ctx.flushGenerate()
	ctx.writeln("return " + ctx.ident("err"))
}