package main

import (
	"fmt"
	"io/ioutil"
	"log"
	"os"
	"strings"

	"github.com/mattermost/mattermost-server/v6/model"
)

func main() {
	url := strings.TrimPrefix(os.Args[1], "-") // -?https://MACHINE/api/v4/files/FILE
	s := strings.Split(url, "/")
	machine, fileId := s[2], s[6]

	login, password := findInNetrc(machine)
	if login == "" || password == "" {
		fmt.Fprintln(os.Stderr, "No credentials found for:", machine)
		os.Exit(1)
	}

	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)
	}
	filename := info.Name
	if len(os.Args) > 2 {
		filename = os.Args[2]
	} else {
		fmt.Fprintln(os.Stderr, "Name:", info.Name)
		fmt.Fprintln(os.Stderr, "Type:", info.MimeType)
		fmt.Fprintln(os.Stderr, "Size:", info.Size)
		fmt.Fprintln(os.Stderr, "Download? Ctrl-D/C")
		os.Stdin.Read(make([]byte, 1))
	}
	data, _, err := client.GetFile(fileId)
	if err != nil {
		log.Fatalln(err)
	}
	if err := ioutil.WriteFile(filename, data, os.FileMode(0666)); err != nil {
		panic(err)
	}
}