lndhub.go/pkg/controllers/create.ctrl.go

package controllers

import (
	"gorm.io/gorm"
	"math/rand"
	"net/http"

	"github.com/labstack/echo/v4"
	"github.com/labstack/gommon/random"

	"github.com/bumi/lndhub.go/pkg/database/models"
)

const alphaNumBytes = random.Alphanumeric

// CreateUserController : Create user controller struct
type CreateUserController struct{}

// CreateUser : Create user Controller
func (CreateUserController) CreateUser(c echo.Context) error {
	type RequestBody struct {
		PartnerID   string `json:"partnerid"`
		AccountType string `json:"accounttype"`
	}
	var body RequestBody

	if err := c.Bind(&body); err != nil {
		return err
	}

	db, _ := c.Get("db").(*gorm.DB)

	user := &models.User{}

	user.Login = randStringBytes(8)
	user.Password = randStringBytes(15)

	if err := db.Create(&user).Error; err != nil {
		return err
	}

	var ResponseBody struct {
		Login    string `json:"login"`
		Password string `json:"password"`
	}
	ResponseBody.Login = user.Login
	ResponseBody.Password = user.Password

	return c.JSON(http.StatusOK, &ResponseBody)
}

func randStringBytes(n int) string {
	b := make([]byte, n)
	for i := range b {
		b[i] = alphaNumBytes[rand.Intn(len(alphaNumBytes))]
	}
	return string(b)
}