phylum/server/internal/command/common/common.go

package common

import (
	"context"
	"fmt"
	"os"

	"github.com/shroff/phylum/server/internal/core/fs"
	"github.com/shroff/phylum/server/internal/core/user"
	"github.com/spf13/cobra"
)

var f fs.FileSystem

func RootFileSystem() fs.FileSystem {
	var rootUser user.User
	if user, err := user.CreateManager(context.Background()).UserByEmail(user.DefaultUserUsername); err != nil {
		fmt.Println("unable to find root user: " + err.Error())
		os.Exit(1)
	} else {
		rootUser = user
	}

	return rootUser.OpenFileSystem(context.Background())
}

func UserFileSystem(cmd *cobra.Command) fs.FileSystem {
	if f == nil {
		if value, err := cmd.Flags().GetString("user"); err != nil {
			fmt.Println("could not read user: " + err.Error())
			os.Exit(4)
		} else {
			if u, err := user.CreateManager(context.Background()).UserByEmail(value); err != nil {
				fmt.Println("could not find user '" + value + "': " + err.Error())
				os.Exit(5)
			} else {
				f = u.OpenFileSystem(context.Background())
			}
		}
	}
	return f
}