2022-03-22 19:17:50 +00:00
|
|
|
package examples
|
|
|
|
|
|
|
|
import (
|
|
|
|
"archive/tar"
|
|
|
|
"bytes"
|
|
|
|
"embed"
|
|
|
|
"path"
|
|
|
|
"sync"
|
|
|
|
|
|
|
|
"github.com/gohugoio/hugo/parser/pageparser"
|
|
|
|
"golang.org/x/sync/singleflight"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
)
|
|
|
|
|
|
|
|
var (
|
|
|
|
//go:embed */*.md
|
|
|
|
//go:embed */*.tf
|
|
|
|
files embed.FS
|
|
|
|
|
|
|
|
examples = make([]Example, 0)
|
|
|
|
parseExamples sync.Once
|
|
|
|
archives = singleflight.Group{}
|
|
|
|
)
|
|
|
|
|
|
|
|
type Example struct {
|
|
|
|
ID string `json:"id"`
|
|
|
|
Name string `json:"name"`
|
|
|
|
Description string `json:"description"`
|
|
|
|
Markdown string `json:"markdown"`
|
|
|
|
}
|
|
|
|
|
|
|
|
// List returns all embedded examples.
|
|
|
|
func List() ([]Example, error) {
|
|
|
|
var returnError error
|
|
|
|
parseExamples.Do(func() {
|
|
|
|
dirs, err := files.ReadDir(".")
|
|
|
|
if err != nil {
|
|
|
|
returnError = xerrors.Errorf("read dir: %w", err)
|
|
|
|
return
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
for _, dir := range dirs {
|
|
|
|
exampleID := dir.Name()
|
|
|
|
// Each one of these is a example!
|
|
|
|
readme, err := files.ReadFile(path.Join(dir.Name(), "README.md"))
|
|
|
|
if err != nil {
|
|
|
|
returnError = xerrors.Errorf("example %q does not contain README.md", exampleID)
|
|
|
|
return
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
frontMatter, err := pageparser.ParseFrontMatterAndContent(bytes.NewReader(readme))
|
|
|
|
if err != nil {
|
|
|
|
returnError = xerrors.Errorf("parse example %q front matter: %w", exampleID, err)
|
|
|
|
return
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
nameRaw, exists := frontMatter.FrontMatter["name"]
|
|
|
|
if !exists {
|
|
|
|
returnError = xerrors.Errorf("example %q front matter does not contain name", exampleID)
|
|
|
|
return
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
name, valid := nameRaw.(string)
|
|
|
|
if !valid {
|
|
|
|
returnError = xerrors.Errorf("example %q name isn't a string", exampleID)
|
|
|
|
return
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
descriptionRaw, exists := frontMatter.FrontMatter["description"]
|
|
|
|
if !exists {
|
|
|
|
returnError = xerrors.Errorf("example %q front matter does not contain name", exampleID)
|
|
|
|
return
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
description, valid := descriptionRaw.(string)
|
|
|
|
if !valid {
|
|
|
|
returnError = xerrors.Errorf("example %q description isn't a string", exampleID)
|
|
|
|
return
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
examples = append(examples, Example{
|
|
|
|
ID: exampleID,
|
|
|
|
Name: name,
|
|
|
|
Description: description,
|
|
|
|
Markdown: string(frontMatter.Content),
|
|
|
|
})
|
|
|
|
}
|
|
|
|
})
|
|
|
|
return examples, returnError
|
|
|
|
}
|
|
|
|
|
|
|
|
// Archive returns a tar by example ID.
|
|
|
|
func Archive(exampleID string) ([]byte, error) {
|
|
|
|
rawData, err, _ := archives.Do(exampleID, func() (interface{}, error) {
|
|
|
|
examples, err := List()
|
|
|
|
if err != nil {
|
2022-04-01 19:42:36 +00:00
|
|
|
return nil, xerrors.Errorf("list: %w", err)
|
2022-03-22 19:17:50 +00:00
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
var selected Example
|
|
|
|
for _, example := range examples {
|
|
|
|
if example.ID != exampleID {
|
|
|
|
continue
|
|
|
|
}
|
|
|
|
selected = example
|
|
|
|
break
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
if selected.ID == "" {
|
|
|
|
return nil, xerrors.Errorf("example with id %q not found", exampleID)
|
|
|
|
}
|
|
|
|
|
|
|
|
entries, err := files.ReadDir(exampleID)
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("read dir: %w", err)
|
|
|
|
}
|
|
|
|
|
|
|
|
var buffer bytes.Buffer
|
|
|
|
tarWriter := tar.NewWriter(&buffer)
|
|
|
|
|
|
|
|
for _, entry := range entries {
|
|
|
|
file, err := files.Open(path.Join(exampleID, entry.Name()))
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("open file: %w", err)
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
info, err := file.Stat()
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("stat file: %w", err)
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
if info.IsDir() {
|
|
|
|
continue
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
data := make([]byte, info.Size())
|
|
|
|
_, err = file.Read(data)
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("read data: %w", err)
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
header, err := tar.FileInfoHeader(info, entry.Name())
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("get file header: %w", err)
|
|
|
|
}
|
|
|
|
header.Mode = 0644
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
err = tarWriter.WriteHeader(header)
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("write file: %w", err)
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
_, err = tarWriter.Write(data)
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("write: %w", err)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
err = tarWriter.Flush()
|
|
|
|
if err != nil {
|
|
|
|
return nil, xerrors.Errorf("flush archive: %w", err)
|
|
|
|
}
|
2022-04-01 19:42:36 +00:00
|
|
|
|
2022-03-22 19:17:50 +00:00
|
|
|
return buffer.Bytes(), nil
|
|
|
|
})
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
data, valid := rawData.([]byte)
|
|
|
|
if !valid {
|
|
|
|
panic("dev error: data must be a byte slice")
|
|
|
|
}
|
|
|
|
return data, nil
|
|
|
|
}
|