tools/check-licenses/result/template.go

// Copyright 2022 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package result

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"go.fuchsia.dev/fuchsia/tools/check-licenses/result/world"
)

// Generate output files for every template defined in the config file.
func expandTemplates() (string, error) {
	if Config.OutDir == "" {
		return "", nil
	}

	outDir := filepath.Join(Config.OutDir, "out")
	if _, err := os.Stat(outDir); os.IsNotExist(err) {
		err := os.Mkdir(outDir, 0755)
		if err != nil {
			return "", err
		}
	}

	var b strings.Builder
	b.WriteString("\n")

	w, err := world.NewWorld()
	if err != nil {
		return "", err
	}
	b.WriteString(w.Status.String())
	b.WriteString("\n")

	for _, o := range Config.Outputs {
		if t, ok := AllTemplates[o]; !ok {
			return "", fmt.Errorf("Couldn't find template %v\n", o)
		} else {
			name := filepath.Join(outDir, o)
			f, err := os.Create(name)
			if err != nil {
				return "", err
			}
			if err := t.Execute(f, w); err != nil {
				return "", err
			}
			if Config.Zip {
				if err := compressGZ(name); err != nil {
					return "", err
				}
			}
			b.WriteString(fmt.Sprintf(" ⦿ Executed template -> %v", name))
			if Config.Zip {
				b.WriteString(" (+ *.gz)")
			}
			b.WriteString("\n")
		}
	}
	return b.String(), nil
}