mirror of
https://github.com/vbatts/imgsrv.git
synced 2024-12-25 15:26:30 +00:00
170 lines
4 KiB
Go
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)")
|
|
|
|
}
|