main.go

package main

import (
	"fmt"
	"os"
	"time"

	"github.com/kelseykm/kelcryptor/colour"
	"github.com/kelseykm/kelcryptor/cryptography"
	"github.com/kelseykm/kelcryptor/errors"
)

const version = "2.0.4"

func main() {
	var retVal int
	defer func() {
		os.Exit(retVal)
	}()

	printBanner()

	toIgnore, toRecordTime, toEncrypt, toDecrypt, files := parseFlags()

	if cleanFiles, err := verifyFiles(files); err != nil {
		fmt.Println(err.Error())
		retVal = 2
		if !toIgnore {
			return
		}
		files = cleanFiles
	}

	password, err := func() ([]byte, error) {
		var acceptableError errors.MismatchedPasswordError
		for {
			password, err := scanPassword()
			if err == nil {
				return password, nil
			} else if err != acceptableError {
				return nil, err
			}
			fmt.Println(err.Error())
		}
	}()
	if err != nil {
		fmt.Println(err.Error())
		retVal = 2
		return
	}

	switch {
	case toEncrypt:
		for _, file := range files {
			if !toRecordTime {
				if err := cryptography.EncryptFile(password, file); err != nil {
					fmt.Println(err.Error())
					retVal = 2
					if toIgnore {
						continue
					}
					return
				}
			} else {
				start := time.Now()

				if err := cryptography.EncryptFile(password, file); err != nil {
					fmt.Println(err.Error())
					retVal = 2
					if toIgnore {
						continue
					}
					return
				}

				timeTaken := time.Since(start).Seconds()

				fmt.Printf("%s %s %s %s %s\n",
					colour.Info,
					colour.Message("Done in"),
					colour.Time(timeTaken),
					colour.Message("seconds:"),
					colour.FileName(file),
				)
			}
		}
	case toDecrypt:
		for _, file := range files {
			if !toRecordTime {
				if err := cryptography.DecryptFile(password, file); err != nil {
					fmt.Println(err.Error())
					retVal = 2
					if toIgnore {
						continue
					}
					return
				}
			} else {
				start := time.Now()

				if err := cryptography.DecryptFile(password, file); err != nil {
					fmt.Println(err.Error())
					retVal = 2
					if toIgnore {
						continue
					}
					return
				}

				timeTaken := time.Since(start).Seconds()

				fmt.Printf("%s %s %s %s %s\n",
					colour.Info,
					colour.Message("Done in"),
					colour.Time(timeTaken),
					colour.Message("seconds:"),
					colour.FileName(file),
				)
			}
		}
	}
}