File: main.go

package info (click to toggle)
golang-github-wildducktheories-go-csv 0.0~git20170625.a843eda-3
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 188 kB
  • sloc: makefile: 5
file content (74 lines) | stat: -rw-r--r-- 2,011 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package main

import (
	"flag"
	"fmt"
	"os"

	"github.com/wildducktheories/go-csv"
	"github.com/wildducktheories/go-csv/utils"
)

func configure(args []string) (*csv.SortProcess, error) {
	flags := flag.NewFlagSet("csv-sort", flag.ExitOnError)
	var key string
	var numericKey string
	var reverseKey string

	flags.StringVar(&key, "key", "", "The columns used to sort the input stream by.")
	flags.StringVar(&numericKey, "numeric", "", "The specified columns are treated as numeric strings.")
	flags.StringVar(&reverseKey, "reverse", "", "The specified columns are sorted in reverse order.")
	if err := flags.Parse(args); err != nil {
		return nil, err
	}

	usage := func() {
		fmt.Printf("usage: csv-sort {options}\n")
		flags.PrintDefaults()
	}

	// Use  a CSV parser to extract the partial keys from the parameter
	keys, err := csv.Parse(key)
	if err != nil || len(keys) < 1 {
		usage()
		return nil, fmt.Errorf("--key must specify one or more columns.")
	}

	numeric, err := csv.Parse(numericKey)
	if err != nil && len(numericKey) > 0 {
		usage()
		return nil, fmt.Errorf("--numeric must specify the list of numeric keys.")
	}

	reversed, err := csv.Parse(reverseKey)
	if err != nil && len(reversed) > 0 {
		usage()
		return nil, fmt.Errorf("--reverse must specify the list of keys to be sorted in reverse order.")
	}

	if i, _, _ := utils.Intersect(keys, numeric); len(i) < len(numeric) {
		return nil, fmt.Errorf("--numeric must be a strict subset of --key")
	}

	if i, _, _ := utils.Intersect(keys, reversed); len(i) < len(reversed) {
		return nil, fmt.Errorf("--reverse must be a strict subset of --key")
	}

	return (&csv.SortKeys{Keys: keys, Numeric: numeric, Reversed: reversed}).AsSortProcess(), nil
}

func main() {
	var p *csv.SortProcess
	var err error
	var errCh = make(chan error, 1)

	if p, err = configure(os.Args[1:]); err == nil {
		p.Run(csv.WithIoReader(os.Stdin), csv.WithIoWriter(os.Stdout), errCh)
		err = <-errCh
	}

	if err != nil {
		fmt.Printf("fatal: %v\n", err)
		os.Exit(1)
	}
}