mirror of
https://github.com/Luzifer/share.git
synced 2024-10-18 05:14:23 +00:00
118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package main
|
|
|
|
//go:generate make pack
|
|
|
|
import (
|
|
"bytes"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
|
|
"github.com/Luzifer/rconfig"
|
|
)
|
|
|
|
var (
|
|
cfg = struct {
|
|
BaseURL string `flag:"base-url" env:"BASE_URL" default:"" description:"URL to prepend before filename"`
|
|
BasePath string `flag:"base-path" env:"BASE_PATH" default:"file/{{ printf \"%.6s\" .Hash }}" description:"Path to upload the file to"`
|
|
Bootstrap bool `flag:"bootstrap" default:"false" description:"Upload frontend files into bucket"`
|
|
Bucket string `flag:"bucket" env:"BUCKET" default:"" description:"S3 bucket to upload files to" validate:"nonzero"`
|
|
ContentType string `flag:"content-type,c" default:"" description:"Force content-type to be set to this value"`
|
|
Listen string `flag:"listen" env:"LISTEN" default:"" description:"Enable HTTP server if set to IP/Port (e.g. ':3000')"`
|
|
Progress bool `flag:"progress" default:"false" description:"Show progress bar while uploading"`
|
|
VersionAndExit bool `flag:"version" default:"false" description:"Prints current version and exits"`
|
|
}{}
|
|
|
|
version = "dev"
|
|
)
|
|
|
|
func init() {
|
|
rconfig.AutoEnv(true)
|
|
if err := rconfig.ParseAndValidate(&cfg); err != nil {
|
|
log.Fatalf("Unable to parse commandline options: %s", err)
|
|
}
|
|
|
|
if cfg.VersionAndExit {
|
|
fmt.Printf("share %s\n", version)
|
|
os.Exit(0)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
switch {
|
|
|
|
case cfg.Bootstrap:
|
|
if err := doBootstrap(); err != nil {
|
|
log.WithError(err).Fatal("Bootstrap failed")
|
|
}
|
|
log.Info("Bucket bootstrap finished: Frontend uploaded")
|
|
|
|
case cfg.Listen != "":
|
|
if err := doListen(); err != nil {
|
|
log.WithError(err).Fatal("HTTP server ended unclean")
|
|
}
|
|
|
|
default:
|
|
if err := doCLIUpload(); err != nil {
|
|
log.WithError(err).Fatal("Upload failed")
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
func doCLIUpload() error {
|
|
if len(rconfig.Args()) == 1 {
|
|
return errors.New("Missing argument: File to upload")
|
|
}
|
|
|
|
if cfg.BaseURL == "" {
|
|
log.Warn("No BaseURL configured, output will be no complete URL")
|
|
}
|
|
|
|
var inFile io.ReadSeeker
|
|
|
|
inFileName := rconfig.Args()[1]
|
|
|
|
if inFileName == "-" {
|
|
inFileName = "stdin"
|
|
if cfg.ContentType == "" {
|
|
// If we don't have an explicitly set content-type assume stdin contains text
|
|
cfg.ContentType = "text/plain"
|
|
}
|
|
|
|
// Stdin is not seekable, so we need to buffer it
|
|
buf := new(bytes.Buffer)
|
|
if _, err := io.Copy(buf, os.Stdin); err != nil {
|
|
log.WithError(err).Fatal("Could not read stdin")
|
|
}
|
|
|
|
inFile = bytes.NewReader(buf.Bytes())
|
|
} else {
|
|
inFileHandle, err := os.Open(inFileName)
|
|
if err != nil {
|
|
return fmt.Errorf("Unable to open source file: %s", err)
|
|
}
|
|
defer inFileHandle.Close()
|
|
inFile = inFileHandle
|
|
}
|
|
|
|
url, err := executeUpload(inFileName, inFile, true, cfg.ContentType, false)
|
|
if err != nil {
|
|
return fmt.Errorf("Unable to upload file: %s", err)
|
|
}
|
|
|
|
fmt.Println(url)
|
|
return nil
|
|
}
|
|
|
|
func doBootstrap() error {
|
|
for _, asset := range []string{"index.html", "app.js"} {
|
|
if _, err := executeUpload(asset, bytes.NewReader(MustAsset("frontend/"+asset)), false, "", true); err != nil {
|
|
return fmt.Errorf("Unable to upload bootstrap asset %q: %s", asset, err)
|
|
}
|
|
}
|
|
return nil
|
|
}
|