routes_users.go

package web

import (
	"github.com/novakit/nova"
	"github.com/novakit/router"
	"github.com/novakit/view"
	"github.com/yankeguo/bastion/types"
)

func routeListUsers(c *nova.Context) (err error) {
	us, v := userService(c), view.Extract(c)
	var res1 *types.ListUsersResponse
	if res1, err = us.ListUsers(c.Req.Context(), &types.ListUsersRequest{}); err != nil {
		return
	}
	v.Data["users"] = res1.Users
	v.DataAsJSON()
	return
}

func routeCreateUser(c *nova.Context) (err error) {
	us, v := userService(c), view.Extract(c)
	var res1 *types.CreateUserResponse
	if res1, err = us.CreateUser(c.Req.Context(), &types.CreateUserRequest{
		Account:  c.Req.FormValue("account"),
		Nickname: c.Req.FormValue("nickname"),
		Password: c.Req.FormValue("password"),
	}); err != nil {
		return
	}
	v.Data["user"] = res1.User
	v.DataAsJSON()
	return
}

func routeGetCurrentUser(c *nova.Context) (err error) {
	a, v := authResult(c), view.Extract(c)
	v.Data["user"] = a.User
	v.DataAsJSON()
	return
}

func routeGetCurrentUserGrantItems(c *nova.Context) (err error) {
	a, gs, v, opts := authResult(c), grantService(c), view.Extract(c), webOptions(c)
	var res1 *types.ListGrantItemsResponse
	if res1, err = gs.ListGrantItems(c.Req.Context(), &types.ListGrantItemsRequest{
		Account: a.User.Account,
	}); err != nil {
		return
	}
	v.Data["grant_items"] = res1.GrantItems
	v.Data["ssh_domain"] = opts.SSHDomain
	v.DataAsJSON()
	return
}

func routeUpdateUserNickname(c *nova.Context) (err error) {
	us, v := userService(c), view.Extract(c)
	var res1 *types.UpdateUserResponse
	if res1, err = us.UpdateUser(c.Req.Context(), &types.UpdateUserRequest{
		Account:        c.Req.FormValue("account"),
		UpdateNickname: true,
		Nickname:       c.Req.FormValue("nickname"),
	}); err != nil {
		return
	}
	v.Data["user"] = res1.User
	v.DataAsJSON()
	return
}

func routeUpdateCurrentUserPassword(c *nova.Context) (err error) {
	a, us, v := authResult(c), userService(c), view.Extract(c)
	if _, err = us.AuthenticateUser(c.Req.Context(), &types.AuthenticateUserRequest{
		Account:  a.User.Account,
		Password: c.Req.FormValue("oldPassword"),
	}); err != nil {
		return
	}
	var res2 *types.UpdateUserResponse
	if res2, err = us.UpdateUser(c.Req.Context(), &types.UpdateUserRequest{
		Account:        a.User.Account,
		UpdatePassword: true,
		Password:       c.Req.FormValue("newPassword"),
	}); err != nil {
		return
	}
	v.Data["user"] = res2.User
	v.DataAsJSON()
	return
}

func routeUpdateUserIsAdmin(c *nova.Context) (err error) {
	us, v := userService(c), view.Extract(c)
	var res1 *types.UpdateUserResponse
	if res1, err = us.UpdateUser(c.Req.Context(), &types.UpdateUserRequest{
		Account:       c.Req.FormValue("account"),
		UpdateIsAdmin: true,
		IsAdmin:       IsFormValueTrue(c.Req.FormValue("is_admin")),
	}); err != nil {
		return
	}
	v.Data["user"] = res1.User
	v.DataAsJSON()
	return
}

func routeUpdateUserIsBlocked(c *nova.Context) (err error) {
	us, v := userService(c), view.Extract(c)
	var res1 *types.UpdateUserResponse
	if res1, err = us.UpdateUser(c.Req.Context(), &types.UpdateUserRequest{
		Account:         c.Req.FormValue("account"),
		UpdateIsBlocked: true,
		IsBlocked:       IsFormValueTrue(c.Req.FormValue("is_blocked")),
	}); err != nil {
		return
	}
	v.Data["user"] = res1.User
	v.DataAsJSON()
	return
}

func routeGetUser(c *nova.Context) (err error) {
	rp, us, v := router.PathParams(c), userService(c), view.Extract(c)
	var res1 *types.GetUserResponse
	if res1, err = us.GetUser(c.Req.Context(), &types.GetUserRequest{
		Account: rp.Get("account"),
	}); err != nil {
		return
	}
	v.Data["user"] = res1.User
	v.DataAsJSON()
	return
}