mediocre-blog/src/cmd/mailinglist-cli/main.go

package main

import (
	"context"
	"errors"
	"io"

	"github.com/mediocregopher/blog.mediocregopher.com/srv/cfg"
	cfgpkg "github.com/mediocregopher/blog.mediocregopher.com/srv/cfg"
	"github.com/mediocregopher/blog.mediocregopher.com/srv/mailinglist"
	"github.com/mediocregopher/mediocre-go-lib/v2/mctx"
	"github.com/mediocregopher/mediocre-go-lib/v2/mlog"
	"github.com/tilinna/clock"
)

func main() {

	ctx := context.Background()

	cfg := cfgpkg.NewBlogCfg(cfg.Params{})

	var dataDir cfgpkg.DataDir
	dataDir.SetupCfg(cfg)
	defer dataDir.Close()
	ctx = mctx.WithAnnotator(ctx, &dataDir)

	var mailerParams mailinglist.MailerParams
	mailerParams.SetupCfg(cfg)
	ctx = mctx.WithAnnotator(ctx, &mailerParams)

	var mlParams mailinglist.Params
	mlParams.SetupCfg(cfg)
	ctx = mctx.WithAnnotator(ctx, &mlParams)

	// 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)
	}

	clock := clock.Realtime()

	var mailer mailinglist.Mailer
	if mailerParams.SMTPAddr == "" {
		logger.Info(ctx, "-smtp-addr not given, using NullMailer")
		mailer = mailinglist.NullMailer
	} else {
		mailer = mailinglist.NewMailer(mailerParams)
	}

	mlStore, err := mailinglist.NewStore(dataDir)
	if err != nil {
		logger.Fatal(ctx, "initializing mailing list storage", err)
	}
	defer mlStore.Close()

	mlParams.Store = mlStore
	mlParams.Mailer = mailer
	mlParams.Clock = clock

	ml := mailinglist.New(mlParams)

	subCmd := cfg.SubCmd()
	ctx = mctx.Annotate(ctx, "subCmd", subCmd)

	switch subCmd {

	case "list":

		for it := mlStore.GetAll(); ; {
			email, err := it()
			if errors.Is(err, io.EOF) {
				break
			} else if err != nil {
				logger.Fatal(ctx, "retrieving next email", err)
			}

			ctx := mctx.Annotate(context.Background(),
				"email", email.Email,
				"createdAt", email.CreatedAt,
				"verifiedAt", email.VerifiedAt,
			)

			logger.Info(ctx, "next")
		}

	case "publish":

		title := cfg.String("title", "", "Title of the post which was published")
		url := cfg.String("url", "", "URL of the post which was published")

		if err := cfg.Init(ctx); err != nil {
			logger.Fatal(ctx, "initializing", err)
		}

		if *title == "" {
			logger.FatalString(ctx, "-title is required")

		} else if *url == "" {
			logger.FatalString(ctx, "-url is required")
		}

		err := ml.Publish(*title, *url)
		if err != nil {
			logger.Fatal(ctx, "publishing", err)
		}

	default:
		logger.FatalString(ctx, "invalid sub-command, must be list|publish")
	}
}