package hash

import (
	"crypto/hmac"
	"crypto/md5"
	"crypto/sha1"
	"crypto/sha256"
	"crypto/sha512"
	"encoding/hex"
	"fmt"
	"golang.org/x/crypto/md4"
	"golang.org/x/crypto/ripemd160"
	"hash"
	"hash/crc32"
	"io"
	"os"

	"b612.me/starlog"
	"github.com/spf13/cobra"
)

var hmacKey string

var Cmd = &cobra.Command{
	Use:   "hash",
	Short: "多种方法哈希值校验",
	Long:  "进行多种方法哈希值校验",
	Run: func(this *cobra.Command, args []string) {
		var cumethod, method []string
		var result = make(map[string]string)
		var err error
		cumethod = []string{"md5", "crc32", "sha512", "sha384", "sha256", "sha224", "sha1", "md4", "ripemd160", "hmacmd5", "hmacmd4", "hmacsha1", "hmacsha224", "hmacsha256", "hmacsha384", "hmacsha512"}
		if ok, _ := this.Flags().GetBool("all"); ok {
			method = cumethod
		} else {
			if len(args) == 0 {
				this.Usage()
				os.Exit(1)
			}

			for _, v := range cumethod {
				if ok, _ := this.Flags().GetBool(v); ok {
					method = append(method, v)
				}
			}
			if len(method) == 0 {
				method = append(method, "md5")
			}
		}
		if ok, _ := this.Flags().GetBool("file"); ok {
			result, err = FileSumAll(args[0], hmacKey, method, func(pect float64) {
				if pect != 100.0 {
					fmt.Printf("校验已完成：%f%%\r", pect)
				} else {
					fmt.Printf("校验已完成：%f%%\n", pect)
				}
			})
		} else {
			var bres map[string][]byte
			bres, err = SumAll([]byte(args[0]), hmacKey, method)
			if err == nil {
				for k, v := range bres {
					result[k] = hex.EncodeToString(v)
				}
			}
		}
		if err != nil {
			starlog.Criticalln("错误：" + err.Error())
		}
		for _, v := range method {
			fmt.Printf("%s：%s\n", v, result[v])
		}
	},
}

func init() {
	Cmd.Flags().StringVarP(&hmacKey, "hmac-key", "k", "b612.me", "HMAC密钥")
	Cmd.Flags().BoolP("all", "a", false, "使用所有的校验方法")
	Cmd.Flags().BoolP("file", "f", false, "对指定文件进行校验")
	Cmd.Flags().BoolP("md5", "m", false, "进行MD5校验（默认）")
	Cmd.Flags().BoolP("crc32", "c", false, "进行CRC32校验")
	Cmd.Flags().BoolP("sha512", "s", false, "进行SHA512校验")
	Cmd.Flags().Bool("sha384", false, "进行SHA384校验")
	Cmd.Flags().Bool("sha256", false, "进行SHA256校验")
	Cmd.Flags().Bool("sha224", false, "进行SHA224校验")
	Cmd.Flags().Bool("sha1", false, "进行SHA1校验")
	Cmd.Flags().Bool("md4", false, "进行MD4校验")
	Cmd.Flags().Bool("ripemd160", false, "进行RIPEMD160校验")
	Cmd.Flags().Bool("hmacmd5", false, "进行HMACMD5校验")
	Cmd.Flags().Bool("hmacmd4", false, "进行HMACMD4校验")
	Cmd.Flags().Bool("hmacsha1", false, "进行HMACSHA1校验")
	Cmd.Flags().Bool("hmacsha224", false, "进行HMACSHA224校验")
	Cmd.Flags().Bool("hmacsha256", false, "进行HMACSHA256校验")
	Cmd.Flags().Bool("hmacsha384", false, "进行HMACSHA384校验")
	Cmd.Flags().Bool("hmacsha512", false, "进行HMACSHA512校验")
}

func FileSumAll(filepath string, key string, method []string, shell func(float64)) (map[string]string, error) {
	result := make(map[string]string)
	methods := make(map[string]hash.Hash)
	var iscrc bool

	if len(method) == 0 {
		method = []string{"sha512", "sha256", "sha384", "sha224", "sha1", "crc32", "md5", "md4", "ripemd160", "hmacmd5", "hmacmd4", "hmacsha1", "hmacsha224", "hmacsha256", "hmacsha384", "hmacsha512"}
	}
	fp, err := os.Open(filepath)
	defer fp.Close()
	if err != nil {
		return result, err
	}
	stat, _ := os.Stat(filepath)
	filebig := float64(stat.Size())
	sum512 := sha512.New()
	sum384 := sha512.New384()
	sum256 := sha256.New()
	sum224 := sha256.New224()
	sum1 := sha1.New()
	crcsum := crc32.NewIEEE()
	md5sum := md5.New()
	md4sum := md4.New()
	ripemd160sum := ripemd160.New()
	hmacmd5sum := hmac.New(md5.New, []byte(key))
	hmacmd4sum := hmac.New(md4.New, []byte(key))
	hmacsha1sum := hmac.New(sha1.New, []byte(key))
	hmacsha224sum := hmac.New(sha256.New224, []byte(key))
	hmacsha256sum := hmac.New(sha256.New, []byte(key))
	hmacsha384sum := hmac.New(sha512.New384, []byte(key))
	hmacsha512sum := hmac.New(sha512.New, []byte(key))
	for _, v := range method {
		switch v {
		case "md5":
			methods["md5"] = md5sum
		case "crc32":
			iscrc = true
		case "sha1":
			methods["sha1"] = sum1
		case "sha224":
			methods["sha224"] = sum224
		case "sha256":
			methods["sha256"] = sum256
		case "sha384":
			methods["sha384"] = sum384
		case "sha512":
			methods["sha512"] = sum512
		case "md4":
			methods["md4"] = md4sum
		case "ripemd160":
			methods["ripemd160"] = ripemd160sum
		case "hmacmd5":
			methods["hmacmd5"] = hmacmd5sum
		case "hmacmd4":
			methods["hmacmd4"] = hmacmd4sum
		case "hmacsha1":
			methods["hmacsha1"] = hmacsha1sum
		case "hmacsha224":
			methods["hmacsha224"] = hmacsha224sum
		case "hmacsha256":
			methods["hmacsha256"] = hmacsha256sum
		case "hmacsha384":
			methods["hmacsha384"] = hmacsha384sum
		case "hmacsha512":
			methods["hmacsha512"] = hmacsha512sum
		}
	}

	writer := 0
	for {
		buf := make([]byte, 1048574)
		n, err := fp.Read(buf)
		if err != nil {
			if err == io.EOF {
				break
			}
			return result, err
		}
		writer += n
		pect := (float64(writer) / filebig) * 100
		go shell(pect)
		for _, v := range methods {
			v.Write(buf[0:n])
		}
		if iscrc {
			crcsum.Write(buf[0:n])
		}
	}
	for k, v := range methods {
		result[k] = hex.EncodeToString(v.Sum(nil))
	}
	if iscrc {
		result["crc32"] = hex.EncodeToString(crcsum.Sum(nil))
	}
	return result, nil
}

func SumAll(data []byte, key string, method []string) (map[string][]byte, error) {
	result := make(map[string][]byte)
	methods := make(map[string]hash.Hash)
	var iscrc bool
	if len(method) == 0 {
		method = []string{"sha512", "sha256", "sha384", "sha224", "sha1", "crc32", "md5", "md4", "ripemd160", "hmacmd5", "hmacmd4", "hmacsha1", "hmacsha224", "hmacsha256", "hmacsha384", "hmacsha512"}
	}
	sum512 := sha512.New()
	sum384 := sha512.New384()
	sum256 := sha256.New()
	sum224 := sha256.New224()
	sum1 := sha1.New()
	crcsum := crc32.NewIEEE()
	md5sum := md5.New()
	md4sum := md4.New()
	ripemd160sum := ripemd160.New()
	hmacmd5sum := hmac.New(md5.New, []byte(key))
	hmacmd4sum := hmac.New(md4.New, []byte(key))
	hmacsha1sum := hmac.New(sha1.New, []byte(key))
	hmacsha224sum := hmac.New(sha256.New224, []byte(key))
	hmacsha256sum := hmac.New(sha256.New, []byte(key))
	hmacsha384sum := hmac.New(sha512.New384, []byte(key))
	hmacsha512sum := hmac.New(sha512.New, []byte(key))
	for _, v := range method {
		switch v {
		case "md5":
			methods["md5"] = md5sum
		case "crc32":
			iscrc = true
		case "sha1":
			methods["sha1"] = sum1
		case "sha224":
			methods["sha224"] = sum224
		case "sha256":
			methods["sha256"] = sum256
		case "sha384":
			methods["sha384"] = sum384
		case "sha512":
			methods["sha512"] = sum512
		case "md4":
			methods["md4"] = md4sum
		case "ripemd160":
			methods["ripemd160"] = ripemd160sum
		case "hmacmd5":
			methods["hmacmd5"] = hmacmd5sum
		case "hmacmd4":
			methods["hmacmd4"] = hmacmd4sum
		case "hmacsha1":
			methods["hmacsha1"] = hmacsha1sum
		case "hmacsha224":
			methods["hmacsha224"] = hmacsha224sum
		case "hmacsha256":
			methods["hmacsha256"] = hmacsha256sum
		case "hmacsha384":
			methods["hmacsha384"] = hmacsha384sum
		case "hmacsha512":
			methods["hmacsha512"] = hmacsha512sum
		}
	}
	for _, v := range methods {
		v.Write(data)
	}
	if iscrc {
		crcsum.Write(data)
	}

	for k, v := range methods {
		result[k] = v.Sum(nil)
	}
	if iscrc {
		result["crc32"] = crcsum.Sum(nil)
	}
	return result, nil
}