commands skeleton

transmission-telegram.go

@@ -11,13 +11,15 @@ import (
 	"github.com/pyed/transmission"
 )
 
+const VERSION = "0.0"
+
 var (
 	// flags
-	token    string
-	master   string
-	url      string
-	username string
-	password string
+	BotToken string
+	Master   string
+	RpcUrl   string
+	Username string
+	Password string
 
 	// transmission
 	Client *transmission.Client
@@ -30,11 +32,11 @@ var (
 // init flags
 func init() {
 	// define arguments and parse them.
-	flag.StringVar(&token, "token", "", "Telegram bot token")
-	flag.StringVar(&master, "master", "", "Your telegram handler, So the bot will only respond to you")
-	flag.StringVar(&url, "url", "http://localhost:9091/transmission/rpc", "Transmission RPC URL")
-	flag.StringVar(&username, "username", "", "Transmission username")
-	flag.StringVar(&password, "password", "", "Transmission password")
+	flag.StringVar(&BotToken, "token", "", "Telegram bot token")
+	flag.StringVar(&Master, "master", "", "Your telegram handler, So the bot will only respond to you")
+	flag.StringVar(&RpcUrl, "url", "http://localhost:9091/transmission/rpc", "Transmission RPC URL")
+	flag.StringVar(&Username, "username", "", "Transmission username")
+	flag.StringVar(&Password, "password", "", "Transmission password")
 
 	// set the usage message
 	flag.Usage = func() {
@@ -45,8 +47,8 @@ func init() {
 	flag.Parse()
 
 	// make sure that we have the two madatory arguments: telegram token & master's handler.
-	if token == "" ||
-		master == "" {
+	if BotToken == "" ||
+		Master == "" {
 		fmt.Fprintf(os.Stderr, "Error: Mandatory argument missing!\n\n")
 		flag.Usage()
 		os.Exit(1)
@@ -57,9 +59,9 @@ func init() {
 func init() {
 	// set transmission.Config, needed to establish a connection with transmission
 	conf := transmission.Config{
-		Address:  url,
-		User:     username,
-		Password: password,
+		Address:  RpcUrl,
+		User:     Username,
+		Password: Password,
 	}
 
 	// transmission.New() never returns an error, we will ignore it and test with client.Session.Update()
@@ -71,7 +73,7 @@ func init() {
 			fmt.Fprintf(os.Stderr, "Transmission's Username or Password is wrong.\n\n")
 
 		} else { // any other error is probaby because of the URL
-			fmt.Fprintf(os.Stderr, "Error: Couldn't connect to: %s\n", url)
+			fmt.Fprintf(os.Stderr, "Error: Couldn't connect to: %s\n", RpcUrl)
 			fmt.Fprintf(os.Stderr, "Make sure to pass the right full RPC URL e.g. http://localhost:9091/transmission/rpc\n\n")
 		}
 		// send the vague error message too
@@ -84,7 +86,7 @@ func init() {
 func init() {
 	// authorize using the token
 	var err error
-	Bot, err = tgbotapi.NewBotAPI(token)
+	Bot, err = tgbotapi.NewBotAPI(BotToken)
 	if err != nil {
 		fmt.Fprintf(os.Stderr, "Telegram Error: %s\n", err)
 		os.Exit(1)
@@ -101,4 +103,63 @@ func init() {
 	}
 }
 
-func main() {}
+func main() {
+	for update := range Updates {
+		// ignore anyone other than 'master'
+		if update.Message.From.UserName != Master {
+			continue
+		}
+		// ignore edited messages
+		if update.Message == nil {
+			continue
+		}
+
+		// tokenize the update
+		tokens := strings.Split(update.Message.Text, " ")
+		command := strings.ToLower(tokens[0])
+
+		switch command {
+		case "list", "/list":
+			// list torrents
+		case "downs", "/downs":
+			// list downloading
+		case "active", "/active":
+			// list active torrents
+		case "errors", "/errors":
+			// list torrents with errors
+		case "trackers", "/trackers":
+			// list trackers
+		case "add", "/add":
+			// takes url to a torrent to add it
+		case "search", "/search":
+			// search for a torrent
+		case "latest", "/latest":
+			// get the latest torrents
+		case "info", "/info":
+			// gets info on specific torrent
+		case "stop", "/stop":
+			// stop one torrent or more
+		case "stopall", "/stopall":
+			// stops all the torrents
+		case "start", "/start":
+			// starts one torrent or more
+		case "startall", "/startall":
+			// starts all the torrents
+		case "stats", "/stats":
+			// print transmission stats
+		case "speed", "/speed":
+			// print current download and upload speeds
+		case "del", "/del":
+			// deletes a torrent but keep its data
+		case "deldata", "/deldata":
+			// deletes a torrents and its data
+		case "help", "/help":
+			// prints a help message
+		case "version", "/version":
+			// print transmission and transmission-telegram versions
+		default:
+			// no such command, try help
+
+		}
+	}
+}