summaryrefslogtreecommitdiff
path: root/srv/src
diff options
context:
space:
mode:
authorBrian Picciano <mediocregopher@gmail.com>2022-05-17 12:41:17 -0600
committerBrian Picciano <mediocregopher@gmail.com>2022-05-17 13:28:56 -0600
commitf9d1f664f0a554b58e674f195572cc8a71d0bbfa (patch)
treea0511f27a8261b98812ff1be0f2dd39bbf7efb78 /srv/src
parent4d2332582355c0b4c70d7817e569139d502d97a1 (diff)
Implement import-asset script
Diffstat (limited to 'srv/src')
-rw-r--r--srv/src/cmd/import-asset/main.go114
-rw-r--r--srv/src/cmd/import-asset/to-import.txt18
-rw-r--r--srv/src/cmd/import-posts/main.go3
3 files changed, 133 insertions, 2 deletions
diff --git a/srv/src/cmd/import-asset/main.go b/srv/src/cmd/import-asset/main.go
new file mode 100644
index 0000000..611bb1c
--- /dev/null
+++ b/srv/src/cmd/import-asset/main.go
@@ -0,0 +1,114 @@
+package main
+
+import (
+ "bufio"
+ "context"
+ "errors"
+ "fmt"
+ "io"
+ "os"
+ "strings"
+
+ 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"
+)
+
+func importAsset(assetStore post.AssetStore, id, path string) error {
+
+ f, err := os.Open(path)
+ if err != nil {
+ return fmt.Errorf("opening file: %w", err)
+ }
+ defer f.Close()
+
+ if err := assetStore.Set(id, f); err != nil {
+ return fmt.Errorf("setting into asset store: %w", err)
+ }
+
+ return 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)
+
+ id := cfg.String("id", "", "ID the asset will be stored under")
+ path := cfg.String("path", "", "Path the asset should be imported from")
+
+ fromStdin := cfg.Bool("from-stdin", false, "If set, ignore id and path, read space separated id/path pairs from stdin")
+
+ // initialization
+ err := cfg.Init(ctx)
+
+ logger := mlog.NewLogger(nil)
+ defer logger.Close()
+
+ if !*fromStdin && (*id == "" || *path == "") {
+ logger.FatalString(ctx, "-id and -path are required if -from-stdin is not given")
+ }
+
+ logger.Info(ctx, "process started")
+ defer logger.Info(ctx, "process exiting")
+
+ if err != nil {
+ logger.Fatal(ctx, "initializing", err)
+ }
+
+ postDB, err := post.NewSQLDB(dataDir)
+ if err != nil {
+ logger.Fatal(ctx, "initializing post sql db", err)
+ }
+ defer postDB.Close()
+
+ assetStore := post.NewAssetStore(postDB)
+
+ if !*fromStdin {
+
+ ctx := mctx.Annotate(ctx, "id", *id, "path", *path)
+
+ if err := importAsset(assetStore, *id, *path); err != nil {
+ logger.Fatal(ctx, "failed to import asset", err)
+ }
+
+ logger.Info(ctx, "asset stored")
+
+ return
+ }
+
+ for stdin := bufio.NewReader(os.Stdin); ; {
+
+ line, err := stdin.ReadString('\n')
+
+ if errors.Is(err, io.EOF) {
+ return
+ } else if err != nil {
+ logger.Fatal(ctx, "reading from stdin", err)
+ }
+
+ fields := strings.Fields(line)
+
+ if len(fields) < 2 {
+ ctx := mctx.Annotate(ctx, "line", line)
+ logger.FatalString(ctx, "cannot process line with fewer than 2 fields")
+ }
+
+ id, path := fields[0], fields[1]
+
+ ctx := mctx.Annotate(ctx, "id", id, "path", path)
+
+ if err := importAsset(assetStore, id, path); err != nil {
+ logger.Fatal(ctx, "failed to import asset", err)
+ }
+
+ logger.Info(ctx, "asset stored")
+ }
+}
diff --git a/srv/src/cmd/import-asset/to-import.txt b/srv/src/cmd/import-asset/to-import.txt
new file mode 100644
index 0000000..92647fa
--- /dev/null
+++ b/srv/src/cmd/import-asset/to-import.txt
@@ -0,0 +1,18 @@
+diamond-square-terrain.png ../../static/src/img/diamond-square/terrain.png
+diamond-square-dsalg.png ../../static/src/img/diamond-square/dsalg.png
+program-structure-diag1.jpg ../../static/src/img/program-structure/diag1.jpg
+program-structure-diag2.jpg ../../static/src/img/program-structure/diag2.jpg
+program-structure-diag3.jpg ../../static/src/img/program-structure/diag3.jpg
+open-infra-keybase.png ../../static/src/img/open-infra/keybase.png
+wedding-1.jpg ../../static/src/img/wedding/1.jpg
+wedding-2.jpg ../../static/src/img/wedding/2.jpg
+wedding-3.jpg ../../static/src/img/wedding/3.jpg
+happy-tree-partial.png ../../static/src/img/happy-tree/partial.png
+happy-tree-atmp1.png ../../static/src/img/happy-tree/happy-tree-atmp1.png
+happy-tree-atmp2.png ../../static/src/img/happy-tree/happy-tree-atmp2.png
+happy-tree-atmp3-pow3.png ../../static/src/img/happy-tree/happy-tree-atmp3-pow3.png
+happy-tree-atmp3-pow4.png ../../static/src/img/happy-tree/happy-tree-atmp3-pow4.png
+happy-tree-atmp3-pow5.png ../../static/src/img/happy-tree/happy-tree-atmp3-pow5.png
+happy-tree-atmp3-pow6.png ../../static/src/img/happy-tree/happy-tree-atmp3-pow6.png
+nfts-disaster-girl.jpg ../../static/src/img/nfts/disaster-girl.jpg
+nfts-gleaners.jpg ../../static/src/img/nfts/gleaners.jpg
diff --git a/srv/src/cmd/import-posts/main.go b/srv/src/cmd/import-posts/main.go
index e4eb9e7..b2dc889 100644
--- a/srv/src/cmd/import-posts/main.go
+++ b/srv/src/cmd/import-posts/main.go
@@ -10,7 +10,6 @@ import (
"time"
"github.com/adrg/frontmatter"
- "github.com/mediocregopher/blog.mediocregopher.com/srv/cfg"
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"
@@ -107,7 +106,7 @@ func main() {
ctx := context.Background()
- cfg := cfg.NewBlogCfg(cfg.Params{})
+ cfg := cfgpkg.NewBlogCfg(cfgpkg.Params{})
var dataDir cfgpkg.DataDir
dataDir.SetupCfg(cfg)