Fix downloading files from telegram

ilyaglow · Jun 1, 2018 · e63827c · e63827c
1 parent 955c365
commit e63827c
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/process.go b/process.go
@@ -27,7 +27,7 @@ func (c *Client) ProcessCortex(input *tgbotapi.Message) error {
 			return err
 		}
 
-		j, err = newFileArtifactFromURL(link, c.TLP)
+		j, err = newFileArtifactFromURL(link, c.TLP, c.Bot.Client)
 		if err != nil {
 			log.Println(err.Error())
 			return err
@@ -116,8 +116,13 @@ func newArtifact(s string, tlp int) (cortex.Observable, error) {
 }
 
 // newFileArtifactFromURL makes a FileArtifact from URL
-func newFileArtifactFromURL(link string, tlp int) (cortex.Observable, error) {
-	resp, err := http.Get(link)
+func newFileArtifactFromURL(link string, tlp int, client *http.Client) (cortex.Observable, error) {
+	req, err := http.NewRequest("GET", link, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	resp, err := client.Do(req)
 	if err != nil {
 		return nil, err
 	}