storj/cmd/certificates/auth.go

// Copyright (C) 2019 Storj Labs, Inc.
// See LICENSE for copying information.

package main

import (
	"bufio"
	"context"
	"encoding/csv"
	"fmt"
	"io"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"text/tabwriter"

	"github.com/spf13/cobra"
	"github.com/zeebo/errs"

	"storj.io/storj/certificate/authorization"
	"storj.io/storj/pkg/process"
)

var (
	authCmd = &cobra.Command{
		Use:   "auth",
		Short: "CSR authorization management",
	}

	authCreateCmd = &cobra.Command{
		Use:   "create <auth_increment_count> [<email>, ...]",
		Short: "Create authorizations from a list of emails",
		Args:  cobra.MinimumNArgs(1),
		RunE:  cmdCreateAuth,
	}

	authInfoCmd = &cobra.Command{
		Use:   "info [<email>, ...]",
		Short: "Get authorization(s) info from CSR authorization DB",
		RunE:  cmdInfoAuth,
	}

	authExportCmd = &cobra.Command{
		Use:   "export [<email>, ...]",
		Short: "Export authorization(s) from CSR authorization DB to a CSV file (or stdout)",
		RunE:  cmdExportAuth,
	}
)

func parseEmailsList(fileName, delimiter string) (emails []string, err error) {
	file, err := os.Open(fileName)
	if err != nil {
		return nil, errs.Wrap(err)
	}
	scanner := bufio.NewScanner(file)
	for scanner.Scan() {
		line := scanner.Text()
		if strings.HasPrefix(line, "#") {
			continue
		}
		emails = append(emails, line)
	}
	return emails, file.Close()
}

func cmdCreateAuth(cmd *cobra.Command, args []string) error {
	ctx, _ := process.Ctx(cmd)
	count, err := strconv.Atoi(args[0])
	if err != nil {
		return errs.New("Count couldn't be parsed: %s", args[0])
	}
	authDB, err := authorization.NewDBFromCfg(authCfg.Config.AuthorizationDB)
	if err != nil {
		return err
	}

	var emails []string
	if len(args) > 1 {
		if authCfg.EmailsPath != "" {
			return errs.New("Either use `--emails-path` or positional args, not both.")
		}
		emails = args[1:]
	} else {
		emails, err = parseEmailsList(authCfg.EmailsPath, authCfg.Delimiter)
		if err != nil {
			return errs.Wrap(err)
		}
	}

	var incErrs errs.Group
	for _, email := range emails {
		if _, err := authDB.Create(ctx, email, count); err != nil {
			incErrs.Add(err)
		}
	}
	return incErrs.Err()
}

func cmdInfoAuth(cmd *cobra.Command, args []string) error {
	ctx, _ := process.Ctx(cmd)
	authDB, err := authorization.NewDBFromCfg(authCfg.Config.AuthorizationDB)
	if err != nil {
		return err
	}

	var emails []string
	if len(args) > 0 {
		if authCfg.EmailsPath != "" && !authCfg.All {
			return errs.New("Either use `--emails-path` or positional args, not both.")
		}
		emails = args
	} else if len(args) == 0 || authCfg.All {
		emails, err = authDB.UserIDs(ctx)
		if err != nil {
			return err
		}
	} else if _, err := os.Stat(authCfg.EmailsPath); err != nil {
		return errs.New("Emails path error: %s", err)
	} else {
		emails, err = parseEmailsList(authCfg.EmailsPath, authCfg.Delimiter)
		if err != nil {
			return errs.Wrap(err)
		}
	}

	var emailErrs, printErrs errs.Group
	w := tabwriter.NewWriter(os.Stdout, 0, 2, 2, ' ', 0)
	if _, err := fmt.Fprintln(w, "Email\tClaimed\tAvail.\t"); err != nil {
		return err
	}

	for _, email := range emails {
		if err := writeAuthInfo(ctx, authDB, email, w); err != nil {
			emailErrs.Add(err)
			continue
		}
	}

	if err := w.Flush(); err != nil {
		return errs.Wrap(err)
	}
	return errs.Combine(emailErrs.Err(), printErrs.Err())
}

func writeAuthInfo(ctx context.Context, authDB *authorization.DB, email string, w io.Writer) error {
	auths, err := authDB.Get(ctx, email)
	if err != nil {
		return err
	}
	if len(auths) < 1 {
		return nil
	}

	claimed, open := auths.GroupByClaimed()
	if _, err := fmt.Fprintf(w,
		"%s\t%d\t%d\t\n",
		email,
		len(claimed),
		len(open),
	); err != nil {
		return err
	}

	if authCfg.ShowTokens {
		if err := writeTokenInfo(claimed, open, w); err != nil {
			return err
		}
	}
	return nil
}

func writeTokenInfo(claimed, open authorization.Group, w io.Writer) error {
	groups := map[string]authorization.Group{
		"Claimed": claimed,
		"Open":    open,
	}
	for label, group := range groups {
		if _, err := fmt.Fprintf(w, "\t%s:\n", label); err != nil {
			return err
		}
		if len(group) > 0 {
			for _, auth := range group {
				if _, err := fmt.Fprintf(w, "\t\t%s\n", auth.Token.String()); err != nil {
					return err
				}
			}
		} else if _, err := fmt.Fprintln(w, "\t\tnone"); err != nil {
			return err
		}
	}
	return nil
}

func cmdExportAuth(cmd *cobra.Command, args []string) error {
	ctx, _ := process.Ctx(cmd)
	authDB, err := authorization.NewDBFromCfg(authCfg.Config.AuthorizationDB)
	if err != nil {
		return err
	}

	var emails []string
	if authCfg.All {
		if authCfg.EmailsPath != "" {
			return errs.New("Cannot use `--emails-path` with --all.")
		}
		emails, err = authDB.UserIDs(ctx)
		if err != nil {
			return err
		}
	} else {
		if authCfg.EmailsPath != "" {
			emails, err = parseEmailsList(authCfg.EmailsPath, authCfg.Delimiter)
			if err != nil {
				return errs.Wrap(err)
			}
		} else if len(args) == 0 {
			return errs.New("Need either `--emails-path` or positional args.")
		}
		emails = append(emails, args...)
	}

	var (
		emailErrs, csvErrs errs.Group
		output             io.Writer
	)
	switch authCfg.Out {
	case "-":
		output = os.Stdout
	default:
		if err := os.MkdirAll(filepath.Dir(authCfg.Out), 0600); err != nil {
			return errs.Wrap(err)
		}
		output, err = os.OpenFile(authCfg.Out, os.O_CREATE, 0600)
		if err != nil {
			return errs.Wrap(err)
		}
	}
	csvWriter := csv.NewWriter(output)

	for _, email := range emails {
		if err := writeAuthExport(ctx, authDB, email, csvWriter); err != nil {
			emailErrs.Add(err)
		}
	}

	csvWriter.Flush()
	return errs.Combine(emailErrs.Err(), csvErrs.Err())
}

func writeAuthExport(ctx context.Context, authDB *authorization.DB, email string, w *csv.Writer) error {
	auths, err := authDB.Get(ctx, email)
	if err != nil {
		return err
	}
	if len(auths) < 1 {
		return nil
	}

	var authErrs errs.Group
	for _, auth := range auths {
		isClaimed := "false"
		if auth.Claim != nil {
			isClaimed = "true"
		}

		if err := w.Write([]string{email, auth.Token.String(), isClaimed}); err != nil {
			authErrs.Add(err)
		}
	}
	return authErrs.Err()
}