github.com/sqlitebrowser/dio@v0.0.0-20240125125356-b587368e5c6b/cmd/licenceGet.go

package cmd

import (
	"errors"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"strings"

	rq "github.com/parnurzeal/gorequest"
	"github.com/spf13/cobra"
)

// Downloads a licence from a DBHub.io cloud.
var licenceGetCmd = &cobra.Command{
	Use:   "get [licence name]",
	Short: "Downloads the text for a licence from a DBHub.io cloud, saving it to [licence name].txt",
	RunE: func(cmd *cobra.Command, args []string) error {
		return licenceGet(args)
	},
}

func init() {
	licenceCmd.AddCommand(licenceGetCmd)
}

func licenceGet(args []string) error {
	// Ensure a licence name was given
	if len(args) == 0 {
		return errors.New("No licence name specified")
	}

	// Check for the presence of "all" as a licence name
	var licenceList []string
	var allFound bool
	for _, j := range args {
		if strings.ToLower(j) == "all" {
			allFound = true
		}
	}

	// If the all keyword was given, then assemble the full licence list.  Otherwise just use whatever was given
	// on the command line
	if allFound {
		l, err := getLicences()
		if err != nil {
			return errors.New(fmt.Sprintf("Error when retrieving list of all licences: %s", err))
		}
		for i := range l {
			licenceList = append(licenceList, i)
		}
	} else {
		licenceList = args
	}

	// Download the licence text
	dlStatus := make(map[string]string)
	for _, lic := range licenceList {
		resp, body, errs := rq.New().TLSClientConfig(&TLSConfig).Get(cloud+"/licence/get").
			Query(fmt.Sprintf("licence=%s", lic)).
			Set("User-Agent", fmt.Sprintf("Dio %s", DIO_VERSION)).
			End()
		if errs != nil {
			for _, err := range errs {
				log.Print(err.Error())
			}
			dlStatus[lic] = "Error when downloading licence text"
			continue
		}
		if resp.StatusCode != http.StatusOK {
			if resp.StatusCode == http.StatusNotFound {
				dlStatus[lic] = "Requested licence not found"
				continue
			}
			dlStatus[lic] = fmt.Sprintf("Download failed with an error: HTTP status %d - '%v'",
				resp.StatusCode, resp.Status)
			continue
		}

		// Write the licence to disk
		var ext string
		if resp.Header.Get("Content-Type") == "text/html" {
			ext = "html"
		} else {
			ext = "txt"
		}
		err := ioutil.WriteFile(fmt.Sprintf("%s.%s", lic, ext), []byte(body), 0644)
		if err != nil {
			dlStatus[lic] = err.Error()
		}
		dlStatus[lic] = fmt.Sprintf("Licence '%s.%s' downloaded", lic, ext)
	}

	// Display the status of the individual licence downloads
	_, err := fmt.Fprintf(fOut, "Downloading licences from: %s...\n\n", cloud)
	if err != nil {
		return err
	}
	for i, j := range dlStatus {
		_, err := fmt.Fprintf(fOut, "  * %s: %s\n", i, j)
		if err != nil {
			return err
		}
	}
	_, err = fmt.Fprintf(fOut, "\nCompleted\n")
	return err
}