1 files changed, 129 insertions, 0 deletions
diff --git a/srv/src/cmd/load-test-data/main.go b/srv/src/cmd/load-test-data/main.go
new file mode 100644
index 0000000..01d3b71
--- /dev/null
+++ b/srv/src/cmd/load-test-data/main.go
@@ -0,0 +1,129 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"os"
+	"path/filepath"
+	"time"
+
+	cfgpkg "github.com/mediocregopher/blog.mediocregopher.com/srv/cfg"
+	"github.com/mediocregopher/blog.mediocregopher.com/srv/post"
+	"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
+	"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
+	"gopkg.in/yaml.v3"
+)
+
+type testData struct {
+	PublishedPosts []post.Post `yaml:"published_posts"`
+	Assets         map[string]string
+}
+
+func loadTestData(path string) (testData, error) {
+
+	f, err := os.Open(path)
+	if err != nil {
+		return testData{}, fmt.Errorf("opening file: %w", err)
+	}
+	defer f.Close()
+
+	var res testData
+
+	if err := yaml.NewDecoder(f).Decode(&res); err != nil {
+		return testData{}, fmt.Errorf("decoding file contents: %w", err)
+	}
+
+	return res, nil
+}
+
+func main() {
+
+	ctx := context.Background()
+
+	cfg := cfgpkg.NewBlogCfg(cfgpkg.Params{})
+
+	var dataDir cfgpkg.DataDir
+	dataDir.SetupCfg(cfg)
+	defer dataDir.Close()
+	ctx = mctx.WithAnnotator(ctx, &dataDir)
+
+	testDataPath := cfg.String("test-data-path", "./test-data.yml", "File containing the data to be loaded in")
+
+	// initialization
+	err := cfg.Init(ctx)
+
+	logger := mlog.NewLogger(nil)
+	defer logger.Close()
+
+	logger.Info(ctx, "process started")
+	defer logger.Info(ctx, "process exiting")
+
+	if err != nil {
+		logger.Fatal(ctx, "initializing", err)
+	}
+
+	testDataDir := filepath.Dir(*testDataPath)
+
+	testData, err := loadTestData(*testDataPath)
+	if err != nil {
+		logger.Fatal(ctx, "loading test data", err)
+	}
+
+	postDB, err := post.NewSQLDB(dataDir)
+	if err != nil {
+		logger.Fatal(ctx, "initializing post sql db", err)
+	}
+	defer postDB.Close()
+
+	{
+
+		postStore := post.NewStore(postDB)
+
+		now := time.Now().Truncate(time.Hour)
+
+		for _, post := range testData.PublishedPosts {
+
+			ctx := mctx.Annotate(ctx,
+				"postID", post.ID,
+				"now", now)
+
+			if _, err := postStore.Set(post, now); err != nil {
+				logger.Fatal(ctx, "setting post", err)
+			}
+
+			logger.Info(ctx, "set post")
+
+			now = now.Add(1 * time.Hour)
+		}
+
+	}
+
+	{
+		assetStore := post.NewAssetStore(postDB)
+
+		setAsset := func(assetID, assetPath string) error {
+			assetFullPath := filepath.Join(testDataDir, assetPath)
+
+			f, err := os.Open(assetFullPath)
+			if err != nil {
+				return fmt.Errorf("opening %q for reading: %w", assetFullPath, err)
+			}
+			defer f.Close()
+
+			return assetStore.Set(assetID, f)
+		}
+
+		for assetID, assetPath := range testData.Assets {
+
+			ctx := mctx.Annotate(ctx,
+				"assetID", assetID,
+				"assetPath", assetPath)
+
+			if err := setAsset(assetID, assetPath); err != nil {
+				logger.Fatal(ctx, "setting asset", err)
+			}
+
+			logger.Info(ctx, "set asset")
+		}
+	}
+}