groups.go

package sshkeymanager

import (
	"log"

	"github.com/pkg/errors"
	"github.com/rs-pro/sshkeymanager/group"
)

func (c *Client) GetGroups() ([]group.Group, error) {
	if c == nil {
		return nil, errors.New("client not initialized")
	}
	if c.GroupsCache == nil {
		raw, se, err := c.Execute("cat /etc/group")
		if err != nil {
			return nil, errors.Wrap(err, raw+se)
		}
		groups, err := group.Parse(raw)
		if err != nil {
			return nil, err
		}
		c.GroupsCache = &groups
	}

	return *c.GroupsCache, nil
}

func (c *Client) ClearGroupCache() error {
	c.GroupsCache = nil
	return nil
}

func (c *Client) FindGroup(group group.Group) (*group.Group, error) {
	groups, err := c.GetGroups()
	if err != nil {
		log.Println("failed to get groups", err)
		return nil, nil
	}
	for _, g := range groups {
		if g.GID == group.GID || g.Name == group.Name {
			return &g, nil
		}
	}
	return nil, nil
}

func (c *Client) AddGroup(group group.Group) (*group.Group, error) {
	if group.Name == "" {
		return nil, errors.New("group name cannot be empty")
	}
	g, _ := c.FindGroup(group)
	if g != nil {
		return g, nil
	}

	so, se, err := c.Execute(group.GroupAdd())
	if err != nil {
		return nil, errors.Wrap(err, so+se)
	}

	c.ClearGroupCache()

	g, _ = c.FindGroup(group)
	if g == nil {
		return nil, errors.New("failed to add group")
	}
	return g, nil
}

func (c *Client) DeleteGroup(group group.Group) (*group.Group, error) {
	if group.Name == "" {
		return nil, errors.New("group name cannot be empty")
	}
	g, _ := c.FindGroup(group)
	if g == nil {
		return g, errors.New("group not found, so not deleted")
	}

	so, se, err := c.Execute(group.GroupDelete())
	if err != nil {
		return g, errors.Wrap(err, so+se)
	}

	c.ClearGroupCache()

	g2, _ := c.FindGroup(group)
	if g2 != nil {
		return g2, errors.New("failed to delete group")
	}
	return g, nil
}