package main import ( "encoding/json" "fmt" "os" "github.com/filecoin-project/lotus/chain/types" "github.com/filecoin-project/lotus/chain/wallet" _ "github.com/filecoin-project/lotus/lib/sigs/bls" _ "github.com/filecoin-project/lotus/lib/sigs/secp" "github.com/urfave/cli/v2" ) func main() { app := cli.NewApp() app.Flags = []cli.Flag{ &cli.StringFlag{ Name: "type", Aliases: []string{"t"}, Value: "bls", Usage: "specify key type to generate (bls or secp256k1)", }, &cli.StringFlag{ Name: "out", Aliases: []string{"o"}, Usage: "specify key file name to generate", }, } app.Action = func(cctx *cli.Context) error { memks := wallet.NewMemKeyStore() w, err := wallet.NewWallet(memks) if err != nil { return err } var kt types.KeyType switch cctx.String("type") { case "bls": kt = types.KTBLS case "secp256k1": kt = types.KTSecp256k1 default: return fmt.Errorf("unrecognized key type: %q", cctx.String("type")) } kaddr, err := w.WalletNew(cctx.Context, kt) if err != nil { return err } ki, err := w.WalletExport(cctx.Context, kaddr) if err != nil { return err } outFile := fmt.Sprintf("%s.key", kaddr) if cctx.IsSet("out") { outFile = fmt.Sprintf("%s.key", cctx.String("out")) } fi, err := os.Create(outFile) if err != nil { return err } defer func() { err2 := fi.Close() if err == nil { err = err2 } }() b, err := json.Marshal(ki) if err != nil { return err } if _, err := fi.Write(b); err != nil { return fmt.Errorf("failed to write key info to file: %w", err) } fmt.Println("Generated new key: ", kaddr) return nil } if err := app.Run(os.Args); err != nil { fmt.Println(err) os.Exit(1) } }