puerta/cmd/admin/user.go

146 lines
3.2 KiB
Go
Raw Normal View History

2022-12-30 06:51:51 +00:00
// SPDX-License-Identifier: Apache-2.0
// Copyright © 2022 Roberto Hidalgo <nidito@un.rob.mx>
package admin
import (
"fmt"
"os"
"time"
"git.rob.mx/nidito/chinampa"
"git.rob.mx/nidito/chinampa/pkg/command"
"git.rob.mx/nidito/puerta/internal/auth"
"git.rob.mx/nidito/puerta/internal/server"
"github.com/sirupsen/logrus"
"github.com/upper/db/v4/adapter/sqlite"
"golang.org/x/crypto/bcrypt"
"gopkg.in/yaml.v3"
)
func init() {
chinampa.Register(userAddCommand)
}
var userAddCommand = &command.Command{
Path: []string{"admin", "user", "create"},
Summary: "Create the initial user",
Description: "",
Arguments: command.Arguments{
{
Name: "handle",
Description: "the username to add",
Required: true,
},
{
Name: "name",
Description: "the user's name",
Required: true,
},
{
Name: "password",
Description: "the password to set for this user",
Required: true,
},
},
Options: command.Options{
"config": {
Type: "string",
Default: "./config.joao.yaml",
},
"db": {
Type: "string",
Default: "./puerta.db",
},
"ttl": {
Type: "string",
Description: "the ttl to set for the user",
Default: "30d",
},
"expires": {
Type: "string",
Description: "the max cookie lifetime",
Default: "",
},
"schedule": {
Type: "string",
Description: "the schedule to set for the user",
Default: "",
},
"greeting": {
Type: "string",
Description: "a custom greeting for the user",
Default: "",
},
},
Action: func(cmd *command.Command) error {
config := cmd.Options["config"].ToValue().(string)
db := cmd.Options["db"].ToValue().(string)
expires := cmd.Options["expires"].ToString()
schedule := cmd.Options["schedule"].ToString()
ttl := cmd.Options["ttl"].ToString()
greeting := cmd.Options["greeting"].ToString()
data, err := os.ReadFile(config)
if err != nil {
return fmt.Errorf("could not read config file: %w", err)
}
cfg := server.ConfigDefaults(db)
if err := yaml.Unmarshal(data, &cfg); err != nil {
return fmt.Errorf("could not unserialize yaml at %s: %w", config, err)
}
sess, err := sqlite.Open(sqlite.ConnectionURL{
Database: cfg.DB,
// Options: {},
})
if err != nil {
return err
}
password, err := bcrypt.GenerateFromPassword([]byte(cmd.Arguments[2].ToString()), bcrypt.DefaultCost)
if err != nil {
return err
}
var ttlDuration auth.Duration
if err := ttlDuration.UnmarshalDB(ttl); err != nil {
return err
}
user := &auth.User{
Name: cmd.Arguments[0].ToString(),
Password: string(password),
Handle: cmd.Arguments[1].ToString(),
Greeting: greeting,
TTL: ttlDuration,
}
if schedule != "" {
user.Schedule = &auth.UserSchedule{}
if err := user.Schedule.UnmarshalDB([]byte(schedule)); err != nil {
return err
}
}
if expires != "" {
t, err := time.Parse(time.RFC3339, expires)
if err != nil {
return err
}
user.Expires = &t
}
res, err := sess.Collection("user").Insert(user)
if err != nil {
return err
}
logrus.Infof("Created user %s with ID: %d", user.Name, res.ID())
return nil
},
}