Added user management

1 files changed, 59 insertions, 0 deletions

diff --git a/src/org/traccar/web/server/model/DataServiceImpl.java b/src/org/traccar/web/server/model/DataServiceImpl.java
index c2d9f72e..663f0e4b 100644
--- a/src/org/traccar/web/server/model/DataServiceImpl.java
+++ b/src/org/traccar/web/server/model/DataServiceImpl.java
@@ -147,6 +147,41 @@ public class DataServiceImpl extends RemoteServiceServlet implements DataService
     }
 
     @Override
+    public List<User> getUsers() {
+        List<User> users = new LinkedList<User>();
+        EntityManager entityManager = entityManagerFactory.createEntityManager();
+        try {
+            users.addAll(entityManager.createQuery("SELECT x FROM User x", User.class).getResultList());
+            return users;
+        } finally {
+            entityManager.close();
+        }
+    }
+
+    @Override
+    public User addUser(User user) {
+        User currentUser = getUser();
+        if (currentUser.getAdmin()) {
+            EntityManager entityManager = entityManagerFactory.createEntityManager();
+            try {
+                entityManager.getTransaction().begin();
+                try {
+                    entityManager.persist(user);
+                    entityManager.getTransaction().commit();
+                    return user;
+                } catch (RuntimeException e) {
+                    entityManager.getTransaction().rollback();
+                    throw e;
+                }
+            } finally {
+                entityManager.close();
+            }
+        } else {
+            throw new SecurityException();
+        }
+    }
+
+    @Override
     public User updateUser(User user) {
         User currentUser = getUser();
         if (currentUser.getAdmin() || (currentUser.getId() == user.getId() && !user.getAdmin())) {
@@ -181,6 +216,30 @@ public class DataServiceImpl extends RemoteServiceServlet implements DataService
         }
     }
 
+    @Override
+    public User removeUser(User user) {
+        User currentUser = getUser();
+        if (currentUser.getAdmin()) {
+            EntityManager entityManager = entityManagerFactory.createEntityManager();
+            try {
+                entityManager.getTransaction().begin();
+                try {
+                    user = entityManager.merge(user);
+                    entityManager.remove(user);
+                    entityManager.getTransaction().commit();
+                    return user;
+                } catch (RuntimeException e) {
+                    entityManager.getTransaction().rollback();
+                    throw e;
+                }
+            } finally {
+                entityManager.close();
+            }
+        } else {
+            throw new SecurityException();
+        }
+    }
+
     private void createUser(User user) {
         EntityManager entityManager = entityManagerFactory.createEntityManager();
         try {