// The Isomorphic Go Project
// Copyright (c) Wirecog, LLC. All rights reserved.
// Use of this source code is governed by a BSD-style
// license, which can be found in the LICENSE file.

package isokit

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"strings"
)

type TemplateSetContents struct {
	items map[string]string
}

func NewTemplateSetContents() *TemplateSetContents {

	return &TemplateSetContents{
		items: map[string]string{},
		// Funcs:   template.FuncMap{},
	}

}

func (tsc *TemplateSetContents) Items() map[string]string {
	return tsc.items
}

func (tsc *TemplateSetContents) importTemplateFileContents() error {

	templateDirectory := filepath.Clean(TemplateFilesPath)

	if err := filepath.Walk(templateDirectory, func(path string, info os.FileInfo, err error) error {
		if strings.HasSuffix(path, TemplateFileExtension) {
			name := strings.TrimSuffix(strings.TrimPrefix(path, templateDirectory+string(os.PathSeparator)), TemplateFileExtension)
			contents, err := ioutil.ReadFile(path)
			tsc.items[name] = string(contents)

			if err != nil {
				fmt.Println("error encountered while walking directory: ", err)
				return err
			}

		}
		return nil
	}); err != nil {
		return err
	}

	return nil

}