Compare commits
No commits in common. "rewrite" and "master" have entirely different histories.
19
cmd/main.go
19
cmd/main.go
@ -1,19 +0,0 @@
|
|||||||
package main
|
|
||||||
|
|
||||||
import (
|
|
||||||
"log"
|
|
||||||
|
|
||||||
"github.com/mattburchett/go_telegram/pkg/core/config"
|
|
||||||
"github.com/mattburchett/go_telegram/pkg/service/telegram"
|
|
||||||
)
|
|
||||||
|
|
||||||
func main() {
|
|
||||||
conf, err := config.GetConfig("config.json")
|
|
||||||
if err != nil {
|
|
||||||
log.Fatal("Failed to read JSON.")
|
|
||||||
}
|
|
||||||
|
|
||||||
tgBot := telegram.Bot{}
|
|
||||||
tgBot.Config = conf
|
|
||||||
tgBot.New(conf.Telegram.Token)
|
|
||||||
}
|
|
||||||
17
go.mod
17
go.mod
@ -1,17 +0,0 @@
|
|||||||
module github.com/mattburchett/go_telegram
|
|
||||||
|
|
||||||
go 1.17
|
|
||||||
|
|
||||||
replace gopkg.in/tucnak/telebot.v2 => github.com/tucnak/telebot v0.0.0-20171121031619-29bd3707020c
|
|
||||||
|
|
||||||
require (
|
|
||||||
github.com/yanzay/tbot v1.0.0
|
|
||||||
github.com/yanzay/tbot/v2 v2.1.0
|
|
||||||
)
|
|
||||||
|
|
||||||
require (
|
|
||||||
github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible // indirect
|
|
||||||
github.com/smartystreets/goconvey v1.7.2 // indirect
|
|
||||||
github.com/technoweenie/multipartstreamer v1.0.1 // indirect
|
|
||||||
github.com/yanzay/log v0.0.0-20160419144809-87352bb23506 // indirect
|
|
||||||
)
|
|
||||||
23
go.sum
23
go.sum
@ -1,23 +0,0 @@
|
|||||||
github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible h1:2cauKuaELYAEARXRkq2LrJ0yDDv1rW7+wrTEdVL3uaU=
|
|
||||||
github.com/go-telegram-bot-api/telegram-bot-api v4.6.4+incompatible/go.mod h1:qf9acutJ8cwBUhm1bqgz6Bei9/C/c93FPDljKWwsOgM=
|
|
||||||
github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1 h1:EGx4pi6eqNxGaHF6qqu48+N2wcFQ5qg5FXgOdqsJ5d8=
|
|
||||||
github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
|
|
||||||
github.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=
|
|
||||||
github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
|
|
||||||
github.com/smartystreets/assertions v1.2.0 h1:42S6lae5dvLc7BrLu/0ugRtcFVjoJNMC/N3yZFZkDFs=
|
|
||||||
github.com/smartystreets/assertions v1.2.0/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo=
|
|
||||||
github.com/smartystreets/goconvey v1.7.2 h1:9RBaZCeXEQ3UselpuwUQHltGVXvdwm6cv1hgR6gDIPg=
|
|
||||||
github.com/smartystreets/goconvey v1.7.2/go.mod h1:Vw0tHAZW6lzCRk3xgdin6fKYcG+G3Pg9vgXWeJpQFMM=
|
|
||||||
github.com/technoweenie/multipartstreamer v1.0.1 h1:XRztA5MXiR1TIRHxH2uNxXxaIkKQDeX7m2XsSOlQEnM=
|
|
||||||
github.com/technoweenie/multipartstreamer v1.0.1/go.mod h1:jNVxdtShOxzAsukZwTSw6MDx5eUJoiEBsSvzDU9uzog=
|
|
||||||
github.com/yanzay/log v0.0.0-20160419144809-87352bb23506 h1:RPrQRi67MUoUPg/3wwL9K9vKVWAB180e2WJgxC02CJE=
|
|
||||||
github.com/yanzay/log v0.0.0-20160419144809-87352bb23506/go.mod h1:x2hAcqPbZzQz79Fzr9xEe7BM73+UsZ3nBucx2u30oSk=
|
|
||||||
github.com/yanzay/tbot v1.0.0 h1:pKLHIvdiHRgU5iTfpWlgMTCWyD4vIQtAsOy7LCYMsAo=
|
|
||||||
github.com/yanzay/tbot v1.0.0/go.mod h1:rS36e4e2P56jkI0bUuZtzjBBIwMIdBBTUHfI4tgG5BM=
|
|
||||||
github.com/yanzay/tbot/v2 v2.1.0 h1:mppieSOIbzaCjp2en66Fz4unIJ57+aalQfdGbtYmaKg=
|
|
||||||
github.com/yanzay/tbot/v2 v2.1.0/go.mod h1:q0+8JblBq9tLAnKHdBIZsHwDvMS9TfO6mNfaAk1VrHg=
|
|
||||||
golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
|
|
||||||
golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg=
|
|
||||||
golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY=
|
|
||||||
golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
|
|
||||||
golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs=
|
|
||||||
@ -1,56 +0,0 @@
|
|||||||
package config
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"log"
|
|
||||||
"os"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Config - This struct will hold configuration components.
|
|
||||||
type Config struct {
|
|
||||||
Telegram struct {
|
|
||||||
Token string `json:"token"`
|
|
||||||
ChatID string `json:"chatId"`
|
|
||||||
Admins []int `json:"admins"`
|
|
||||||
AuthorizedChats []string `json:"authorizedChats"`
|
|
||||||
} `json:"telegram"`
|
|
||||||
|
|
||||||
Sonarr struct {
|
|
||||||
URL string `json:"url"`
|
|
||||||
APIKey string `json:"apiKey"`
|
|
||||||
SeasonLimit int `json:"seasonLimit"`
|
|
||||||
ProfileID int `json:"profileId"`
|
|
||||||
} `json:"sonarr"`
|
|
||||||
CouchPotato struct {
|
|
||||||
URL string `json:"url"`
|
|
||||||
APIKey string `json:"apiKey"`
|
|
||||||
ProfileID string `json:"profileId`
|
|
||||||
} `json:"couchpotato"`
|
|
||||||
}
|
|
||||||
|
|
||||||
//GetConfig gets the configuration values for the api using the file in the supplied configPath.
|
|
||||||
func GetConfig(configPath string) (Config, error) {
|
|
||||||
if _, err := os.Stat(configPath); os.IsNotExist(err) {
|
|
||||||
return Config{}, fmt.Errorf("could not find the config file at path %s", configPath)
|
|
||||||
}
|
|
||||||
log.Println("Loading Configuration File: " + configPath)
|
|
||||||
return loadConfigFromFile(configPath)
|
|
||||||
}
|
|
||||||
|
|
||||||
//if the config loaded from the file errors, no defaults will be loaded and the app will exit.
|
|
||||||
func loadConfigFromFile(configPath string) (conf Config, err error) {
|
|
||||||
file, err := os.Open(configPath)
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Error opening config file: %v", err)
|
|
||||||
} else {
|
|
||||||
defer file.Close()
|
|
||||||
|
|
||||||
err = json.NewDecoder(file).Decode(&conf)
|
|
||||||
if err != nil {
|
|
||||||
log.Printf("Error decoding config file: %v", err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return conf, err
|
|
||||||
}
|
|
||||||
@ -1 +0,0 @@
|
|||||||
package couchpotato
|
|
||||||
@ -1,87 +0,0 @@
|
|||||||
package couchpotato
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"net/http"
|
|
||||||
"net/url"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/mattburchett/go_telegram/pkg/core/config"
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
type couchpotatoSearch struct {
|
|
||||||
Movies []struct {
|
|
||||||
Title string `json:"original_title"`
|
|
||||||
Imdb string `json:"imdb"`
|
|
||||||
Year int `json:"year"`
|
|
||||||
InLibrary bool `json:"in_library"`
|
|
||||||
InWanted bool `json:"in_wanted"`
|
|
||||||
} `json:"movies"`
|
|
||||||
Success bool `json:"success"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type request struct {
|
|
||||||
ImdbID string `json:"imdbid"`
|
|
||||||
Title string `json:"title"`
|
|
||||||
Year int `json:"year"`
|
|
||||||
Requested bool `json:"requested"`
|
|
||||||
Downloaded bool `json:"downloaded"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (search couchpotatoSearch) Convert() []request {
|
|
||||||
requests := []request{}
|
|
||||||
for _, result := range search.Movies {
|
|
||||||
requests = append(requests, request{
|
|
||||||
ImdbID: result.Imdb,
|
|
||||||
Title: result.Title,
|
|
||||||
Year: result.Year,
|
|
||||||
Requested: result.InWanted,
|
|
||||||
Downloaded: result.InLibrary,
|
|
||||||
})
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
return requests
|
|
||||||
}
|
|
||||||
|
|
||||||
type response struct {
|
|
||||||
Button string `json:"button"`
|
|
||||||
Callback string `json:"callback"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Search performs the lookup actions within CouchPotato
|
|
||||||
func Search(m *tbot.Message, config config.Config) ([]response, error) {
|
|
||||||
searchLookup, err := http.Get(config.CouchPotato.URL + config.CouchPotato.APIKey + "/movie.search?q=" + url.QueryEscape(strings.TrimPrefix(strings.TrimPrefix(m.Text, "/m"), " ")))
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
search := couchpotatoSearch{}
|
|
||||||
|
|
||||||
searchData, err := ioutil.ReadAll(searchLookup.Body)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
requestJSON := json.Unmarshal(searchData, &search)
|
|
||||||
if requestJSON != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
requests := search.Convert()
|
|
||||||
|
|
||||||
responseData := []response{}
|
|
||||||
for _, r := range requests {
|
|
||||||
responseData = append(responseData,
|
|
||||||
response{
|
|
||||||
fmt.Sprintf("%v (%v)", r.Title, r.Year),
|
|
||||||
fmt.Sprintf("%v", r.ImdbID),
|
|
||||||
})
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
return responseData, err
|
|
||||||
}
|
|
||||||
@ -1,24 +0,0 @@
|
|||||||
package sonarr
|
|
||||||
|
|
||||||
import (
|
|
||||||
"io/ioutil"
|
|
||||||
"net/http"
|
|
||||||
|
|
||||||
"github.com/mattburchett/go_telegram/pkg/core/config"
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Status contains the Sonarr request for system status.
|
|
||||||
func Status(m *tbot.Message, config config.Config) (string, error) {
|
|
||||||
r, err := http.Get(config.Sonarr.URL + "system/status?apikey=" + config.Sonarr.APIKey)
|
|
||||||
if err != nil {
|
|
||||||
return "Failed to contact Sonarr for data", err
|
|
||||||
}
|
|
||||||
|
|
||||||
rd, err := ioutil.ReadAll(r.Body)
|
|
||||||
if err != nil {
|
|
||||||
return "Failed to read Sonarr status data.", err
|
|
||||||
}
|
|
||||||
|
|
||||||
return string(rd), err
|
|
||||||
}
|
|
||||||
@ -1,245 +0,0 @@
|
|||||||
package sonarr
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
|
||||||
"io/ioutil"
|
|
||||||
"net/http"
|
|
||||||
"net/url"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/mattburchett/go_telegram/pkg/core/config"
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Response holds the information needed for Telegram callback.
|
|
||||||
type response struct {
|
|
||||||
Button string `json:"button"`
|
|
||||||
Callback string `json:"callback"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type sonarrSearch []struct {
|
|
||||||
Title string `json:"title"`
|
|
||||||
SeasonCount int `json:"seasonCount"`
|
|
||||||
Year int `json:"year"`
|
|
||||||
TvdbID int `json:"tvdbId"`
|
|
||||||
Downloaded bool `json:"downloaded"`
|
|
||||||
QualityProfileID int `json:"qualityProfileId"`
|
|
||||||
TitleSlug string `json:"titleSlug"`
|
|
||||||
Images []struct {
|
|
||||||
CoverType string `json:"coverType"`
|
|
||||||
URL string `json:"url"`
|
|
||||||
} `json:"images"`
|
|
||||||
Seasons []struct {
|
|
||||||
SeasonNumber int `json:"seasonNumber"`
|
|
||||||
Monitored bool `json:"monitored"`
|
|
||||||
} `json:"seasons"`
|
|
||||||
ProfileID int `json:"profileId"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type sonarrAdd struct {
|
|
||||||
Title string `json:"title"`
|
|
||||||
TvdbID int `json:"tvdbId"`
|
|
||||||
QualityProfileID int `json:"qualityProfileId"`
|
|
||||||
TitleSlug string `json:"titleSlug"`
|
|
||||||
Images []struct {
|
|
||||||
CoverType string `json:"coverType"`
|
|
||||||
URL string `json:"url"`
|
|
||||||
} `json:"images"`
|
|
||||||
Seasons []struct {
|
|
||||||
SeasonNumber int `json:"seasonNumber"`
|
|
||||||
Monitored bool `json:"monitored"`
|
|
||||||
} `json:"seasons"`
|
|
||||||
ProfileID int `json:"profileId"`
|
|
||||||
RootFolderPath string `json:"rootFolderPath"`
|
|
||||||
AddOptions struct {
|
|
||||||
IgnoreEpisodesWithFiles bool `json:"ignoreEpisodesWithFiles"`
|
|
||||||
IgnoreEpisodesWithoutFiles bool `json:"ignoreEpisodesWithoutFiles"`
|
|
||||||
SearchForMIssingEpisodes bool `json:"searchForMissingEpisodes"`
|
|
||||||
} `json:"addOptions"`
|
|
||||||
}
|
|
||||||
|
|
||||||
type rootFolderLookup []struct {
|
|
||||||
Path string `json:"path"`
|
|
||||||
FreeSpace int64 `json:"freeSpace"`
|
|
||||||
TotalSpace int64 `json:"totalSpace"`
|
|
||||||
UnmappedFolders []struct {
|
|
||||||
Name string `json:"name"`
|
|
||||||
Path string `json:"path"`
|
|
||||||
} `json:"unmappedFolders"`
|
|
||||||
ID int `json:"id"`
|
|
||||||
}
|
|
||||||
|
|
||||||
// Search performs the lookup actions within Sonarr.
|
|
||||||
func Search(m *tbot.Message, config config.Config) ([]response, error) {
|
|
||||||
|
|
||||||
var remote sonarrSearch
|
|
||||||
var local sonarrSearch
|
|
||||||
|
|
||||||
// Perform series lookup
|
|
||||||
remoteLookup, err := http.Get(config.Sonarr.URL +
|
|
||||||
"series/lookup?apikey=" +
|
|
||||||
config.Sonarr.APIKey +
|
|
||||||
"&term=" +
|
|
||||||
url.QueryEscape(strings.TrimPrefix(strings.TrimPrefix(m.Text, "/s"), " ")))
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Perform series local database lookup.
|
|
||||||
localLookup, err := http.Get(config.Sonarr.URL + "series?apikey=" + config.Sonarr.APIKey)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Read remote and local Data
|
|
||||||
remoteData, err := ioutil.ReadAll(remoteLookup.Body)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
localData, err := ioutil.ReadAll(localLookup.Body)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Unmarshal remote and local JSON
|
|
||||||
remoteJSON := json.Unmarshal(remoteData, &remote)
|
|
||||||
if remoteJSON != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
localJSON := json.Unmarshal(localData, &local)
|
|
||||||
if localJSON != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// Check for downloaded items.
|
|
||||||
for r := range remote {
|
|
||||||
for l := range local {
|
|
||||||
if remote[r].TvdbID == local[l].TvdbID {
|
|
||||||
remote[r].Downloaded = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Form URLs and return to Telegram
|
|
||||||
responseData := []response{}
|
|
||||||
for k := range remote {
|
|
||||||
responseData = append(responseData,
|
|
||||||
response{
|
|
||||||
fmt.Sprintf("%v%v%v (%v) - %v Seasons",
|
|
||||||
seasonHandler(remote[k].SeasonCount, config.Sonarr.SeasonLimit),
|
|
||||||
downloadedHandler(remote[k].Downloaded),
|
|
||||||
remote[k].Title,
|
|
||||||
remote[k].Year,
|
|
||||||
remote[k].SeasonCount),
|
|
||||||
fmt.Sprintf("%d%s%s", remote[k].TvdbID,
|
|
||||||
strings.TrimSuffix(downloadedHandler(remote[k].Downloaded), " "),
|
|
||||||
strings.TrimSuffix(seasonHandler(remote[k].SeasonCount, config.Sonarr.SeasonLimit), " ")),
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
return responseData, err
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add will take the callback data and add the show to Sonarr.
|
|
||||||
func Add(callback string, config config.Config) string {
|
|
||||||
// Separate the TVDB ID
|
|
||||||
tvdbid := strings.TrimPrefix(strings.TrimSuffix(strings.TrimSuffix(callback, "+"), "*"), "tv_")
|
|
||||||
|
|
||||||
// Look it up, to gather the information needed and the title.
|
|
||||||
seriesLookup, err := http.Get(config.Sonarr.URL + "/series/lookup?apikey=" + config.Sonarr.APIKey + "&term=tvdb:" + tvdbid)
|
|
||||||
if err != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
|
|
||||||
seriesData, err := ioutil.ReadAll(seriesLookup.Body)
|
|
||||||
if err != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
|
|
||||||
series := sonarrSearch{}
|
|
||||||
seriesJSON := json.Unmarshal(seriesData, &series)
|
|
||||||
if seriesJSON != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
|
|
||||||
// If the "downloaded" asterisk is already in place, go ahead and return.
|
|
||||||
if strings.Contains(callback, "*") {
|
|
||||||
return fmt.Sprintf("%v has already been requested for download.", series[0].Title)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Gather the root folder location.
|
|
||||||
rootFolderLookupQuery, err := http.Get(config.Sonarr.URL + "/rootfolder?apikey=" + config.Sonarr.APIKey)
|
|
||||||
if err != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
rootFolderData, err := ioutil.ReadAll(rootFolderLookupQuery.Body)
|
|
||||||
if err != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
|
|
||||||
rootFolder := rootFolderLookup{}
|
|
||||||
rootFolderJSON := json.Unmarshal(rootFolderData, &rootFolder)
|
|
||||||
if rootFolderJSON != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
|
|
||||||
// Form the JSON needed for adding to Sonarr.
|
|
||||||
seriesAdd := sonarrAdd{
|
|
||||||
TvdbID: series[0].TvdbID,
|
|
||||||
Title: series[0].Title,
|
|
||||||
// QualityProfileID: series[0].QualityProfileID,
|
|
||||||
TitleSlug: series[0].TitleSlug,
|
|
||||||
Images: series[0].Images,
|
|
||||||
Seasons: series[0].Seasons,
|
|
||||||
RootFolderPath: rootFolder[0].Path,
|
|
||||||
ProfileID: config.Sonarr.ProfileID,
|
|
||||||
}
|
|
||||||
seriesAdd.AddOptions.IgnoreEpisodesWithFiles = false
|
|
||||||
seriesAdd.AddOptions.IgnoreEpisodesWithoutFiles = false
|
|
||||||
seriesAdd.AddOptions.SearchForMIssingEpisodes = true
|
|
||||||
|
|
||||||
// Post it to Sonarr to be added.
|
|
||||||
seriesAddJSON, err := json.Marshal(seriesAdd)
|
|
||||||
if err != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
seriesAddReq, err := http.Post(config.Sonarr.URL+"/series?apikey="+config.Sonarr.APIKey, "application/json", bytes.NewBuffer(seriesAddJSON))
|
|
||||||
if err != nil {
|
|
||||||
return err.Error()
|
|
||||||
}
|
|
||||||
|
|
||||||
if seriesAddReq.StatusCode != 201 {
|
|
||||||
return "There was an error processing this request."
|
|
||||||
}
|
|
||||||
return fmt.Sprintf("%s has been queued for download.", series[0].Title)
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// downloadHandler returns the proper string that should be shown in the Telegram response.
|
|
||||||
func downloadedHandler(downloaded bool) string {
|
|
||||||
if downloaded {
|
|
||||||
return "* "
|
|
||||||
}
|
|
||||||
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
// seasonHandler returns the proper string that should be shown in the Telegram response.
|
|
||||||
// it is also used to prevent non-admins from downloading shows with a high number of seasons.
|
|
||||||
func seasonHandler(seasonCount int, seasonLimit int) string {
|
|
||||||
if seasonLimit == 0 {
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
|
|
||||||
if seasonCount >= seasonLimit {
|
|
||||||
return "+ "
|
|
||||||
}
|
|
||||||
|
|
||||||
return ""
|
|
||||||
}
|
|
||||||
@ -1,36 +0,0 @@
|
|||||||
package telegram
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strconv"
|
|
||||||
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (tb *Bot) myID(m *tbot.Message) {
|
|
||||||
if tb.adminCheck(m.From.ID, false) {
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, strconv.Itoa(m.From.ID))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "You are not an authorized admin.")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (tb *Bot) chatID(m *tbot.Message) {
|
|
||||||
if tb.adminCheck(m.From.ID, false) {
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, m.Chat.ID)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "You are not an authorized admin.")
|
|
||||||
}
|
|
||||||
|
|
||||||
// adminCheck checks for valid bot admins.
|
|
||||||
func (tb *Bot) adminCheck(id int, callback bool) bool {
|
|
||||||
for _, admin := range tb.Config.Telegram.Admins {
|
|
||||||
if id == admin {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
@ -1,73 +0,0 @@
|
|||||||
package telegram
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Handler creates the active Telegram handlers.
|
|
||||||
func (tb *Bot) Handler() {
|
|
||||||
|
|
||||||
// Bot Healthcheck
|
|
||||||
tb.Bot.HandleMessage("/ping", func(m *tbot.Message) {
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "pong")
|
|
||||||
})
|
|
||||||
|
|
||||||
// telegram/couchpotato.go
|
|
||||||
tb.Bot.HandleMessage("/m", tb.couchpotatoSearch)
|
|
||||||
|
|
||||||
// telegram/sonar.go
|
|
||||||
tb.Bot.HandleMessage("/s", tb.sonarrSearch)
|
|
||||||
tb.Bot.HandleMessage("/admin sonarrStatus", tb.sonarrStatus)
|
|
||||||
|
|
||||||
// telegram/testhandler.go
|
|
||||||
tb.Bot.HandleMessage("/test", tb.testHandler)
|
|
||||||
|
|
||||||
// telegram/admin.go
|
|
||||||
tb.Bot.HandleMessage("/admin myID", tb.myID)
|
|
||||||
tb.Bot.HandleMessage("/admin chatID", tb.chatID)
|
|
||||||
|
|
||||||
// Help
|
|
||||||
tb.Bot.HandleMessage("/help$", tb.helpHandler)
|
|
||||||
tb.Bot.HandleMessage("/h$", tb.helpHandler)
|
|
||||||
|
|
||||||
// Callback Handler
|
|
||||||
tb.Bot.HandleCallback(tb.callbackHandler)
|
|
||||||
}
|
|
||||||
|
|
||||||
// callbackHandler handles callbacks.
|
|
||||||
func (tb *Bot) callbackHandler(cq *tbot.CallbackQuery) {
|
|
||||||
go func() {
|
|
||||||
tb.Client.AnswerCallbackQuery(cq.ID, tbot.OptText("Request received."))
|
|
||||||
tb.Client.DeleteMessage(tb.CallbackChatID, tb.CallbackMessageID)
|
|
||||||
}()
|
|
||||||
|
|
||||||
if strings.Contains(cq.Data, "tv_") {
|
|
||||||
tb.sonarrAdd(cq)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb.Client.SendMessage(tb.CallbackChatID, cq.Data)
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
func (tb *Bot) helpHandler(m *tbot.Message) {
|
|
||||||
if !tb.whitelistHandler(m) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "USAGE:\n\n/movie <Movie Name> or /m <Movie Name>\n/show <TV Show Name> or /s <TV Show Name>\n\nEXAMPLES:\n\n/s The Walking Dead\n/m Avatar")
|
|
||||||
}
|
|
||||||
|
|
||||||
func (tb *Bot) whitelistHandler(m *tbot.Message) bool {
|
|
||||||
for _, id := range tb.Config.Telegram.AuthorizedChats {
|
|
||||||
if id == m.Chat.ID {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "This bot is not authorized for use in this chat.")
|
|
||||||
return false
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,85 +0,0 @@
|
|||||||
package telegram
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/mattburchett/go_telegram/pkg/service/sonarr"
|
|
||||||
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Sonarr Search
|
|
||||||
func (tb *Bot) sonarrSearch(m *tbot.Message) {
|
|
||||||
if !tb.whitelistHandler(m) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
text := strings.TrimPrefix(strings.TrimPrefix(m.Text, "/s"), " ")
|
|
||||||
if len(text) == 0 {
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "You must specify a show. Type /help for help.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
request, err := sonarr.Search(m, tb.Config)
|
|
||||||
if err != nil {
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, err.Error())
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
inlineResponse := make([][]tbot.InlineKeyboardButton, 0)
|
|
||||||
for _, i := range request {
|
|
||||||
inlineResponse = append(inlineResponse, []tbot.InlineKeyboardButton{{
|
|
||||||
Text: i.Button,
|
|
||||||
CallbackData: "tv_" + i.Callback,
|
|
||||||
}})
|
|
||||||
}
|
|
||||||
|
|
||||||
if len(request) == 0 {
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "No results found, try harder.")
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
response, _ := tb.Client.SendMessage(m.Chat.ID, "Please select the show you would like to download.", tbot.OptInlineKeyboardMarkup(&tbot.InlineKeyboardMarkup{InlineKeyboard: inlineResponse}))
|
|
||||||
tb.CallbackMessageID = response.MessageID
|
|
||||||
tb.CallbackChatID = m.Chat.ID
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// sonarrAdd will perform the add requests to Sonarr.
|
|
||||||
func (tb *Bot) sonarrAdd(cq *tbot.CallbackQuery) {
|
|
||||||
if strings.Contains(cq.Data, "+") {
|
|
||||||
if tb.adminCheck(cq.From.ID, true) {
|
|
||||||
tb.Client.SendMessage(tb.CallbackChatID, sonarr.Add(cq.Data, tb.Config))
|
|
||||||
} else {
|
|
||||||
tb.Client.AnswerCallbackQuery(cq.ID, tbot.OptText("This request is over the season limit."))
|
|
||||||
}
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb.Client.SendMessage(tb.CallbackChatID, sonarr.Add(cq.Data, tb.Config))
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
// Admin Functions
|
|
||||||
|
|
||||||
// sonarrStatus queries Sonarr for it's system status information.
|
|
||||||
func (tb *Bot) sonarrStatus(m *tbot.Message) {
|
|
||||||
if !tb.whitelistHandler(m) {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
if tb.adminCheck(m.From.ID, false) {
|
|
||||||
request, err := sonarr.Status(m, tb.Config)
|
|
||||||
|
|
||||||
if err != nil {
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, fmt.Sprintf("%v: \n %v", request, err))
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, "Sonarr Status:")
|
|
||||||
tb.Client.SendMessage(m.Chat.ID, request)
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,37 +0,0 @@
|
|||||||
package telegram
|
|
||||||
|
|
||||||
import (
|
|
||||||
"log"
|
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/mattburchett/go_telegram/pkg/core/config"
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Bot contains all the necessary bot and callback information.
|
|
||||||
type Bot struct {
|
|
||||||
Client *tbot.Client
|
|
||||||
Config config.Config
|
|
||||||
Bot *tbot.Server
|
|
||||||
CallbackChatID string
|
|
||||||
CallbackMessageID int
|
|
||||||
}
|
|
||||||
|
|
||||||
// Stat middleware.
|
|
||||||
func stat(h tbot.UpdateHandler) tbot.UpdateHandler {
|
|
||||||
return func(u *tbot.Update) {
|
|
||||||
start := time.Now()
|
|
||||||
h(u)
|
|
||||||
log.Printf("Handle time: %v", time.Now().Sub(start))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// New creates an active telegram bot and loads the handlers.
|
|
||||||
func (tb *Bot) New(token string) {
|
|
||||||
tb.Bot = tbot.New(token)
|
|
||||||
tb.Bot.Use(stat)
|
|
||||||
tb.Client = tb.Bot.Client()
|
|
||||||
tb.Handler()
|
|
||||||
tb.Bot.Start()
|
|
||||||
|
|
||||||
}
|
|
||||||
@ -1,25 +0,0 @@
|
|||||||
package telegram
|
|
||||||
|
|
||||||
import (
|
|
||||||
"strings"
|
|
||||||
|
|
||||||
"github.com/yanzay/tbot/v2"
|
|
||||||
)
|
|
||||||
|
|
||||||
func (tb *Bot) testHandler(m *tbot.Message) {
|
|
||||||
buttons := make([]string, 0)
|
|
||||||
buttons = append(buttons, "ping", "is", "stupid")
|
|
||||||
|
|
||||||
inline2 := make([][]tbot.InlineKeyboardButton, 0)
|
|
||||||
|
|
||||||
for _, i := range buttons {
|
|
||||||
inline2 = append(inline2, []tbot.InlineKeyboardButton{{
|
|
||||||
Text: i,
|
|
||||||
CallbackData: i,
|
|
||||||
}})
|
|
||||||
}
|
|
||||||
|
|
||||||
msg, _ := tb.Client.SendMessage(m.Chat.ID, "Inline test. "+strings.TrimPrefix(m.Text, "/test "), tbot.OptInlineKeyboardMarkup(&tbot.InlineKeyboardMarkup{InlineKeyboard: inline2}))
|
|
||||||
tb.CallbackMessageID = msg.MessageID
|
|
||||||
tb.CallbackChatID = m.Chat.ID
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue
Block a user