From def59347022475ca608f2bb6f116da6ea76cb452 Mon Sep 17 00:00:00 2001
From: Dipak Acharya <dpakach@gmail.com>
Date: Mon, 10 Aug 2020 14:03:08 +0545
Subject: [PATCH] Add UidNumber and GidNumber when creating new user

---
 pkg/service/v0/data/user.go |  2 ++
 pkg/service/v0/users.go     | 20 ++++++++++++++++++++
 2 files changed, 22 insertions(+)

diff --git a/pkg/service/v0/data/user.go b/pkg/service/v0/data/user.go
index c43cc9c..4b3c0c5 100644
--- a/pkg/service/v0/data/user.go
+++ b/pkg/service/v0/data/user.go
@@ -14,6 +14,8 @@ type User struct {
 	DisplayName string `json:"displayname" xml:"displayname"`
 	Email       string `json:"email" xml:"email"`
 	Quota       *Quota `json:"quota" xml:"quota"`
+	UIDNumber   int64  `json:"uidNumber" xml:"uidNumber"`
+	GIDNumber   int64  `json:"gidNumber" xml:"gidNumber"`
 }
 
 // Quota holds quota information
diff --git a/pkg/service/v0/users.go b/pkg/service/v0/users.go
index 42b87f3..41ed11a 100644
--- a/pkg/service/v0/users.go
+++ b/pkg/service/v0/users.go
@@ -5,6 +5,7 @@ import (
 	"encoding/hex"
 	"fmt"
 	"net/http"
+	"strconv"
 	"strings"
 
 	"github.com/cs3org/reva/pkg/user"
@@ -60,6 +61,8 @@ func (o Ocs) GetUser(w http.ResponseWriter, r *http.Request) {
 		Username:    account.PreferredName,
 		DisplayName: account.DisplayName,
 		Email:       account.Mail,
+		UIDNumber:   account.UidNumber,
+		GIDNumber:   account.GidNumber,
 		Enabled:     account.AccountEnabled,
 		// FIXME only return quota for users/{userid} endpoint (not /user)
 		// TODO query storage registry for free space? of home storage, maybe...
@@ -81,6 +84,21 @@ func (o Ocs) AddUser(w http.ResponseWriter, r *http.Request) {
 	username := r.PostFormValue("username")
 	displayname := r.PostFormValue("displayname")
 	email := r.PostFormValue("email")
+	uid := r.PostFormValue("uidNumber")
+	gid := r.PostFormValue("gidNumber")
+
+	uidNumber, err := strconv.ParseInt(uid, 10, 64)
+	if err != nil {
+		render.Render(w, r, response.ErrRender(data.MetaBadRequest.StatusCode, "Cannot use the uidNumber provided"))
+		o.logger.Error().Err(err).Str("userid", userid).Msg("Cannot use the uidNumber provided")
+		return
+	}
+	gidNumber, err := strconv.ParseInt(gid, 10, 64)
+	if err != nil {
+		render.Render(w, r, response.ErrRender(data.MetaBadRequest.StatusCode, "Cannot use the gidNumber provided"))
+		o.logger.Error().Err(err).Str("userid", userid).Msg("Cannot use the gidNumber provided")
+		return
+	}
 
 	// fallbacks
 	/* TODO decide if we want to make these fallbacks. Keep in mind:
@@ -99,6 +117,8 @@ func (o Ocs) AddUser(w http.ResponseWriter, r *http.Request) {
 			DisplayName:              displayname,
 			PreferredName:            username,
 			OnPremisesSamAccountName: username,
+			UidNumber:                uidNumber,
+			GidNumber:                gidNumber,
 			PasswordProfile: &accounts.PasswordProfile{
 				Password: password,
 			},