gddo-admin/main.go

// Copyright 2013 The Go Authors. All rights reserved.
//
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file or at
// https://developers.google.com/open-source/licenses/bsd.

// Command gddo-admin is the GoDoc.org command line administration tool.
package main

import (
	"flag"
	"fmt"
	"os"
	"strings"
	"time"
)

type command struct {
	name  string
	run   func(c *command)
	flag  flag.FlagSet
	usage string
}

func (c *command) printUsage() {
	fmt.Fprintf(os.Stderr, "%s %s\n", os.Args[0], c.usage)
	c.flag.PrintDefaults()
}

var (
	project       = flag.String("project", "", "App Engine project ID used to interact with remote API.")
	redisServer   = flag.String("db-server", "redis://127.0.0.1:6379", "URI of Redis server.")
	dbIdleTimeout = flag.Duration("db-idle-timeout", 250*time.Second, "Close database connections after remaining idle for this duration.")
)

var gaeEndpoint string

var commands = []*command{
	blockCommand,
	reindexCommand,
	deleteCommand,
	popularCommand,
	dangleCommand,
	crawlCommand,
	statsCommand,
}

func printUsage() {
	var n []string
	for _, c := range commands {
		n = append(n, c.name)
	}
	fmt.Fprintf(os.Stderr, "%s %s\n", os.Args[0], strings.Join(n, "|"))
	flag.PrintDefaults()
	for _, c := range commands {
		c.printUsage()
	}
}

func setDefaults() {
	if *project != "" {
		gaeEndpoint = fmt.Sprintf("serviceproxy-dot-%s.appspot.com", *project)
	}
}

func main() {
	flag.Usage = printUsage
	flag.Parse()
	args := flag.Args()
	setDefaults()

	if len(args) >= 1 {
		for _, c := range commands {
			if args[0] == c.name {
				c.flag.Usage = func() {
					c.printUsage()
					os.Exit(2)
				}
				c.flag.Parse(args[1:])
				c.run(c)
				return
			}
		}
	}
	printUsage()
	os.Exit(2)
}