ephemerup/upctl/lib/output.go

/*
Copyright © 2023 Thomas von Dein

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/

package lib

import (
	"encoding/json"
	"errors"
	"fmt"
	"github.com/imroc/req/v3"
	"github.com/olekukonko/tablewriter"
	"github.com/tlinden/cenophane/common"
	"os"
	"time"
)

// make a human readable version of the expire setting
func prepareExpire(expire string, start common.Timestamp) string {
	switch expire {
	case "asap":
		return "On first access"
	default:
		return time.Unix(start.Unix()+int64(common.Duration2int(expire)), 0).Format("2006-01-02 15:04:05")
	}

	return ""
}

// generic table writer
func WriteTable(headers []string, data [][]string) {
	table := tablewriter.NewWriter(os.Stdout)

	table.SetHeader(headers)
	table.AppendBulk(data)

	table.SetAutoWrapText(false)
	table.SetAutoFormatHeaders(true)
	table.SetHeaderAlignment(tablewriter.ALIGN_LEFT)
	table.SetAlignment(tablewriter.ALIGN_LEFT)
	table.SetCenterSeparator("")
	table.SetColumnSeparator("")
	table.SetRowSeparator("")
	table.SetHeaderLine(false)
	table.SetBorder(false)
	table.SetTablePadding("\t")
	table.SetNoWhiteSpace(true)

	table.Render()
}

// output like psql \x
func WriteExtended(uploads *common.Uploads) {
	format := fmt.Sprintf("%%%ds: %%s\n", Maxwidth)

	// we shall only have 1 element, however, if we ever support more, here we go
	for _, entry := range uploads.Entries {
		expire := prepareExpire(entry.Expire, entry.Uploaded)
		fmt.Printf(format, "Id", entry.Id)
		fmt.Printf(format, "Expire", expire)
		fmt.Printf(format, "Context", entry.Context)
		fmt.Printf(format, "Uploaded", entry.Uploaded)
		fmt.Printf(format, "Filename", entry.File)
		fmt.Printf(format, "Url", entry.Url)
		fmt.Println()
	}
}

// extract an common.Uploads{} struct from json response
func GetUploadsFromResponse(resp *req.Response) (*common.Uploads, error) {
	uploads := common.Uploads{}

	if err := json.Unmarshal([]byte(resp.String()), &uploads); err != nil {
		return nil, errors.New("Could not unmarshall JSON response: " + err.Error())
	}

	if !uploads.Success {
		return nil, errors.New(uploads.Message)
	}

	return &uploads, nil
}

// turn the Uploads{} struct into a table and print it
func RespondTable(resp *req.Response) error {
	uploads, err := GetUploadsFromResponse(resp)
	if err != nil {
		return err
	}

	if uploads.Message != "" {
		fmt.Println(uploads.Message)
	}

	// tablewriter
	data := [][]string{}
	for _, entry := range uploads.Entries {
		data = append(data, []string{
			entry.Id, entry.Expire, entry.Context, entry.Uploaded.Format("2006-01-02 15:04:05"),
		})
	}

	WriteTable([]string{"ID", "EXPIRE", "CONTEXT", "UPLOADED"}, data)

	return nil
}

// turn the Uploads{} struct into xtnd output and print it
func RespondExtended(resp *req.Response) error {
	uploads, err := GetUploadsFromResponse(resp)
	if err != nil {
		return err
	}

	if uploads.Message != "" {
		fmt.Println(uploads.Message)
	}

	WriteExtended(uploads)

	return nil
}