diff --git a/bytes.go b/bytes.go new file mode 100644 index 0000000..06e815f --- /dev/null +++ b/bytes.go @@ -0,0 +1,41 @@ +package keys + +// Bytes64 converts byte slice to *[64]byte or panics. +func Bytes64(b []byte) *[64]byte { + if len(b) != 64 { + panic("not 64 bytes") + } + var b64 [64]byte + copy(b64[:], b) + return &b64 +} + +// Bytes32 converts byte slice to *[32]byte or panics. +func Bytes32(b []byte) *[32]byte { + if len(b) != 32 { + panic("not 32 bytes") + } + var b32 [32]byte + copy(b32[:], b) + return &b32 +} + +// Bytes24 converts byte slice to *[24]byte or panics. +func Bytes24(b []byte) *[24]byte { + if len(b) != 24 { + panic("not 24 bytes") + } + var b24 [24]byte + copy(b24[:], b) + return &b24 +} + +// Bytes16 converts byte slice to *[16]byte or panics. +func Bytes16(b []byte) *[16]byte { + if len(b) != 16 { + panic("not 16 bytes") + } + var b16 [16]byte + copy(b16[:], b) + return &b16 +} diff --git a/rand.go b/rand.go index b23fd93..559281b 100644 --- a/rand.go +++ b/rand.go @@ -95,42 +95,13 @@ func RandFileName() string { return encoding.MustEncode(RandBytes(16), encoding.Base32, encoding.NoPadding()) } -// Bytes64 converts byte slice to *[64]byte. -func Bytes64(b []byte) *[64]byte { - if len(b) != 64 { - panic("not 64 bytes") - } - var b64 [64]byte - copy(b64[:], b) - return &b64 -} - -// Bytes32 converts byte slice to *[32]byte. -func Bytes32(b []byte) *[32]byte { - if len(b) != 32 { - panic("not 32 bytes") - } - var b32 [32]byte - copy(b32[:], b) - return &b32 -} - -// Bytes24 converts byte slice to *[24]byte. -func Bytes24(b []byte) *[24]byte { - if len(b) != 24 { - panic("not 24 bytes") - } - var b24 [24]byte - copy(b24[:], b) - return &b24 -} - -// Bytes16 converts byte slice to *[16]byte. -func Bytes16(b []byte) *[16]byte { - if len(b) != 16 { - panic("not 16 bytes") +// RandDigits returns string of random digits of length. +// RandDigits(6) => "745566" +func RandDigits(length int) string { + charSet := numbers + b := make([]byte, 0, length) + for i := 0; i < length; i++ { + b = append(b, randomChar(charSet, b)) } - var b16 [16]byte - copy(b16[:], b) - return &b16 + return string(b) } diff --git a/rand_test.go b/rand_test.go index 1eca3f9..96b6e1b 100644 --- a/rand_test.go +++ b/rand_test.go @@ -51,3 +51,11 @@ func TestRandTempPath(t *testing.T) { p2 := keys.RandTempPath() require.NotEqual(t, p, p2) } + +func TestRandDigits(t *testing.T) { + p := keys.RandDigits(10) + require.NotEmpty(t, p) + t.Logf("%s", p) + p2 := keys.RandDigits(10) + require.NotEqual(t, p, p2) +}