package main

import (
	"crypto/rand"
	"crypto/rsa"
	"fmt"
	"github.com/xanzy/ssh-agent"
	"golang.org/x/crypto/ed25519"
	"golang.org/x/crypto/ssh/agent"
)

func addEd25519(a agent.Agent) error {
	_, priv, err := ed25519.GenerateKey(rand.Reader)
	if err != nil {
		return err
	}
	toadd := agent.AddedKey{
		PrivateKey: priv,
	}
	if err := a.Add(toadd); err != nil {
		return err
	}
	fmt.Println("ed25519 key added")
	return nil
}

func addRSA(a agent.Agent) error {
	priv, err := rsa.GenerateKey(rand.Reader, 2048)
	if err != nil {
		return err
	}
	toadd := agent.AddedKey{
		PrivateKey: priv,
	}
	if err := a.Add(toadd); err != nil {
		return err
	}
	fmt.Println("RSA key added")
	return nil
}

func main() {
	local, _, _ := sshagent.New()
	//soc, err := net.Dial("unix", os.Getenv("SSH_AUTH_SOCK"))
	//if err != nil {
	//	fmt.Printf("Error connecting to agent: %v\n", err)
	//	os.Exit(1)
	//}
	//local := agent.NewClient(soc)
	if err := addRSA(local); err != nil {
		fmt.Println(err)
	}
	if err := addEd25519(local); err != nil {
		fmt.Println(err)
	}
}