1
0
Fork 0
mirror of https://github.com/vbatts/imgsrv.git synced 2024-12-24 23:06:31 +00:00
imgsrv/imgsrv.go
Vincent Batts f768d2f9f6
*: bubble the dbhandler up to cli
Signed-off-by: Vincent Batts <vbatts@hashbangbash.com>
2017-04-07 13:42:12 -04:00

170 lines
4 KiB
Go

package main
/*
Fun Times. This is to serve as a single utility that
* can be an image server, that stores into a mongo backend,
OR
* the client side tool that pushes/pulls images to the running server.
*/
import (
"flag"
"fmt"
"log"
"net/url"
"os"
"github.com/vbatts/imgsrv/client"
"github.com/vbatts/imgsrv/config"
"github.com/vbatts/imgsrv/util"
)
var (
ConfigFile = fmt.Sprintf("%s/.imgsrv.yaml", os.Getenv("HOME"))
DefaultConfig = &config.Config{
Server: false,
Ip: "0.0.0.0",
Port: "7777",
DbHandler: "mongo",
MongoHost: "localhost",
MongoDbName: "filesrv",
MongoUsername: "",
MongoPassword: "",
RemoteHost: "",
}
PutFile = ""
FetchUrl = ""
FileKeywords = ""
)
func main() {
flag.Parse()
for _, arg := range flag.Args() {
// TODO What to do with these floating args ...
// Assume they're files and upload them?
log.Printf("%s", arg)
}
// loads either default or flag specified config
// to override variables
if c, err := config.ReadConfigFile(ConfigFile); err == nil {
DefaultConfig.Merge(c)
}
if DefaultConfig.Server {
// Run the server!
runServer(DefaultConfig)
} else if len(FetchUrl) > 0 {
// not sure that this ought to be exposed in the client tool
file, err := util.FetchFileFromURL(FetchUrl)
if err != nil {
log.Println(err)
return
}
log.Println(file)
} else {
// we're pushing up a file
if len(DefaultConfig.RemoteHost) == 0 {
log.Println("Please provide a remotehost!")
return
}
if len(PutFile) == 0 { //&& len(flag.Args()) == 0) {
log.Println("Please provide files to be uploaded!")
return
}
params := map[string]string{}
if len(FileKeywords) > 0 {
params["keywords"] = FileKeywords
} else {
log.Println("WARN: you didn't provide any keywords :-(")
}
url, err := url.Parse(DefaultConfig.RemoteHost + "/f/")
if err != nil {
log.Println(err)
return
}
url_path, err := client.PutFileFromPath(url.String(), PutFile, params)
if err != nil {
log.Println(err)
return
}
fmt.Printf("%s%s\n", DefaultConfig.RemoteHost, url_path)
}
}
/*
http://golang.org/doc/effective_go.html#init
TODO refactor flags and config, to assign, instead of pass by reference.
http://play.golang.org/p/XhGqn-MOjL
*/
func init() {
flag.StringVar(&ConfigFile,
"config",
ConfigFile,
"Provide alternate configuration file")
/* Server-side */
flag.BoolVar(&DefaultConfig.Server,
"server",
DefaultConfig.Server,
"Run as an image server (defaults to client-side)")
flag.StringVar(&DefaultConfig.Ip,
"ip",
DefaultConfig.Ip,
"IP to bind to (if running as a server)('ip' in the config)")
flag.StringVar(&DefaultConfig.Port,
"port",
DefaultConfig.Port,
"Port to listen on (if running as a server)('port' in the config)")
flag.StringVar(&DefaultConfig.DbHandler,
"dbhandler",
DefaultConfig.DbHandler,
"Database backend handler (if runnint as a server)")
/* MongoDB settings */
flag.StringVar(&DefaultConfig.MongoHost,
"mongo-host",
DefaultConfig.MongoHost,
"Mongo Host to connect to ('mongohost' in the config)")
flag.StringVar(&DefaultConfig.MongoDbName,
"mongo-db",
DefaultConfig.MongoDbName,
"Mongo db to connect to ('mongodb' in the config)")
flag.StringVar(&DefaultConfig.MongoUsername,
"mongo-username",
DefaultConfig.MongoUsername,
"Mongo username to auth with (if needed) ('mongousername' in the config)")
flag.StringVar(&DefaultConfig.MongoPassword,
"mongo-password",
DefaultConfig.MongoPassword,
"Mongo password to auth with (if needed) ('mongopassword' in the config)")
/* Client-side */
flag.StringVar(&FetchUrl,
"fetch",
FetchUrl,
"Just fetch the file from this url")
flag.StringVar(&DefaultConfig.RemoteHost,
"remotehost",
DefaultConfig.RemoteHost,
"Remote host to get/put files on ('remotehost' in the config)")
flag.StringVar(&PutFile,
"put",
PutFile,
"Put file on remote server (needs -remotehost)")
flag.StringVar(&FileKeywords,
"keywords",
FileKeywords,
"Keywords to associate with file. (comma delimited) (needs -put)")
}