snapshot of https://github.com/traggo/server.git (master, 2026-09-16; blobs > 131072B omitted)

device/create.go

@@ -0,0 +1,83 @@
+package device
+
+import (
+	"context"
+	"errors"
+	"time"
+
+	"github.com/jinzhu/copier"
+	"github.com/traggo/server/auth"
+	"github.com/traggo/server/auth/rand"
+	"github.com/traggo/server/generated/gqlmodel"
+	"github.com/traggo/server/model"
+	"github.com/traggo/server/user/password"
+)
+
+var (
+	timeNow          = time.Now
+	randToken        = rand.Token
+	comparePassword  = password.ComparePassword
+	errUserPassWrong = errors.New("username/password combination does not exist")
+)
+
+// Login creates a device.
+func (r *ResolverForDevice) Login(ctx context.Context, username string, pass string, deviceName string, deviceType gqlmodel.DeviceType, cookie bool) (*gqlmodel.Login, error) {
+
+	user := new(model.User)
+	find := r.DB.Where("name = ?", username).Find(user)
+
+	if find.RecordNotFound() {
+		return nil, errUserPassWrong
+	}
+
+	if !comparePassword(user.Pass, []byte(pass)) {
+		return nil, errUserPassWrong
+	}
+
+	return r.createDeviceInternal(ctx, user, deviceName, deviceType, cookie)
+}
+
+// CreateDevice creates a device.
+func (r *ResolverForDevice) CreateDevice(ctx context.Context, deviceName string, deviceType gqlmodel.DeviceType) (*gqlmodel.Login, error) {
+
+	user := auth.GetUser(ctx)
+
+	return r.createDeviceInternal(ctx, user, deviceName, deviceType, false)
+}
+
+func (r *ResolverForDevice) createDeviceInternal(ctx context.Context, user *model.User, deviceName string, deviceType gqlmodel.DeviceType, cookie bool) (*gqlmodel.Login, error) {
+
+	token := randToken(20)
+	for !r.DB.Where("token = ?", token).Find(new(model.Device)).RecordNotFound() {
+		token = randToken(20)
+	}
+
+	now := timeNow()
+	device := &model.Device{
+		Token:     token,
+		UserID:    user.ID,
+		Name:      deviceName,
+		Type:      model.DeviceType(deviceType),
+		CreatedAt: now.UTC(),
+		ActiveAt:  now.UTC(),
+	}
+
+	if err := device.Type.Valid(); err != nil {
+		return nil, err
+	}
+
+	if cookie {
+		auth.GetCreateSession(ctx)(token, device.Type.Seconds())
+	}
+
+	create := r.DB.Create(device)
+
+	gqlUser := &gqlmodel.User{}
+	copier.Copy(gqlUser, user)
+	gqlDevice := &gqlmodel.Device{}
+	copier.Copy(gqlDevice, device)
+	return &gqlmodel.Login{
+		User:   gqlUser,
+		Device: gqlDevice,
+		Token:  token}, create.Error
+}