feat: add key/import command

Co-authored-by: Henrique Dias <[email protected]>

Author: thedmdim

key.go

@@ -1,6 +1,11 @@
 package shell
 
-import "context"
+import (
+	"context"
+	"io"
+
+	files "github.com/ipfs/go-ipfs-files"
+)
 
 type Key struct {
 	Id   string
@@ -82,3 +87,45 @@ func (s *Shell) KeyRm(ctx context.Context, name string) ([]*Key, error) {
 	}
 	return out.Keys, nil
 }
+
+type KeyImportOpt func(*RequestBuilder) error
+type keyImportOpt struct{}
+
+var KeyImportGen keyImportOpt
+
+func (keyImportOpt) IpnsBase(enc string) KeyImportOpt {
+	return func(rb *RequestBuilder) error {
+		rb.Option("ipns-base", enc)
+		return nil
+	}
+}
+
+func (keyImportOpt) Format(format string) KeyImportOpt {
+	return func(rb *RequestBuilder) error {
+		rb.Option("format", format)
+		return nil
+	}
+}
+
+func (keyImportOpt) AllowAnyKeyType(allow bool) KeyImportOpt {
+	return func(rb *RequestBuilder) error {
+		rb.Option("allow-any-key-type", allow)
+		return nil
+	}
+}
+
+// KeyImport imports key as file.
+func (s *Shell) KeyImport(ctx context.Context, name string, key io.Reader, options ...KeyImportOpt) error {
+	fr := files.NewReaderFile(key)
+	slf := files.NewSliceDirectory([]files.DirEntry{files.FileEntry("", fr)})
+	fileReader := files.NewMultiFileReader(slf, true)
+
+	rb := s.Request("key/import", name)
+	for _, opt := range options {
+		if err := opt(rb); err != nil {
+			return err
+		}
+	}
+
+	return rb.Body(fileReader).Exec(ctx, nil)
+}

key_test.go

@@ -2,6 +2,7 @@ package shell
 
 import (
 	"context"
+	"os"
 	"testing"
 
 	"github.com/cheekybits/is"
@@ -114,3 +115,19 @@ func TestKeyRm(t *testing.T) {
 	is.NotNil(err)
 	is.Equal(err.Error(), "key/rm: no key named testKey was found")
 }
+
+func TestKeyImport(t *testing.T) {
+	is := is.New(t)
+	s := NewShell(shellUrl)
+
+	// Key generated as per: https://github.com/ipfs/kubo/blob/c9d51bbe0133968858aa9991b7f69ec269126599/test/sharness/t0165-keystore-data/README.md
+	f, err := os.Open("./tests/key_test.pem")
+	is.Nil(err)
+	defer f.Close()
+
+	err = s.KeyImport(context.Background(), "testImportKey", f, KeyImportGen.Format("pem-pkcs8-cleartext"))
+	is.Nil(err)
+
+	_, err = s.KeyRm(context.Background(), "testImportKey")
+	is.Nil(err)
+}

tests/key_test.pem

@@ -0,0 +1,3 @@
+-----BEGIN PRIVATE KEY-----
+MC4CAQAwBQYDK2VwBCIEIJ2M1na2f3dRm4b1FcAQvsn7q08+XfBZcr4MgH4yiBdz
+-----END PRIVATE KEY-----