feat: create vote model

3 files changed, 33 insertions, 1 deletions

diff --git a/models/polls/poll.schema.ts b/models/polls/poll.schema.ts
index d495e4b..2f39aa1 100644
--- a/models/polls/poll.schema.ts
+++ b/models/polls/poll.schema.ts
@@ -17,7 +17,12 @@ export interface PollSchema extends Document {
 
 export const imageDataSchema = {
   url: String,
-  votes: [Types.ObjectId]
+  votes: [
+    {
+      type: Types.ObjectId,
+      ref: 'vote'
+    }
+  ]
 };
 
 export const pollSchema = new Schema({
diff --git a/models/votes/vote.model.ts b/models/votes/vote.model.ts
new file mode 100644
index 0000000..1dbb146
--- /dev/null
+++ b/models/votes/vote.model.ts
@@ -0,0 +1,5 @@
+import { Model, model } from 'mongoose';
+import { VoteSchema, voteSchema } from './vote.schema';
+
+export default model<VoteSchema, Model<VoteSchema>>('Vote', voteSchema);
+
diff --git a/models/votes/vote.schema.ts b/models/votes/vote.schema.ts
new file mode 100644
index 0000000..36b38f2
--- /dev/null
+++ b/models/votes/vote.schema.ts
@@ -0,0 +1,22 @@
+import { Document, Schema, Types } from 'mongoose';
+import { Vote } from 'which-types';
+
+export interface VoteSchema extends Document, Omit<Vote, '_id'> {
+  password: string;
+}
+
+export const voteSchema = new Schema({
+  userId: {
+    type: Types.ObjectId,
+    ref: 'user'
+  },
+  pollId: {
+    type: Types.ObjectId,
+    ref: 'poll'
+  },
+  which: {
+    type: String,
+    match: /left|right/g
+  }
+}, { timestamps: true });
+