package main

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

type siteData struct {
	name       string
	rawName    string
	baseUrl    string
	outputPath string
	sourcePath string
	pages      map[string]page
	nav        []navItem
}

func getSites(cfg config) (sites []siteData) {
	files, err := ioutil.ReadDir(cfg.DocsRoot)
	if err != nil {
		panic(err)
	}

	for _, f := range files {
		if f.IsDir() && f.Name() != ".git" {
			newSite := siteData{
				name:       nameify(f.Name()),
				rawName:    f.Name(),
				baseUrl:    filepath.Join(cfg.BaseUrl, f.Name()),
				outputPath: filepath.Join(cfg.TargetDir, f.Name()),
				sourcePath: filepath.Join(cfg.DocsRoot, f.Name())}
			sites = append(sites, newSite)
		}
	}

	return
}

func generateSite(site siteData, cfg config, outputDir string) {
	template := loadTemplate(cfg)
	templData := templateData{SiteTitle: site.name,
		StylesheetUrl: cfg.BaseUrl + "/style.css",
		Nav:           site.nav}
	for k, v := range site.pages {
		templData.Content = v.content
		err := applyTemplate(filepath.Join(outputDir, site.rawName, k), templData, template)
		if err != nil {
			panic(err)
		}
	}

	// link images to generation directory
	err := os.Symlink(filepath.Join(site.sourcePath, "img"), filepath.Join(outputDir, site.rawName, "img"))
	if err != nil {
		panic(err)
	}
}