gmantaos/Goirate

package main

import (
    "bytes"
    "encoding/json"
    "log"

    "goirate/torrents"

    "github.com/olekukonko/tablewriter"
)

// SearchCommand defines the search command and holds its options.
type SearchCommand struct {
    torrentSearchArgs
    Args positionalArgs `positional-args:"1" required:"1"`
}

// Execute is the callback of the mirrors command.
func (m *SearchCommand) Execute(args []string) error {

    torrents, err := m.GetTorrents(m.Args.Query)

    if err != nil {
        return err
    }

    torrents = m.GetFilters().FilterTorrentsCount(torrents, m.Count)

    if Options.JSON {
        torrentsJSON, err := json.MarshalIndent(torrents, "", "   ")

        if err != nil {
            return err
        }

        log.Println(string(torrentsJSON))

    } else if m.MagnetLink {

        for _, torrent := range torrents {
            log.Println(torrent.Magnet)
        }

    } else if m.TorrentURL {

        for _, torrent := range torrents {
            log.Println(torrent.FullURL())
        }

    } else {

        log.Println(getTorrentsTable(torrents))

    }

    return nil
}

func getTorrentsTable(torrents []torrents.Torrent) string {
    buf := bytes.NewBufferString("")

    table := tablewriter.NewWriter(buf)
    table.SetHeader([]string{"Title", "Size", "Seeds/Peers"})
    table.SetColumnAlignment([]int{tablewriter.ALIGN_DEFAULT, tablewriter.ALIGN_CENTER, tablewriter.ALIGN_DEFAULT})
    table.SetBorders(tablewriter.Border{Left: true, Top: false, Right: false, Bottom: false})
    table.SetCenterSeparator("")
    table.SetColumnSeparator("")
    table.SetRowLine(true)
    table.SetAutoFormatHeaders(false)
    table.SetAutoWrapText(false)

    for _, torrent := range torrents {

        table.Append([]string{torrent.Title + "\n" + torrent.FullURL(), torrent.SizeString(), torrent.PeersString()})
    }

    table.Render()

    return buf.String()
}