{{/*
Copyright 2019-present Facebook Inc. All rights reserved.
This source code is licensed under the Apache 2.0 license found
in the LICENSE file in the root directory of this source tree.
*/}}

{{/* gotype: entgo.io/ent/entc/gen.Graph */}}

{{ define "import" }}
import (
	"bytes"
	"context"
	"errors"
	"fmt"
	"io"
	"math"
	"strings"
	"sync"
	"time"
	{{- /* Ignore generating on graph templates (e.g. config.tmpl, client.tmpl, etc). */}}
	{{- if ne $.Config.Package $.Package }}
		"{{ $.Config.Package }}/predicate"
		{{- if $.HasOneFieldID }}
			{{- with $.ID.Type.PkgPath }}
				{{- $name := $.ID.Type.PkgName }}
				{{ if ne $name (base $.ID.Type.PkgPath) }}{{ $name }} {{ end }}"{{ . }}"
			{{- end }}
		{{- end }}
		{{- /* Import external packages */}}
        {{- template "import/types" $ }}
	{{- end }}
	"entgo.io/ent"
	"entgo.io/ent/dialect"
	"entgo.io/ent/schema/field"
	{{- with $.Storage }}
		{{- range $import := .Imports }}
			"{{ $import }}"
		{{- end }}
	{{- end }}
	{{- if and (hasField $ "Scope") ($.Scope.Imports) }}
		{{- range $import := $.Scope.Imports }}
			{{ $import.Alias }} "{{ $import.Path }}"
		{{- end }}
	{{- end }}
	{{- template "import/additional" $ }}
)
{{ end }}

{{/* A template for importing fields with custom types */}}
{{ define "import/types" -}}
	{{- $seen := dict }}
	{{- $selfPkg := "" }}{{ if ne $.Config.Package $.Package }}{{ $selfPkg = print $.Config.Package "/" $.PackageDir }}{{ end }}
	{{- $fields := $.Fields }}{{ if $.HasOneFieldID }}{{ if $.ID.UserDefined }}{{ $fields = append $fields $.ID }}{{ end }}{{ end }}
	{{- range $f := $fields }}
		{{- $pkg := $f.Type.PkgPath }}
		{{- if and $pkg (not (hasImport (base $pkg))) (not (hasKey $seen $pkg)) (ne $pkg $selfPkg) }}
			{{- $name := $f.Type.PkgName }}
			{{ if ne $name (base $pkg) }}{{ $name }} {{ end}}"{{ $pkg }}"
			{{- $seen = set $seen $pkg true }}
		{{- end }}
	{{- end }}
	{{- /* Import packages for edge target ID types referenced in mutation fields. */}}
	{{- range $e := $.EdgesWithID }}
		{{- $pkg := $e.Type.ID.Type.PkgPath }}
		{{- if and $pkg (not (hasImport (base $pkg))) (not (hasKey $seen $pkg)) (ne $pkg $selfPkg) }}
			{{- $name := $e.Type.ID.Type.PkgName }}
			{{ if ne $name (base $pkg) }}{{ $name }} {{ end}}"{{ $pkg }}"
			{{- $seen = set $seen $pkg true }}
		{{- end }}
	{{- end }}
{{- end }}

{{/* A template for allowing additional imports by ent extensions or user templates.*/}}
{{ define "import/additional" }}
	{{- with $tmpls := matchTemplate "import/additional/*" }}
		{{- range $tmpl := $tmpls }}
			{{- xtemplate $tmpl $ }}
		{{- end }}
	{{- end }}
	{{- if $.Storage }}
		{{- with $tmpls := matchTemplate (printf "dialect/%s/import/additional/*" $.Storage) }}
			{{- range $tmpl := $tmpls }}
				{{- xtemplate $tmpl $ }}
			{{- end }}
		{{- end }}
	{{- end }}
{{ end }}