DockerFiles/transmission-telegram
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

init telegram

Browse Source
This commit is contained in:
pyed 2016-06-19 22:02:15 +03:00
parent 9f56711ea0
commit 844e316b24
1 changed files with 39 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
transmission-telegram.go
View File

@ -6,19 +6,28 @@ import (
"os"
"strings"
"gopkg.in/telegram-bot-api.v4"
"github.com/pyed/transmission"
)
var (
// flags
token string
master string
url string
username string
password string
client *transmission.Client
// transmission
Client *transmission.Client
// telegram
Bot *tgbotapi.BotAPI
Updates <-chan tgbotapi.Update
)
// init flags
func init() {
// define arguments and parse them.
flag.StringVar(&token, "token", "", "Telegram bot token")
@ -42,7 +51,10 @@ func init() {
flag.Usage()
os.Exit(1)
}
}
// init transmission
func init() {
// set transmission.Config, needed to establish a connection with transmission
conf := transmission.Config{
Address: url,
@ -51,19 +63,40 @@ func init() {
}
// transmission.New() never returns an error, we will ignore it and test with client.Session.Update()
client, _ = transmission.New(conf)
if err := client.Session.Update(); err != nil {
Client, _ = transmission.New(conf)
if err := Client.Session.Update(); err != nil {
// try to predict the error message, as it vague coming from pyed/transmission
if strings.HasPrefix(err.Error(), "invalid character") { // means the user or the pass is wrong.
fmt.Fprintln(os.Stderr, "Transmission's Username or Password is wrong.\n")
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.Fprintln(os.Stderr, "Make sure to pass the right full RPC URL e.g. http://localhost:9091/transmission/rpc\n")
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
fmt.Fprintln(os.Stderr, "RawError: "+err.Error())
fmt.Fprintf(os.Stderr, "JSONError: %s\n", err)
os.Exit(1)
}
}
// init telegram
func init() {
// authorize using the token
var err error
Bot, err = tgbotapi.NewBotAPI(token)
if err != nil {
fmt.Fprintf(os.Stderr, "Telegram Error: %s\n", err)
os.Exit(1)
}
// get a channel and sign it to 'Updates'
u := tgbotapi.NewUpdate(0)
u.Timeout = 60
Updates, err = Bot.GetUpdatesChan(u)
if err != nil {
fmt.Fprintf(os.Stderr, "Telegram Error: %s\n", err)
os.Exit(1)
}
}