New mattermost's model API

[mmfileget.git] / main.go

diff --git a/main.go b/main.go
index 38d2cb08d6434900b790d370a1856a106880fd78..a31bb51f758acc5abd77f058c0ee46ed9d8ca35c 100644 (file)
--- a/main.go
+++ b/main.go
@@ -3,10 +3,11 @@ package main
 import (
        "fmt"
        "io/ioutil"
+       "log"
        "os"
        "strings"
 
-       "github.com/mattermost/mattermost-server/v5/model"
+       "github.com/mattermost/mattermost-server/v6/model"
 )
 
 func main() {
@@ -20,9 +21,15 @@ func main() {
                os.Exit(1)
        }
 
-       Client := model.NewAPIv4Client("https://" + machine)
-       Client.Login(login, password)
-       info, resp := Client.GetFileInfo(fileId)
+       client := model.NewAPIv4Client("https://" + machine)
+       _, _, err := client.Login(login, password)
+       if err != nil {
+               log.Fatalln(err)
+       }
+       info, resp, err := client.GetFileInfo(fileId)
+       if err != nil {
+               log.Fatalln(err)
+       }
        if info == nil {
                fmt.Fprintln(os.Stderr, resp)
                os.Exit(1)
@@ -37,7 +44,10 @@ func main() {
                fmt.Fprintln(os.Stderr, "Download? Ctrl-D/C")
                os.Stdin.Read(make([]byte, 1))
        }
-       data, _ := Client.GetFile(fileId)
+       data, _, err := client.GetFile(fileId)
+       if err != nil {
+               log.Fatalln(err)
+       }
        if err := ioutil.WriteFile(filename, data, os.FileMode(0666)); err != nil {
                panic(err)
        }