Repository binding to users

Author: TobyNartowski

src/main/java/pl/simplemethod/codebin/githubOauth/PostLogin.java

@@ -45,9 +45,8 @@ ResponseEntity postlogin(HttpServletResponse response, @RequestParam("code") Str
             githubId = githubClient.getUserInfo(token).get("id").toString();
             Users users = usersRepository.getFirstByGithub(Integer.valueOf(githubId));
             if (users == null) {
-                usersList.add(
-                        new Users(token, Integer.valueOf(githubId), "user", null, null)
-                );
+                users = new Users(token, Integer.valueOf(githubId), "user", null, null);
+                usersList.add(users);
                 usersList.forEach(usersRepository::save);
             }
             else
@@ -57,13 +56,16 @@ ResponseEntity postlogin(HttpServletResponse response, @RequestParam("code") Str
             }
             Cookie cookie = new Cookie("token", token);
             cookie.setMaxAge(999999999);
+
+            Cookie idCookie = new Cookie("id", Integer.toString(users.getId()));
+            idCookie.setMaxAge(999999999);
+
+            response.addCookie(idCookie);
             response.addCookie(cookie);
         } catch (org.json.JSONException e) {
             return new ResponseEntity<>(e, headers, HttpStatus.valueOf(404));
         }
 
         return new ResponseEntity<>("<script type=\"text/javascript\">window.location.href = \"/\"</script>", headers, HttpStatus.valueOf(200));
     }
-
-
 }

src/main/java/pl/simplemethod/codebin/model/Users.java

@@ -11,9 +11,6 @@
 @Table(name = "users")
 public class Users implements Serializable {
 
-
-    @OneToMany()
-    List<Containers> containers = new ArrayList<>();
     @Id
     @GeneratedValue(strategy = GenerationType.IDENTITY)
     @Column(name = "id")
@@ -28,6 +25,9 @@ public class Users implements Serializable {
     @Column(name = "role")
     private String role;
 
+    @OneToMany
+    private List<Containers> containers = new ArrayList<>();
+
     public Users() {
     }
 

src/main/java/pl/simplemethod/codebin/srv/SrvRestController.java

@@ -7,12 +7,17 @@
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.*;
+import org.springframework.web.util.WebUtils;
 import pl.simplemethod.codebin.linkDeploy.LinkClient;
 import pl.simplemethod.codebin.model.Containers;
 import pl.simplemethod.codebin.model.Images;
+import pl.simplemethod.codebin.model.Users;
 import pl.simplemethod.codebin.repository.ContainersRepository;
 import pl.simplemethod.codebin.repository.ImagesRepository;
+import pl.simplemethod.codebin.repository.UsersRepository;
 
+import javax.servlet.http.Cookie;
+import javax.servlet.http.HttpServletResponse;
 import java.time.Instant;
 import java.util.ArrayList;
 import java.util.List;
@@ -33,6 +38,9 @@ public class SrvRestController {
     @Autowired
     private LinkClient linkClient;
 
+    @Autowired
+    private UsersRepository usersRepository;
+
 /*
     @GetMapping("/createtest")
     public @ResponseBody
@@ -79,6 +87,35 @@ ResponseEntity createContainer(@RequestParam("dockerimage") String dockerImage,
         return new ResponseEntity<>(response.toString(), headers, HttpStatus.valueOf(status));
     }
 
+    @GetMapping("container/new")
+    public @ResponseBody
+    ResponseEntity createContainerForUser(@RequestParam("dockerimage") String dockerImage, @RequestParam("exposedports") Integer exposedPorts, @RequestParam("hostports") Integer hostPorts, @RequestParam("name") String name, @RequestParam("rammemory") Long ramMemory, @RequestParam("diskquota") Long diskQuota, @CookieValue("id") String id) {
+        HttpHeaders headers = new HttpHeaders();
+        headers.setContentType(MediaType.APPLICATION_JSON);
+        org.json.JSONObject response = srvClient.createAndRunContainer(srvClient.generateCreateConfig(dockerImage, exposedPorts, hostPorts, ramMemory, diskQuota), name);
+        int status;
+        if (response.get("status").toString().equals("204")) {
+            status = 200;
+            List<Containers> containers = new ArrayList<>();
+            Images images = imagesRepository.getFirstByName(dockerImage);
+            Containers newContainer = new Containers(name, response.get("id").toString(), images, exposedPorts, hostPorts, ramMemory, diskQuota, "XD", 1, Instant.now().getEpochSecond());
+            containers.add(newContainer);
+            containers.forEach(containersRepository::save);
+
+            try {
+                // TODO: 01/06/2019 oczekuje integera, a dajemy mu longa, pojebane
+                Users users = usersRepository.getOne((long) Integer.parseInt(id));
+                users.getContainers().add(newContainer);
+                usersRepository.save(users);
+            } catch (NumberFormatException e) {
+                return new ResponseEntity<>(e.toString(), headers, HttpStatus.NOT_FOUND);
+            }
+        } else {
+            status = Integer.valueOf(response.get("status").toString());
+        }
+        return new ResponseEntity<>(response.toString(), headers, HttpStatus.valueOf(status));
+    }
+
     /**
      * The method restarts the container
      *