// Copyright 2022 CloudWeGo Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package tpl

// MainTpl is the template for generating main.go
var MainTpl string = `package main

import (
	{{- range $path, $aliases := .Imports}}
		{{- if not $aliases}}
			"{{$path}}"
		{{- else}}
			{{- range $alias, $is := $aliases}}
				{{$alias}} "{{$path}}"
			{{- end}}
		{{- end}}
	{{- end}}
)

func main() {
    svr := {{.PkgRefName}}.NewServer(new({{.ServiceName}}Impl))

    err := svr.Run()

    if err != nil {
        log.Println(err.Error())
    }
}
`

var MainMultipleServicesTpl string = `package main

import (
	{{- range $path, $aliases := .Imports}}
		{{- if not $aliases}}
			"{{$path}}"
		{{- else}}
			{{- range $alias, $is := $aliases}}
				{{$alias}} "{{$path}}"
			{{- end}}
		{{- end}}
	{{- end}}
)

func main() {
    svr := server.NewServer()

    {{- range $idx, $svc := .Services}}
    if err := {{$svc.RefName}}.RegisterService(svr, new({{$svc.ServiceName}}Impl)); err != nil {
        panic(err)
    }
    {{- end}}

    err := svr.Run()

    if err != nil {
        log.Println(err.Error())
    }
}

`
