main.go

package main

import (
	"os"

	"github.com/codegangsta/cli"
	"github.com/sirupsen/logrus"
)

var (
	logger *logrus.Logger
)

func init() {
	logger = logrus.New()
	logger.Level = logrus.DebugLevel
	logger.Out = os.Stdout
}

func main() {
	app := cli.NewApp()
	app.Name = "twitter-graph"
	app.Usage = "A docker log pump to splunk"
	app.Flags = []cli.Flag{
		cli.StringFlag{
			Name:   "host, d",
			Usage:  "Neo4j host",
			EnvVar: "NEO4_HOST",
			Value:  "localhost",
		},
		cli.IntFlag{
			Name:   "port, P",
			Usage:  "Neo4j port",
			EnvVar: "NEO4_PORT",
			Value:  7474,
		},
		cli.StringFlag{
			Name:   "user, u",
			Usage:  "Neo4j username",
			EnvVar: "NEO4_USERNAME",
		},
		cli.StringFlag{
			Name:   "password, p",
			Usage:  "Neo4j password",
			EnvVar: "NEO4_PASSWORD",
		},
		cli.StringFlag{
			Name:  "screenname, s",
			Usage: "Twitter screen name",
		},
		cli.StringFlag{
			Name:   "user-key",
			Usage:  "Twitter User Key",
			EnvVar: "TWITTER_USER_KEY",
		},
		cli.StringFlag{
			Name:   "user-secret",
			Usage:  "Twitter User Secret",
			EnvVar: "TWITTER_USER_SECRET",
		},
		cli.StringFlag{
			Name:   "consumer-key",
			Usage:  "Twitter Consumer Key",
			EnvVar: "TWITTER_CONSUMER_KEY",
		},
		cli.StringFlag{
			Name:   "consumer-secret",
			Usage:  "Twitter Consumer Secret",
			EnvVar: "TWITTER_CONSUMER_SECRET",
		},
	}

	app.Commands = []cli.Command{
		cli.Command{
			Name: "add",
			Subcommands: []cli.Command{
				cli.Command{
					Name: "followers",
					Action: func(ctx *cli.Context) {
						exec(ctx, func(eng *Engine) error {
							return eng.AddFollowers()
						})
					},
				},
				cli.Command{
					Name: "mentions",
					Action: func(ctx *cli.Context) {
						exec(ctx, func(eng *Engine) error {
							return eng.AddMentions()
						})
					},
				},
				cli.Command{
					Name: "tweets",
					Action: func(ctx *cli.Context) {
						exec(ctx, func(eng *Engine) error {
							return eng.AddTweets()
						})
					},
				},
			},
		},
		cli.Command{
			Name: "maintain",
			Subcommands: []cli.Command{
				cli.Command{
					Name: "users",
					Action: func(ctx *cli.Context) {
						exec(ctx, func(eng *Engine) error {
							return eng.CompleteUsers()
						})
					},
				},
				cli.Command{
					Name: "following",
					Action: func(ctx *cli.Context) {
						exec(ctx, func(eng *Engine) error {
							return eng.MaintainFollowing()
						})
					},
				},
				cli.Command{
					Name: "followers",
					Action: func(ctx *cli.Context) {
						exec(ctx, func(eng *Engine) error {
							return eng.MaintainFollowers()
						})
					},
				},
			},
		},
	}

	app.Run(os.Args)
}