fix

apps/api/internal/auth/owned_company.go

@@ -0,0 +1,83 @@
+package auth
+
+import (
+	"context"
+	"errors"
+	"strings"
+
+	"github.com/google/uuid"
+	"github.com/jackc/pgx/v5"
+)
+
+// defaultOwnedCompanyName builds a workspace name when invite accept creates a personal company.
+func defaultOwnedCompanyName(email, name string) string {
+	name = strings.TrimSpace(name)
+	if name != "" {
+		return name + "'s workspace"
+	}
+	email = strings.ToLower(strings.TrimSpace(email))
+	local := email
+	if at := strings.IndexByte(email, '@'); at > 0 {
+		local = email[:at]
+	}
+	local = strings.TrimSpace(local)
+	if local == "" {
+		return "My workspace"
+	}
+	return local + "'s workspace"
+}
+
+// ensureOwnedCompanyTx returns the user's owned company, creating one (admin membership +
+// settings + credit wallet) when they do not already own a tenant.
+func ensureOwnedCompanyTx(ctx context.Context, tx pgx.Tx, userID uuid.UUID, companyName string) (uuid.UUID, bool, error) {
+	var existing uuid.UUID
+	err := tx.QueryRow(ctx, `
+		SELECT id FROM companies WHERE owner_user_id = $1
+		ORDER BY created_at ASC
+		LIMIT 1`, userID).Scan(&existing)
+	if err == nil {
+		_, merr := tx.Exec(ctx, `
+			INSERT INTO memberships (company_id, user_id, role, status)
+			VALUES ($1, $2, 'admin', 'active')
+			ON CONFLICT (company_id, user_id) DO UPDATE
+			SET status = 'active',
+				role = 'admin',
+				updated_at = now()`, existing, userID)
+		return existing, false, merr
+	}
+	if !errors.Is(err, pgx.ErrNoRows) {
+		return uuid.Nil, false, err
+	}
+
+	companyName = strings.TrimSpace(companyName)
+	if companyName == "" {
+		companyName = "My workspace"
+	}
+
+	var companyID uuid.UUID
+	err = tx.QueryRow(ctx, `
+		INSERT INTO companies (name, owner_user_id) VALUES ($1, $2) RETURNING id`,
+		companyName, userID).Scan(&companyID)
+	if err != nil {
+		return uuid.Nil, false, err
+	}
+	_, err = tx.Exec(ctx, `
+		INSERT INTO memberships (company_id, user_id, role, status)
+		VALUES ($1, $2, 'admin', 'active')`, companyID, userID)
+	if err != nil {
+		return uuid.Nil, false, err
+	}
+	_, err = tx.Exec(ctx, `
+		INSERT INTO company_settings (company_id) VALUES ($1)
+		ON CONFLICT DO NOTHING`, companyID)
+	if err != nil {
+		return uuid.Nil, false, err
+	}
+	_, err = tx.Exec(ctx, `
+		INSERT INTO credit_balances (company_id) VALUES ($1)
+		ON CONFLICT DO NOTHING`, companyID)
+	if err != nil {
+		return uuid.Nil, false, err
+	}
+	return companyID, true, nil
+}