// Copyright 2013-2017 the u-root Authors. All rights reserved
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

// comm compares two files.
//
// Synopsis:
//
//	comm [-123h] FILE1 FILE2
//
// Descrption:
//
//	Comm reads file1 and file2, which are in lexicographical order, and
//	produces a three column output: lines only in file1; lines only in
//	file2; and lines in both files. The file name – means the standard
//	input.
//
// Options:
//
//	-1: suppress printing of column 1
//	-2: suppress printing of column 2
//	-3: suppress printing of column 3
//	-h: print this help message and exit
package main

import (
	"bufio"
	"errors"
	"flag"
	"fmt"
	"io"
	"log"
	"os"
	"strings"
)

var (
	s1   = flag.Bool("1", false, "suppress printing of column 1")
	s2   = flag.Bool("2", false, "suppress printing of column 2")
	s3   = flag.Bool("3", false, "suppress printing of column 3")
	help = flag.Bool("h", false, "print this help message and exit")

	// ErrUsage is the error for incorrect usage.
	ErrUsage = errors.New("comm: comm [-123h] file1 file2")
)

func reader(r io.Reader, c chan string) {
	b := bufio.NewReader(r)
	for {
		s, err := b.ReadString('\n')
		c <- strings.TrimRight(s, "\r\n")
		if err != nil {
			break
		}
	}
	close(c)
}

type out struct {
	s1, s2, s3 string
}

func outer(c1, c2 chan string, c chan out) {
	s1, ok1 := <-c1
	s2, ok2 := <-c2
	for {
		if ok1 && ok2 {
			switch {
			case s1 < s2:
				c <- out{s1, "", ""}
				s1, ok1 = <-c1
			case s1 > s2:
				c <- out{"", s2, ""}
				s2, ok2 = <-c2
			default:
				c <- out{"", "", s2}
				s1, ok1 = <-c1
				s2, ok2 = <-c2
			}
		} else if ok1 {
			c <- out{s1, "", ""}
			s1, ok1 = <-c1
		} else if ok2 {
			c <- out{"", s2, ""}
			s2, ok2 = <-c2
		} else {
			break
		}
	}
	close(c)
}

func comm(w io.Writer, args ...string) error {
	if len(args) != 2 || *help {
		return ErrUsage
	}

	c1 := make(chan string, 100)
	c2 := make(chan string, 100)
	c := make(chan out, 100)

	f1, err := os.Open(args[0])
	if err != nil {
		return fmt.Errorf("can't open %s: %w", args[0], err)
	}

	f2, err := os.Open(args[1])
	if err != nil {
		return fmt.Errorf("can't open %s: %w", args[1], err)
	}
	go reader(f1, c1)
	go reader(f2, c2)
	go outer(c1, c2, c)

	for {
		out, ok := <-c
		if !ok {
			break
		}

		line := ""
		if !*s1 {
			line += out.s1
		}
		line += "\t"
		if !*s2 {
			line += out.s2
		}
		line += "\t"
		if !*s3 {
			line += out.s3
		}
		if line != "\t\t" {
			fmt.Fprintln(w, strings.TrimRight(line, "\t")) // the unix comm utility does this
		}
	}
	return nil
}

func main() {
	flag.Parse()
	if err := comm(os.Stdout, flag.Args()...); err != nil {
		if err == ErrUsage {
			log.Println(err.Error())
			flag.Usage()
		}
		log.Fatal(err)
	}
}
