package main

import (
	"bufio"
	"flag"
	"fmt"
	"os"

	"github.com/kalloc/dkim"
)

// ./test path/to/emls/*.eml

func main() {
	var (
		filename string
		fp       *os.File
		err      error
		dk       *dkim.DKIM
	)

	flag.Parse()

	for _, filename = range flag.Args() {
		fmt.Printf("Check: %s — ", filename)
		if fp, err = os.Open(filename); err != nil {
			fmt.Printf("ERR-WRONG_FILE")
		} else if dk, _ = dkim.ParseEml(bufio.NewReader(fp)); dk == nil {
			fmt.Printf("ERR-DKIM_NOT_FOUND")
		} else if _, err = dk.GetPublicKey(); err != nil {
			fmt.Printf("ERR-DKIM_PK_NOT_FOUND")
		} else if dk.Verify() == false {
			fmt.Printf("ERR-DKIM_NOT_VERIFIED (Body is %v, Sig is %v)", dk.Status.ValidBody, dk.Status.Valid)
		} else {
			fmt.Printf("OK")
		}
		fmt.Printf("\n")
	}
}