// Copyright 2017 AMIS Technologies
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package compose

import (
	"bytes"
	"fmt"
	"text/template"

	"github.com/istanbul-tools/docker/service"
)

type quorum struct {
	*istanbul
	Number         int
	QuorumServices []*service.Quorum
}

func newQuorum(ist *istanbul, number int) Compose {
	q := &quorum{
		istanbul: ist,
		Number:   number,
	}
	q.init()
	return q
}

func (q *quorum) init() {
	// set constellations
	var constellations []*service.Constellation
	for i := 0; i < q.Number; i++ {
		constellations = append(constellations,
			service.NewConstellation(q.Services[i].Identity,
				// from subnet ip 100
				fmt.Sprintf("%v.%v", q.IPPrefix, i+100),
				10000+i,
			),
		)
	}
	for i := 0; i < q.Number; i++ {
		// set othernodes
		var nodes []string
		for j := 0; j < q.Number; j++ {
			if i != j {
				nodes = append(nodes, constellations[j].Host())
			}
		}
		constellations[i].SetOtherNodes(nodes)

		// update quorum service
		q.QuorumServices = append(q.QuorumServices,
			service.NewQuorum(q.Services[i], constellations[i]))
	}
}

func (q *quorum) String() string {
	tmpl, err := template.New("quorum").Funcs(template.FuncMap(
		map[string]interface{}{
			"PrintVolumes": func() (result string) {
				for i := 0; i < q.Number; i++ {
					result += fmt.Sprintf("  \"%v\":\n", i)
				}
				return
			},
		})).Parse(quorumTemplate)
	if err != nil {
		fmt.Printf("Failed to parse template, %v", err)
		return ""
	}

	result := new(bytes.Buffer)
	err = tmpl.Execute(result, q)
	if err != nil {
		fmt.Printf("Failed to render template, %v", err)
		return ""
	}

	return result.String()
}

var quorumTemplate = `version: '3'
services:
  {{ .EthStats }}
  {{- range .QuorumServices }}
  {{ . }}
  {{- end }}
networks:
  app_net:
    driver: bridge
    ipam:
      driver: default
      config:
      - subnet: {{ .IPPrefix }}.0/24
volumes:
{{ PrintVolumes }}
`