2 files changed, 41 insertions, 9 deletions
diff --git a/src/org/traccar/database/DataManager.java b/src/org/traccar/database/DataManager.java
index c32f0b017..2b7d15b3d 100644
--- a/src/org/traccar/database/DataManager.java
+++ b/src/org/traccar/database/DataManager.java
@@ -26,9 +26,15 @@ import java.sql.ResultSet;
 import java.sql.SQLException;
 import java.sql.Statement;
 import java.text.ParseException;
-import java.util.*;
+import java.util.Calendar;
+import java.util.Collection;
+import java.util.Date;
+import java.util.HashMap;
+import java.util.LinkedList;
+import java.util.List;
+import java.util.Map;
+import java.util.Properties;
 import javax.sql.DataSource;
-
 import org.traccar.Context;
 import org.traccar.helper.DriverDelegate;
 import org.traccar.helper.Log;
@@ -36,6 +42,7 @@ import org.traccar.http.JsonConverter;
 import org.traccar.model.Device;
 import org.traccar.model.Permission;
 import org.traccar.model.Position;
+import org.traccar.model.Server;
 import org.traccar.model.User;
 
 public class DataManager {
@@ -307,6 +314,14 @@ public class DataManager {
         admin.setPassword("admin");
         admin.setAdmin(true);
         addUser(admin);
+        
+        Server server = new Server();
+        server.setRegistration(true);
+        QueryBuilder.create(dataSource,
+                "INSERT INTO server (registration, latitude, longitude, zoom)" +
+                "VALUES (:registration, :latitude, :longitude, :zoom);")
+                .setObject(server)
+                .executeUpdate();
 
         mockData(admin.getId());
     }
@@ -348,17 +363,12 @@ public class DataManager {
     }
 
     public User login(String email, String password) throws SQLException {
-        Collection<User> result = QueryBuilder.create(dataSource,
+        return QueryBuilder.create(dataSource,
                 "SELECT * FROM user WHERE email = :email AND " +
                 "password = CAST(HASH('SHA256', STRINGTOUTF8(:password), 1000) AS VARCHAR);")
                 .setString("email", email)
                 .setString("password", password)
-                .executeQuery(new User());
-        if (!result.isEmpty()) {
-            return result.iterator().next();
-        } else {
-            return null;
-        }
+                .executeQuerySingle(new User());
     }
 
     public void addUser(User user) throws SQLException {
@@ -420,5 +430,18 @@ public class DataManager {
                 .setDate("to", to)
                 .executeQuery(new Position());
     }
+    
+    public Server getServer() throws SQLException {
+        return QueryBuilder.create(dataSource,
+                "SELECT * FROM server;")
+                .executeQuerySingle(new Server());
+    }
+    
+    public void updateServer(Server server) throws SQLException {
+        QueryBuilder.create(dataSource,
+                "UPDATE server SET registration = :registration WHERE id = :id;")
+                .setObject(server)
+                .executeUpdate();
+    }
 
 }
diff --git a/src/org/traccar/database/QueryBuilder.java b/src/org/traccar/database/QueryBuilder.java
index a35b07fee..386ed86d7 100644
--- a/src/org/traccar/database/QueryBuilder.java
+++ b/src/org/traccar/database/QueryBuilder.java
@@ -201,6 +201,15 @@ public class QueryBuilder {
         public void process(T object, ResultSet resultSet) throws SQLException;
     }
     
+    public <T extends Factory> T executeQuerySingle(T prototype) throws SQLException {
+        Collection<T> result = executeQuery(prototype);
+        if (!result.isEmpty()) {
+            return result.iterator().next();
+        } else {
+            return null;
+        }
+    }
+    
     public <T extends Factory> Collection<T> executeQuery(T prototype) throws SQLException {
         List<T> result = new LinkedList<T>();