Function EnsureProjectUser

cmd/cql-proxy/model/project_user.go:206–294

EnsureProjectUser add/update user info to project database.s

(db *gorp.DbMap, provider string,
	uid string, name string, email string, extra gin.H,
	enableSignUp bool, signUpState ProjectUserState)

// EnsureProjectUser add/update user info to project database.s
func EnsureProjectUser(db *gorp.DbMap, provider string,
	uid string, name string, email string, extra gin.H,
	enableSignUp bool, signUpState ProjectUserState) (
	u *ProjectUser, err error) {

	// find by uid
	err = db.SelectOne(&u,
		`SELECT * FROM "____user" WHERE "provider" = ? AND "provider_uid" = ? LIMIT 1`,
		provider, uid)
	exists := true
	now := time.Now().Unix()

	if err != nil {
		// find by provider and email or provider and name
		switch provider {
		case "weibo":
			err = db.SelectOne(&u,
				`SELECT * FROM "____user" WHERE "provider" = ? AND "name" = ? AND "state" = ? LIMIT 1`,
				provider, name, ProjectUserStatePreRegistered)
		default:
			// google, facebook, twitter, etc.
			err = db.SelectOne(&u,
				`SELECT * FROM "____user" WHERE "provider" = ? AND "email" = ? AND "state" = ? LIMIT 1`,
				provider, email, ProjectUserStatePreRegistered)
		}
	}

	if err != nil && !enableSignUp {
		// new user, not even pre-registered
		err = errors.New("user does not exists, sign up disabled")
		return
	}

	if err != nil {
		u = &ProjectUser{
			Name:        name,
			Email:       email,
			State:       signUpState,
			Provider:    provider,
			ProviderUID: uid,
			Extra:       extra,
		}

		exists = false
		err = nil
	} else {
		if u.State == ProjectUserStateDisabled {
			// disabled
			err = errors.New("account is disabled")
			return
		}

		u.LastLogin = now
		u.Name = name
		u.Email = email
		u.Extra = extra
	}