package packages

import (
	"fmt"
	"io"
	"sort"
	"strings"

	"github.com/anchore/syft/syft/pkg"
	"github.com/olekukonko/tablewriter"
)

type TablePresenter struct {
	catalog *pkg.Catalog
}

func NewTablePresenter(catalog *pkg.Catalog) *TablePresenter {
	return &TablePresenter{
		catalog: catalog,
	}
}

func (pres *TablePresenter) Present(output io.Writer) error {
	rows := make([][]string, 0)

	columns := []string{"Name", "Version", "Type"}
	for _, p := range pres.catalog.Sorted() {
		row := []string{
			p.Name,
			p.Version,
			string(p.Type),
		}
		rows = append(rows, row)
	}

	if len(rows) == 0 {
		fmt.Fprintln(output, "No packages discovered")
		return nil
	}

	// sort by name, version, then type
	sort.SliceStable(rows, func(i, j int) bool {
		for col := 0; col < len(columns); col++ {
			if rows[i][col] != rows[j][col] {
				return rows[i][col] < rows[j][col]
			}
		}
		return false
	})
	rows = removeDuplicateRows(rows)

	table := tablewriter.NewWriter(output)

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

	table.AppendBulk(rows)
	table.Render()

	return nil
}

func removeDuplicateRows(items [][]string) [][]string {
	seen := map[string][]string{}
	var result [][]string

	for _, v := range items {
		key := strings.Join(v, "|")
		if seen[key] != nil {
			// dup!
			continue
		}

		seen[key] = v
		result = append(result, v)
	}
	return result
}