metadata.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"
	"os"

	"github.com/yannh/r10k-go/puppetmodule"
)

type dependency struct {
	name               string
	versionRequirement string
}

type metadata struct {
	name         string
	dependencies []dependency
}

type metadataFile struct {
	*os.File
	filename string
	env      environment
}

func newMetadataFile(mf string, env environment) *metadataFile {
	// We just ignore if the file doesn't exist'
	f, err := os.Open(mf)
	if err != nil {
		return nil
	}

	return &metadataFile{File: f, filename: mf, env: env}
}

func (m *metadataFile) Close() { m.File.Close() }

func (m *metadataFile) Process(drs chan<- downloadRequest) error {
	var meta metadata
	done := make(chan bool)

	metadataFile, err := ioutil.ReadAll(m.File)
	if err != nil {
		return fmt.Errorf("could not read JSON file %v", err)
	}

	if err = json.Unmarshal(metadataFile, &meta); err != nil {
		return fmt.Errorf("JSON file malformed: %v", err)
	}

	for _, req := range meta.dependencies {
		dr := downloadRequest{
			m:    puppetmodule.NewForgeModule(req.name, ""),
			env:  m.env,
			done: done,
		}

		go func(dr downloadRequest) {
			drs <- dr
		}(dr)
	}

	for i := 0; i < len(meta.dependencies); i++ {
		<-done
	}

	return nil
}