package main

import (
	"bufio"
	"os"
	"os/user"
	"syscall"

	"pfish.zone/go/multipass/file"
)

const (
	InternalError = 111
	Failed        = 1
)

func main() {
	infile := os.NewFile(3, "")
	reader := bufio.NewReader(os.Stdin)
	username, err := reader.ReadString(0)
	if err != nil {
		os.Exit(InternalError)
	}
	username = username[:len(username)-1]
	pass, err := reader.ReadString(0)
	if err != nil {
		os.Exit(InternalError)
	}
	pass = pass[:len(pass)-1]
	infile.Close()
	passfile, err := file.ForUser(username)
	if err != nil {
		os.Exit(Failed)
	}
	success, _ := passfile.Authenticate(pass)
	if !success {
		os.Exit(Failed)
	}
	user, err := user.Lookup(username)
	if err != nil {
		os.Exit(Failed)
	}
	os.Setenv("userdb_uid", user.Uid)
	os.Setenv("userdb_gid", user.Gid)
	os.Setenv("EXTRA", "userdb_uid userdb_gid")
	environ := []string{
		"USER=" + user.Username,
		"userdb_uid=" + user.Uid,
		"userdb_gid=" + user.Gid,
		"HOME=" + user.HomeDir,
		"EXTRA=userdb_uid userdb_gid",
	}
	syscall.Exec(os.Args[1], os.Args[1:], environ)
}