imgsrv/imgsrv.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"
  "labix.org/v2/mgo"
  "log"
  "net/url"
  "os"
  "path/filepath"
)

var (
  ConfigFile   = fmt.Sprintf("%s/.imgsrv.yaml", os.Getenv("HOME"))

  DefaultRunAsServer    = false
  RunAsServer           = DefaultRunAsServer

  DefaultServerIP       = "0.0.0.0"
  ServerIP              = DefaultServerIP
  DefaultServerPort     = "7777"
  ServerPort            = DefaultServerPort
  DefaultMongoHost      = "localhost"
  MongoHost             = DefaultMongoHost
  DefaultMongoDB        = "filesrv"
  MongoDB               = DefaultMongoDB
  DefaultMongoUsername  = ""
  MongoUsername         = DefaultMongoUsername
  DefaultMongoPassword  = ""
  MongoPassword         = DefaultMongoPassword

  mongo_session *mgo.Session
  images_db *mgo.Database
  gfs *mgo.GridFS

  DefaultRemoteHost     = ""
  RemoteHost            = DefaultRemoteHost
  PutFile               = ""
  FetchUrl              = ""
  FileKeywords          = ""
)

func main() {
  flag.Parse()
  for _, arg := range flag.Args() {
    // What to do with these floating args ...
    log.Printf("%s", arg)
  }

  // loads either default or flag specified config
  // to override variables
  loadConfiguration(ConfigFile)

  if (len(FetchUrl) > 0) {
    file, err := FetchFileFromURL(FetchUrl)
    if (err != nil) {
      log.Println(err)
      return
    }
    log.Println(file)
  } else if (RunAsServer) {
    log.Printf("%s", ServerIP)
    runServer(ServerIP,ServerPort)
  } else {
    if (len(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
    }
    _,basename := filepath.Split(PutFile)
    queryParams := "?filename=" + basename
    if (len(FileKeywords) > 0) {
      queryParams = queryParams + "&keywords=" + FileKeywords
    } else {
      log.Println("WARN: you didn't provide any keywords :-(")
    }
    url, err := url.Parse(RemoteHost + "/f/" + queryParams)
    if (err != nil) {
      log.Println(err)
      return
    }
    log.Printf("POSTing: %s\n", url.String())
    url_path, err := PutFileFromPath(url.String(), PutFile)
    if (err != nil) {
      log.Println(err)
      return
    }
    log.Printf("New Image!: %s%s\n", RemoteHost, url_path)
  }
}

/* http://golang.org/doc/effective_go.html#init */
func init() {
  flag.StringVar(&ConfigFile,
       "config",
       ConfigFile,
       "Provide alternate configuration file")

  /* Server-side */
  flag.BoolVar(&RunAsServer,
       "server",
       RunAsServer,
       "Run as an image server (defaults to client-side)")
  flag.StringVar(&ServerIP,
       "ip",
       ServerIP,
       "IP to bind to (if running as a server)('ip' in the config)")
  flag.StringVar(&ServerPort,
       "port",
       ServerPort,
       "Port to listen on (if running as a server)('port' in the config)")

  /* MongoDB settings */
  flag.StringVar(&MongoHost,
       "mongo-host",
       MongoHost,
       "Mongo Host to connect to ('mongohost' in the config)")
  flag.StringVar(&MongoDB,
       "mongo-db",
       MongoDB,
       "Mongo db to connect to ('mongodb' in the config)")
  flag.StringVar(&MongoUsername,
       "mongo-username",
       MongoUsername,
       "Mongo username to auth with (if needed) ('mongousername' in the config)")
  flag.StringVar(&MongoPassword,
       "mongo-password",
       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(&RemoteHost,
       "remotehost",
       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)")

}

func loadConfiguration(filename string) (c Config) {
  //log.Printf("Attempting to load config file: %s", filename)
  c, err := ReadConfigFile(filename)
  if (err != nil) {
    //log.Println(err)
    return Config{}
  }

  cRunAsServer := c.GetBool("server")
  cServerIp := c.GetString("ip")
  cServerPort := c.GetString("port")
  cMongoHost := c.GetString("mongohost")
  cMongoDB := c.GetString("mongodb")
  cMongoUsername := c.GetString("mongousername")
  cMongoPassword := c.GetString("mongopassword")
  cRemoteHost := c.GetString("remotehost")

  // Only set variables from config file,
  // if they weren't passed as flags
  if (DefaultRunAsServer == RunAsServer && cRunAsServer) {
    RunAsServer = cRunAsServer
  }
  if (DefaultServerIP == ServerIP && len(cServerIp) > 0) {
    ServerIP = cServerIp
  }
  if (DefaultServerPort == ServerPort && len(cServerPort) > 0) {
    ServerPort = cServerPort
  }
  if (DefaultMongoHost == MongoHost && len(cMongoHost) > 0) {
    MongoHost = cMongoHost
  }
  if (DefaultMongoDB == MongoDB && len(cMongoDB) > 0) {
    MongoDB = cMongoDB
  }
  if (DefaultMongoUsername == MongoUsername && len(cMongoUsername) > 0) {
    MongoUsername = cMongoUsername
  }
  if (DefaultMongoPassword == MongoPassword && len(cMongoPassword) > 0) {
    MongoPassword = cMongoPassword
  }
  if (DefaultRemoteHost == RemoteHost && len(cRemoteHost) > 0) {
    RemoteHost = cRemoteHost
  }

  return c
}