package parser

import (
	"bufio"
	"errors"
	"fmt"
    "io"
	"regexp"
	"strconv"
	"strings"
	"time"
)

type QEntry struct {
	id         string
	status     string
	date       time.Time
	size       int
	sender     string
	recipients []string
	reason     string
}

const sortableDateFormat = "2006-01-02 15:04:05"

func (m QEntry) ShortString() string {
	return fmt.Sprintf("[%s] %s <%s> -> %d recipients (%d bytes)", m.date.Format(sortableDateFormat), m.id, m.sender, len(m.recipients), m.size)
}

func (m QEntry) String() string {
    var recipientSuffix string
    if len(m.recipients) > 1 {
        recipientSuffix = ",..."
    }
	return fmt.Sprintf("[%s] %s <%s> -> {%d}<%s>%s (%s, %d bytes)", m.date.Format(sortableDateFormat), m.id, m.sender, len(m.recipients), m.recipients[0], recipientSuffix, m.status, m.size)
}

func (m QEntry) DetailedString() string {
    var reasonStr string
    if m.reason == "" {
        reasonStr = "-/-"
    }
    return fmt.Sprintf("Id: %s\nDate: %s\nStatus: %s\nReason: %s\nSize: %d\nSender: %s\nRecipients: %s", m.id, m.date.Format(sortableDateFormat), m.status, reasonStr, m.size, m.sender, strings.Join(m.recipients, ", "))
}

func ParseMailQ(dataSource io.Reader) ([]QEntry, error) {
	const dateFormat = "2006 Mon Jan _2 15:04:05"
	var messageIdStart = regexp.MustCompile("^[0-9A-Za-z]+[*!]? ")
	scanner := bufio.NewScanner(dataSource)
    var line string
    scanner.Scan()
    line = scanner.Text()
    if strings.HasPrefix(line, "Mail queue is empty") {
		// If mail queue is empty, there is nothing to do
		return []QEntry{}, nil
	} else if strings.HasPrefix(line, "-Queue ID-") == false {
		// Abort if input does not look like output from mailq(1)
		return nil, errors.New("Sorry, this does not look like output from mailq(1).")
	}
	var currentMail QEntry
	var queueEntries []QEntry

	for scanner.Scan() {
		// Read input line by line
		line := scanner.Text()
		fields := strings.Fields(line)
		if strings.HasPrefix(line, "--") {
			// Handle the summary line at the end of mailq output
			break
		} else if messageIdStart.MatchString(line) {
			// Handle line starting next mail in queue
			messageId := fields[0]
			if strings.HasSuffix(messageId, "*") {
				currentMail.status = "active"
			} else if strings.HasSuffix(messageId, "!") {
				currentMail.status = "hold"
			} else {
				currentMail.status = "deferred"
			}
			dateString := strconv.Itoa(time.Now().Year()) + " " + strings.Join(fields[2:6], " ")
			mailDate, _ := time.Parse(dateFormat, dateString)
			currentMail.id = strings.TrimRight(messageId, "*!")
			currentMail.size, _ = strconv.Atoi(fields[1])
			currentMail.date = mailDate
			currentMail.sender = fields[6]
			continue
		} else if strings.HasPrefix(line, "(") && strings.HasSuffix(line, ")") {
			// Handle reason for deferred status (if deferred at all, may be missing)
			currentMail.reason = strings.Trim(strings.TrimSpace(line), "()")
			continue
		} else if len(fields) == 1 {
			// Handle line with one of the mail recipients
			currentMail.recipients = append(currentMail.recipients, fields[0])
			continue
		} else if len(fields) == 0 {
			// If the next line is empty, make sure to push current mail to list
			// and create a new struct for the next mail to process
			queueEntries = append(queueEntries, currentMail)
			currentMail = QEntry{}
		}
	}
	if scanner.Err() != nil {
		// If the scanner failed, let our caller know.
		return nil, errors.New("Something went wrong with reading from stdin. Sorry!")
	}
	return queueEntries, nil
}