2022-03-19 02:35:04 +00:00
|
|
|
package saltyim
|
|
|
|
|
|
|
|
import (
|
2022-03-19 11:42:36 +00:00
|
|
|
"bufio"
|
2022-03-19 06:02:40 +00:00
|
|
|
"fmt"
|
2022-03-19 11:42:36 +00:00
|
|
|
"io"
|
2022-03-19 02:35:04 +00:00
|
|
|
"os"
|
2022-03-19 11:42:36 +00:00
|
|
|
"strings"
|
2022-03-19 02:35:04 +00:00
|
|
|
|
|
|
|
"github.com/keys-pub/keys"
|
|
|
|
"go.mills.io/salty"
|
|
|
|
)
|
|
|
|
|
2022-03-19 11:42:36 +00:00
|
|
|
func readUser(fd io.Reader) (Addr, error) {
|
|
|
|
scan := bufio.NewScanner(fd)
|
|
|
|
|
|
|
|
var a Addr
|
|
|
|
for scan.Scan() {
|
|
|
|
if strings.HasPrefix(scan.Text(), "# user:") {
|
|
|
|
user := strings.Split(strings.TrimSpace(strings.TrimPrefix(scan.Text(), "# user:")), "@")
|
|
|
|
if len(user) != 2 {
|
|
|
|
return Addr{}, fmt.Errorf("user not found")
|
|
|
|
}
|
|
|
|
a.User, a.Domain = user[0], user[1]
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return a, scan.Err()
|
|
|
|
}
|
|
|
|
|
2022-03-19 06:34:33 +00:00
|
|
|
// DefaultIdentity returns a default identity file (if one exists) otherwise
|
|
|
|
// returns an empty string
|
|
|
|
func DefaultIdentity() string {
|
2022-03-19 16:34:39 +00:00
|
|
|
return os.ExpandEnv("$HOME/.salty/$USER.key")
|
2022-03-19 06:34:33 +00:00
|
|
|
}
|
|
|
|
|
2022-03-19 11:42:36 +00:00
|
|
|
// CreateIdentity ...
|
|
|
|
func CreateIdentity(fn, user string) error {
|
|
|
|
f, err := os.OpenFile(fn, os.O_CREATE|os.O_EXCL|os.O_WRONLY, 0600)
|
|
|
|
if err != nil {
|
|
|
|
return fmt.Errorf("error opening identity %s for writing: %w", fn, err)
|
|
|
|
}
|
|
|
|
defer f.Close()
|
|
|
|
|
|
|
|
salty.GenerateKeys(f)
|
|
|
|
|
|
|
|
f.Write([]byte(fmt.Sprintf("# user: %s\n", user)))
|
|
|
|
|
|
|
|
if err := f.Sync(); err != nil {
|
|
|
|
return fmt.Errorf("error syncing identity %s for writing: %w", fn, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := f.Close(); err != nil {
|
|
|
|
return fmt.Errorf("error closing identity %s for writing: %w", fn, err)
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2022-03-19 02:35:04 +00:00
|
|
|
// GetIdentity ...
|
|
|
|
func GetIdentity(fn string) (*keys.EdX25519Key, Addr, error) {
|
|
|
|
id, err := os.Open(fn)
|
|
|
|
if err != nil {
|
2022-03-19 11:42:36 +00:00
|
|
|
return nil, Addr{}, fmt.Errorf("error opening identity file: %q", fn)
|
2022-03-19 02:35:04 +00:00
|
|
|
}
|
|
|
|
defer id.Close()
|
|
|
|
|
|
|
|
key, err := salty.ParseIdentity(id)
|
|
|
|
if err != nil {
|
2022-03-19 11:42:36 +00:00
|
|
|
return nil, Addr{}, fmt.Errorf("error reading private key: %q", fn)
|
2022-03-19 02:35:04 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
id.Seek(0, 0)
|
2022-03-19 11:42:36 +00:00
|
|
|
me, err := readUser(id)
|
2022-03-19 02:35:04 +00:00
|
|
|
if err != nil {
|
2022-03-19 06:02:40 +00:00
|
|
|
return key, Addr{}, fmt.Errorf("error reading user from keyfile: %q", fn)
|
2022-03-19 02:35:04 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return key, me, nil
|
|
|
|
}
|