bring models back

Author: khvn26

src/main/java/com/flagsmith/flagengine/environments/EnvironmentModel.java

@@ -0,0 +1,28 @@
+package com.flagsmith.flagengine.environments;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.flagsmith.flagengine.features.FeatureStateModel;
+import com.flagsmith.flagengine.identities.IdentityModel;
+import com.flagsmith.flagengine.projects.ProjectModel;
+import com.flagsmith.utils.models.BaseModel;
+import java.util.List;
+import lombok.Data;
+
+@Data
+public class EnvironmentModel extends BaseModel {
+  private Integer id;
+
+  @JsonProperty("api_key")
+  private String apiKey;
+
+  @JsonProperty("name")
+  private String name;
+
+  private ProjectModel project;
+
+  @JsonProperty("feature_states")
+  private List<FeatureStateModel> featureStates;
+
+  @JsonProperty("identity_overrides")
+  private List<IdentityModel> identityOverrides;
+}

src/main/java/com/flagsmith/flagengine/features/FeatureModel.java

@@ -0,0 +1,10 @@
+package com.flagsmith.flagengine.features;
+
+import lombok.Data;
+
+@Data
+public class FeatureModel {
+  private Integer id;
+  private String name;
+  private String type;
+}

src/main/java/com/flagsmith/flagengine/features/FeatureSegmentModel.java

@@ -0,0 +1,9 @@
+package com.flagsmith.flagengine.features;
+
+import com.flagsmith.flagengine.utils.models.BaseModel;
+import lombok.Data;
+
+@Data
+public class FeatureSegmentModel extends BaseModel {
+  private Integer priority;
+}

src/main/java/com/flagsmith/flagengine/features/FeatureStateModel.java

@@ -0,0 +1,23 @@
+package com.flagsmith.flagengine.features;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.flagsmith.utils.models.BaseModel;
+import java.util.List;
+import java.util.UUID;
+import lombok.Data;
+
+@Data
+public class FeatureStateModel extends BaseModel {
+  private FeatureModel feature;
+  private Boolean enabled;
+  @JsonProperty("django_id")
+  private Integer djangoId;
+  @JsonProperty("featurestate_uuid")
+  private String featurestateUuid = UUID.randomUUID().toString();
+  @JsonProperty("multivariate_feature_state_values")
+  private List<MultivariateFeatureStateValueModel> multivariateFeatureStateValues;
+  @JsonProperty("feature_state_value")
+  private Object value;
+  @JsonProperty("feature_segment")
+  private FeatureSegmentModel featureSegment;
+}

src/main/java/com/flagsmith/flagengine/features/MultivariateFeatureOptionModel.java

@@ -0,0 +1,9 @@
+package com.flagsmith.flagengine.features;
+
+import com.flagsmith.utils.models.BaseModel;
+import lombok.Data;
+
+@Data
+public class MultivariateFeatureOptionModel extends BaseModel {
+  private String value;
+}

src/main/java/com/flagsmith/flagengine/features/MultivariateFeatureStateValueModel.java

@@ -0,0 +1,17 @@
+package com.flagsmith.flagengine.features;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.flagsmith.utils.models.BaseModel;
+import java.util.UUID;
+import lombok.Data;
+
+@Data
+public class MultivariateFeatureStateValueModel extends BaseModel {
+  @JsonProperty("multivariate_feature_option")
+  private MultivariateFeatureOptionModel multivariateFeatureOption;
+  @JsonProperty("percentage_allocation")
+  private Float percentageAllocation;
+  private Integer id;
+  @JsonProperty("mv_fs_value_uuid")
+  private String mvFsValueUuid = UUID.randomUUID().toString();
+}

src/main/java/com/flagsmith/flagengine/identities/IdentityModel.java

@@ -0,0 +1,23 @@
+package com.flagsmith.flagengine.identities;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.flagsmith.flagengine.features.FeatureStateModel;
+import com.flagsmith.utils.models.BaseModel;
+import java.sql.Date;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.UUID;
+import lombok.Data;
+
+@Data
+public class IdentityModel extends BaseModel {
+  @JsonProperty("django_id")
+  private Integer djangoId;
+  private String identifier;
+  @JsonProperty("created_date")
+  private Date createdDate;
+  @JsonProperty("identity_uuid")
+  private String identityUuid = UUID.randomUUID().toString();
+  @JsonProperty("identity_features")
+  private List<FeatureStateModel> identityFeatures = new ArrayList<>();
+}

src/main/java/com/flagsmith/flagengine/projects/ProjectModel.java

@@ -0,0 +1,11 @@
+package com.flagsmith.flagengine.projects;
+
+import com.flagsmith.flagengine.segments.SegmentModel;
+import com.flagsmith.utils.models.BaseModel;
+import java.util.List;
+import lombok.Data;
+
+@Data
+public class ProjectModel extends BaseModel {
+  private List<SegmentModel> segments;
+}

src/main/java/com/flagsmith/flagengine/segments/SegmentConditionModel.java

@@ -0,0 +1,13 @@
+package com.flagsmith.flagengine.segments;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.flagsmith.flagengine.segments.constants.SegmentConditions;
+import lombok.Data;
+
+@Data
+public class SegmentConditionModel {
+  private SegmentConditions operator;
+  private String value;
+  @JsonProperty("property_")
+  private String property;
+}

src/main/java/com/flagsmith/flagengine/segments/SegmentModel.java

@@ -0,0 +1,16 @@
+package com.flagsmith.flagengine.segments;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.flagsmith.flagengine.features.FeatureStateModel;
+import com.flagsmith.utils.models.BaseModel;
+import java.util.List;
+import lombok.Data;
+
+@Data
+public class SegmentModel extends BaseModel {
+  private Integer id;
+  private String name;
+  private List<SegmentRuleModel> rules;
+  @JsonProperty("feature_states")
+  private List<FeatureStateModel> featureStates;
+}