package utils

import (
	"bufio"
	"bytes"
	"gitee.com/anesec/chameleon/pkg/types"
	"path/filepath"
	"strings"
)

func LoadUsers(data []byte) (users types.Users, err error) {
	users = make(types.Users)
	scanner := bufio.NewScanner(bytes.NewReader(data))
	for scanner.Scan() {
		line := scanner.Text()
		if len(line) == 0 {
			continue
		}

		fields := strings.Split(line, ":")
		if len(fields) != 7 {
			continue
		}

		users.Add(&types.User{
			Name:  fields[0],
			Uid:   fields[2],
			Gid:   fields[3],
			Home:  fields[5],
			Shell: filepath.Base(fields[6]),
		})
	}

	return users, scanner.Err()
}

func LoadGroups(data []byte) (groups types.Groups, err error) {
	groups = make(types.Groups)
	scanner := bufio.NewScanner(bytes.NewReader(data))
	for scanner.Scan() {
		line := scanner.Text()
		if len(line) == 0 {
			continue
		}

		fields := strings.Split(line, ":")
		if len(fields) != 4 {
			continue
		}

		groups.Add(&types.Group{
			Name:  fields[0],
			Gid:   fields[2],
			Users: strings.Split(fields[3], ","),
		})
	}

	return groups, scanner.Err()
}