163 lines
3.3 KiB
Go
163 lines
3.3 KiB
Go
package actions
|
|
|
|
import (
|
|
"bufio"
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"git.t-juice.club/torjus/gpaste/client"
|
|
"git.t-juice.club/torjus/gpaste/files"
|
|
"github.com/urfave/cli/v2"
|
|
"golang.org/x/term"
|
|
)
|
|
|
|
const defaultTimeout = 10 * time.Second
|
|
|
|
func ActionUpload(c *cli.Context) error {
|
|
clnt := client.Client{
|
|
BaseURL: c.String("url"),
|
|
}
|
|
|
|
for _, arg := range c.Args().Slice() {
|
|
f, err := os.Open(arg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer f.Close()
|
|
|
|
file := &files.File{
|
|
OriginalFilename: arg,
|
|
Body: f,
|
|
}
|
|
|
|
resp, err := clnt.Upload(c.Context, file)
|
|
if err != nil {
|
|
errmsg := fmt.Sprintf("Error uploading file: %s", err)
|
|
return cli.Exit(errmsg, 1)
|
|
}
|
|
|
|
fmt.Printf("Uploaded file %s - %s", file.OriginalFilename, resp.Files[0].URL)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func ActionDelete(c *cli.Context) error {
|
|
clnt := client.Client{
|
|
BaseURL: c.String("url"),
|
|
}
|
|
|
|
for _, arg := range c.Args().Slice() {
|
|
ctx, cancel := context.WithTimeout(c.Context, defaultTimeout)
|
|
defer cancel()
|
|
|
|
if err := clnt.Delete(ctx, arg); err != nil {
|
|
fmt.Printf("Error deleting file %s\n", arg)
|
|
fmt.Printf("%s\n", err)
|
|
}
|
|
|
|
fmt.Printf("Deleted %s\n", arg)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func ActionLogin(c *cli.Context) error {
|
|
username := c.Args().First()
|
|
if username == "" {
|
|
return cli.Exit("USERNAME not supplied.", 1)
|
|
}
|
|
|
|
password, err := readPassword()
|
|
if err != nil {
|
|
return fmt.Errorf("error reading password: %w", err)
|
|
}
|
|
|
|
clnt := client.Client{
|
|
BaseURL: c.String("url"),
|
|
}
|
|
if err := clnt.Login(c.Context, username, password); err != nil {
|
|
errmsg := fmt.Sprintf("Error logging in: %s", err)
|
|
return cli.Exit(errmsg, 1)
|
|
}
|
|
|
|
if err := clnt.WriteConfig(); err != nil {
|
|
errMsg := fmt.Sprintf("Failed to write config: %s", err)
|
|
return cli.Exit(errMsg, 1)
|
|
}
|
|
// TODO: Store this somewhere, so we don't need to log in each time
|
|
fmt.Println("Successfully logged in.")
|
|
|
|
return nil
|
|
}
|
|
|
|
func ActionUserCreate(c *cli.Context) error {
|
|
// TODO: Needs to supply auth token to actually work
|
|
fmt.Println("Need to be logged in to create user")
|
|
|
|
username := readString("Enter username: ")
|
|
|
|
password, err := readPassword()
|
|
if err != nil {
|
|
return fmt.Errorf("error reading password: %w", err)
|
|
}
|
|
|
|
clnt := client.Client{
|
|
BaseURL: c.String("url"),
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(c.Context, defaultTimeout)
|
|
defer cancel()
|
|
|
|
if err := clnt.Login(ctx, username, password); err != nil {
|
|
errmsg := fmt.Sprintf("Error logging in: %s", err)
|
|
return cli.Exit(errmsg, 1)
|
|
}
|
|
|
|
fmt.Println("User to create:")
|
|
|
|
username = readString("Enter username: ")
|
|
|
|
password, err = readPassword()
|
|
if err != nil {
|
|
return fmt.Errorf("error reading password: %w", err)
|
|
}
|
|
|
|
if err := clnt.UserCreate(ctx, username, password); err != nil {
|
|
errmsg := fmt.Sprintf("Error creating user: %s", err)
|
|
return cli.Exit(errmsg, 1)
|
|
}
|
|
|
|
fmt.Printf("Created user %s\n", username)
|
|
|
|
return nil
|
|
}
|
|
|
|
func readPassword() (string, error) {
|
|
fmt.Print("Enter Password: ")
|
|
|
|
bytePassword, err := term.ReadPassword(int(syscall.Stdin))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
password := string(bytePassword)
|
|
|
|
return strings.TrimSpace(password), nil
|
|
}
|
|
|
|
func readString(prompt string) string {
|
|
fmt.Print(prompt)
|
|
|
|
scanner := bufio.NewScanner(os.Stdin)
|
|
for scanner.Scan() {
|
|
return scanner.Text()
|
|
}
|
|
|
|
return ""
|
|
}
|