2 files changed, 23 insertions, 0 deletions
diff --git a/src/main/java/com/blog/web/models/Article.java b/src/main/java/com/blog/web/models/Article.java
index 2682994..42d27a1 100644
--- a/src/main/java/com/blog/web/models/Article.java
+++ b/src/main/java/com/blog/web/models/Article.java
@@ -1,5 +1,6 @@
 package com.blog.web.models;
 
+import com.blog.web.dto.ArticleDto;
 import jakarta.persistence.*;
 import org.hibernate.annotations.CreationTimestamp;
 import org.hibernate.annotations.UpdateTimestamp;
@@ -42,6 +43,16 @@ public class Article {
 
     public Article() {};
 
+    public Article(ArticleDto articleDto) {
+        this.id = articleDto.getId();
+        this.title = articleDto.getTitle();
+        this.photoUrl = articleDto.getPhotoUrl();
+        this.content = articleDto.getContent();
+        this.createdBy = articleDto.getCreatedBy();
+        this.createdOn = articleDto.getCreatedOn();
+        this.updatedOn = articleDto.getUpdatedOn();
+    }
+
     public Long getId() {
         return id;
     }
diff --git a/src/main/java/com/blog/web/models/UserEntity.java b/src/main/java/com/blog/web/models/UserEntity.java
index ac7b288..d121580 100644
--- a/src/main/java/com/blog/web/models/UserEntity.java
+++ b/src/main/java/com/blog/web/models/UserEntity.java
@@ -1,9 +1,12 @@
 package com.blog.web.models;
 
 import jakarta.persistence.*;
+import org.springframework.security.core.authority.SimpleGrantedAuthority;
+import org.springframework.security.core.userdetails.User;
 
 import java.util.ArrayList;
 import java.util.List;
+import java.util.stream.Collectors;
 
 @Entity(name = "users")
 // Named UserEntity to prevent conflicts with Java User object
@@ -27,6 +30,15 @@ public class UserEntity {
         return this.id == user.getId();
     }
 
+    public User toSecurityUser() {
+        return new User(
+                this.getEmail(),
+                this.getPassword(),
+                this.getRoles().stream().map((role) -> new SimpleGrantedAuthority(role.getName()))
+                        .collect(Collectors.toList())
+        );
+    }
+
     public Long getId() {
         return id;
     }