package keys

import (

"strings"

"unicode/utf8"

)

// DataType is the type of key data.

type DataType = string

const (

// UnknownType is unknown

UnknownType DataType = ""

// IDType is string identifier (keys.ID)

IDType DataType = "id"

// SaltpackArmoredType is armored saltpack encoding.

SaltpackArmoredType DataType = "saltpack-armored"

// SaltpackType is binary saltpack encoding.

// SaltpackType DataType = "saltpack"

// SSHPublicType is ssh public key "ssh-ed25519 AAAAC3Nz..."

SSHPublicType DataType = "ssh-public"

// SSHType is ssh private key "-----BEGIN OPENSSH PRIVATE..."

SSHType DataType = "ssh"

)

// DetectDataType tries to find out what data type the bytes are.

func DetectDataType(b []byte) DataType {

s := ""

if utf8.Valid(b) {

s = strings.TrimSpace(string(b))

}

if s != "" {

if _, err := ParseID(s); err == nil {

return IDType

} else if len(s) < 100 && (strings.HasPrefix(s, "kex1") || strings.HasPrefix(s, "kbx1")) {

return IDType

}

if strings.Contains(s, "BEGIN ") && strings.Contains(s, " MESSAGE") {

return SaltpackArmoredType

}

if strings.HasPrefix(s, "-----BEGIN ") {

return SSHType

}

if strings.HasPrefix(s, "ssh-") {

return SSHPublicType

}

}

// TODO: SaltpackType

return UnknownType

}