Newsfeed management Updated and required changes (PalisadoesFoundation#1383)

* implementation of the newsfeed feature

* implementation of the newsfeed feature

Author: aashimawadhwa

schema.graphql

@@ -623,6 +623,7 @@ type Post {
   likeCount: Int
   likedBy: [User]
   organization: Organization!
+  pinned: Boolean
   text: String!
   title: String
   videoUrl: URL
@@ -669,10 +670,10 @@ enum PostOrderByInput {
 }
 
 input PostUpdateInput {
-  imageUrl: URL
+  imageUrl: String
   text: String
   title: String
-  videoUrl: URL
+  videoUrl: String
 }
 
 input PostWhereInput {

src/app.ts

@@ -68,6 +68,8 @@ app.use(
 );
 
 app.use("/images", express.static(path.join(__dirname, "./../images")));
+app.use("/videos", express.static(path.join(__dirname, "./../videos")));
+
 app.use(requestContext.middleware());
 
 if (process.env.NODE_ENV !== "production")

src/models/EncodedVideo.ts

@@ -0,0 +1,39 @@
+import type { Types, Model } from "mongoose";
+import { Schema, model, models } from "mongoose";
+/**
+ * This is an interface that represents a database(MongoDB) document for Encoded Video.
+ */
+export interface InterfaceEncodedVideo {
+  _id: Types.ObjectId;
+  fileName: string;
+  content: string;
+  numberOfUses: number;
+}
+/**
+ * This describes the schema for a `encodedVideo` that corresponds to `InterfaceEncodedVideo` document.
+ * @param fileName - File name.
+ * @param content - Content.
+ * @param numberOfUses - Number of Uses.
+ */
+const encodedVideoSchema = new Schema({
+  fileName: {
+    type: String,
+    required: true,
+  },
+  content: {
+    type: String,
+    required: true,
+  },
+  numberOfUses: {
+    type: Number,
+    required: true,
+    default: 1,
+  },
+});
+
+const encodedVideoModel = (): Model<InterfaceEncodedVideo> =>
+  model<InterfaceEncodedVideo>("EncodedVideo", encodedVideoSchema);
+
+// This syntax is needed to prevent Mongoose OverwriteModelError while running tests.
+export const EncodedVideo = (models.EncodedVideo ||
+  encodedVideoModel()) as ReturnType<typeof encodedVideoModel>;

src/models/Post.ts

@@ -13,7 +13,7 @@ export interface InterfacePost {
   status: string;
   createdAt: Date;
   imageUrl: string | undefined | null;
-  videoUrl: string | undefined;
+  videoUrl: string | undefined | null;
   creator: PopulatedDoc<InterfaceUser & Document>;
   organization: PopulatedDoc<InterfaceOrganization & Document>;
   likedBy: PopulatedDoc<InterfaceUser & Document>[];

src/resolvers/Mutation/createPost.ts

@@ -9,8 +9,10 @@ import {
 } from "../../constants";
 import { isValidString } from "../../libraries/validators/validateString";
 import { uploadEncodedImage } from "../../utilities/encodedImageStorage/uploadEncodedImage";
+import { uploadEncodedVideo } from "../../utilities/encodedVideoStorage/uploadEncodedVideo";
 import { findOrganizationsInCache } from "../../services/OrganizationCache/findOrganizationsInCache";
 import { cacheOrganizations } from "../../services/OrganizationCache/cacheOrganizations";
+
 /**
  * This function enables to create a post.
  * @param _parent - parent of current request
@@ -65,10 +67,18 @@ export const createPost: MutationResolvers["createPost"] = async (
     );
   }
 
-  let uploadImageFileName;
+  let uploadImageFileName = null;
+  let uploadVideoFileName = null;
 
   if (args.file) {
-    uploadImageFileName = await uploadEncodedImage(args.file, null);
+    const dataUrlPrefix = "data:";
+    if (args.file.startsWith(dataUrlPrefix + "image/")) {
+      uploadImageFileName = await uploadEncodedImage(args.file, null);
+    } else if (args.file.startsWith(dataUrlPrefix + "video/")) {
+      uploadVideoFileName = await uploadEncodedVideo(args.file, null);
+    } else {
+      throw new Error("Unsupported file type.");
+    }
   }
 
   // Checks if the recieved arguments are valid according to standard input norms
@@ -118,7 +128,8 @@ export const createPost: MutationResolvers["createPost"] = async (
     pinned: args.data.pinned ? true : false,
     creator: context.userId,
     organization: args.data.organizationId,
-    imageUrl: args.file ? uploadImageFileName : null,
+    imageUrl: uploadImageFileName,
+    videoUrl: uploadVideoFileName,
   });
 
   if (args.data.pinned) {
@@ -141,8 +152,11 @@ export const createPost: MutationResolvers["createPost"] = async (
   // Returns createdPost.
   return {
     ...createdPost.toObject(),
-    imageUrl: createdPost.imageUrl
+    imageUrl: uploadImageFileName
       ? `${context.apiRootUrl}${uploadImageFileName}`
       : null,
+    videoUrl: uploadVideoFileName
+      ? `${context.apiRootUrl}${uploadVideoFileName}`
+      : null,
   };
 };

src/resolvers/Mutation/updatePost.ts

@@ -8,6 +8,8 @@ import {
   LENGTH_VALIDATION_ERROR,
 } from "../../constants";
 import { isValidString } from "../../libraries/validators/validateString";
+import { uploadEncodedImage } from "../../utilities/encodedImageStorage/uploadEncodedImage";
+import { uploadEncodedVideo } from "../../utilities/encodedVideoStorage/uploadEncodedVideo";
 
 export const updatePost: MutationResolvers["updatePost"] = async (
   _parent,
@@ -51,6 +53,20 @@ export const updatePost: MutationResolvers["updatePost"] = async (
     );
   }
 
+  if (args.data?.imageUrl && args.data?.imageUrl !== null) {
+    args.data.imageUrl = await uploadEncodedImage(
+      args.data.imageUrl,
+      post.imageUrl
+    );
+  }
+
+  if (args.data?.videoUrl && args.data?.videoUrl !== null) {
+    args.data.videoUrl = await uploadEncodedVideo(
+      args.data.videoUrl,
+      post.videoUrl
+    );
+  }
+
   // Checks if the recieved arguments are valid according to standard input norms
   const validationResultTitle = isValidString(args.data?.title ?? "", 256);
   const validationResultText = isValidString(args.data?.text ?? "", 500);

src/resolvers/Query/postsByOrganization.ts

@@ -25,6 +25,7 @@ export const postsByOrganization: QueryResolvers["postsByOrganization"] =
     const postsWithImageURLResolved = postsInOrg.map((post) => ({
       ...post,
       imageUrl: post.imageUrl ? `${context.apiRootUrl}${post.imageUrl}` : null,
+      videoUrl: post.videoUrl ? `${context.apiRootUrl}${post.videoUrl}` : null,
     }));
 
     return postsWithImageURLResolved;

src/resolvers/Query/postsByOrganizationConnection.ts

@@ -57,6 +57,9 @@ export const postsByOrganizationConnection: QueryResolvers["postsByOrganizationC
         ? `${context.apiRootUrl}${post.imageUrl}`
         : null;
 
+      post.videoUrl = post.videoUrl
+        ? `${context.apiRootUrl}${post.videoUrl}`
+        : null;
       return post;
     });
 

src/typeDefs/inputs.ts

@@ -375,7 +375,7 @@ export const inputs = gql`
   input PostUpdateInput {
     text: String
     title: String
-    imageUrl: URL
-    videoUrl: URL
+    imageUrl: String
+    videoUrl: String
   }
 `;

src/typeDefs/types.ts

@@ -303,6 +303,7 @@ export const types = gql`
     comments: [Comment]
     likeCount: Int
     commentCount: Int
+    pinned: Boolean
   }
 
   """

src/types/generatedGraphQLTypes.ts

@@ -1130,6 +1130,7 @@ export type Post = {
   likeCount?: Maybe<Scalars['Int']>;
   likedBy?: Maybe<Array<Maybe<User>>>;
   organization: Organization;
+  pinned?: Maybe<Scalars['Boolean']>;
   text: Scalars['String'];
   title?: Maybe<Scalars['String']>;
   videoUrl?: Maybe<Scalars['URL']>;
@@ -1174,10 +1175,10 @@ export type PostOrderByInput =
   | 'videoUrl_DESC';
 
 export type PostUpdateInput = {
-  imageUrl?: InputMaybe<Scalars['URL']>;
+  imageUrl?: InputMaybe<Scalars['String']>;
   text?: InputMaybe<Scalars['String']>;
   title?: InputMaybe<Scalars['String']>;
-  videoUrl?: InputMaybe<Scalars['URL']>;
+  videoUrl?: InputMaybe<Scalars['String']>;
 };
 
 export type PostWhereInput = {
@@ -2473,6 +2474,7 @@ export type PostResolvers<ContextType = any, ParentType extends ResolversParentT
   likeCount?: Resolver<Maybe<ResolversTypes['Int']>, ParentType, ContextType>;
   likedBy?: Resolver<Maybe<Array<Maybe<ResolversTypes['User']>>>, ParentType, ContextType>;
   organization?: Resolver<ResolversTypes['Organization'], ParentType, ContextType>;
+  pinned?: Resolver<Maybe<ResolversTypes['Boolean']>, ParentType, ContextType>;
   text?: Resolver<ResolversTypes['String'], ParentType, ContextType>;
   title?: Resolver<Maybe<ResolversTypes['String']>, ParentType, ContextType>;
   videoUrl?: Resolver<Maybe<ResolversTypes['URL']>, ParentType, ContextType>;

src/utilities/encodedVideoStorage/deletePreviousVideo.ts

@@ -0,0 +1,29 @@
+import { unlink } from "fs/promises";
+import path from "path";
+import { EncodedVideo } from "../../models/EncodedVideo";
+
+export const deletePreviousVideo = async (
+  videoToBeDeletedPath: string
+): Promise<void> => {
+  const videoToBeDeleted = await EncodedVideo.findOne({
+    fileName: videoToBeDeletedPath!,
+  });
+
+  if (videoToBeDeleted?.numberOfUses === 1) {
+    await unlink(path.join(__dirname, "../../../" + videoToBeDeleted.fileName));
+    await EncodedVideo.deleteOne({
+      fileName: videoToBeDeletedPath,
+    });
+  }
+
+  await EncodedVideo.findOneAndUpdate(
+    {
+      fileName: videoToBeDeletedPath,
+    },
+    {
+      $inc: {
+        numberOfUses: -1,
+      },
+    }
+  );
+};

src/utilities/encodedVideoStorage/encodedVideoExtensionCheck.ts

@@ -0,0 +1,15 @@
+export const encodedVideoExtentionCheck = (encodedUrl: string): boolean => {
+  const extension = encodedUrl.substring(
+    "data:".length,
+    encodedUrl.indexOf(";base64")
+  );
+
+  console.log(extension);
+
+  const isValidVideo = extension === "video/mp4";
+  if (isValidVideo) {
+    return true;
+  }
+
+  return false;
+};

src/utilities/encodedVideoStorage/uploadEncodedVideo.ts

@@ -0,0 +1,72 @@
+import shortid from "shortid";
+import * as fs from "fs";
+import { writeFile } from "fs/promises";
+import { encodedVideoExtentionCheck } from "./encodedVideoExtensionCheck";
+import { errors, requestContext } from "../../libraries";
+import { INVALID_FILE_TYPE } from "../../constants";
+import { EncodedVideo } from "../../models/EncodedVideo";
+import path from "path";
+import { deletePreviousVideo } from "./deletePreviousVideo";
+
+export const uploadEncodedVideo = async (
+  encodedVideoURL: string,
+  // eslint-disable-next-line @typescript-eslint/no-unused-vars
+  previousVideoPath?: string | null
+): Promise<string> => {
+  const isURLValidVideo = encodedVideoExtentionCheck(encodedVideoURL);
+
+  if (!isURLValidVideo) {
+    throw new errors.InvalidFileTypeError(
+      requestContext.translate(INVALID_FILE_TYPE.MESSAGE),
+      INVALID_FILE_TYPE.CODE,
+      INVALID_FILE_TYPE.PARAM
+    );
+  }
+
+  const encodedVideoAlreadyExist = await EncodedVideo.findOne({
+    content: encodedVideoURL,
+  });
+
+  if (previousVideoPath) {
+    await deletePreviousVideo(previousVideoPath);
+  }
+
+  if (encodedVideoAlreadyExist) {
+    await EncodedVideo.findOneAndUpdate(
+      {
+        content: encodedVideoURL,
+      },
+      {
+        $inc: {
+          numberOfUses: 1,
+        },
+      }
+    );
+    return encodedVideoAlreadyExist.fileName;
+  }
+
+  let id = shortid.generate();
+
+  id = "videos/" + id + "video.mp4";
+
+  const uploadedEncodedVideo = await EncodedVideo.create({
+    fileName: id,
+    content: encodedVideoURL,
+  });
+
+  const data = encodedVideoURL.replace(/^data:video\/\w+;base64,/, "");
+
+  const buf = Buffer.from(data, "base64");
+
+  if (!fs.existsSync(path.join(__dirname, "../../../videos"))) {
+    fs.mkdir(path.join(__dirname, "../../../videos"), (error) => {
+      if (error) {
+        throw error;
+      }
+    });
+  }
+
+  await writeFile(path.join(__dirname, "../../../" + id), buf);
+
+  return uploadedEncodedVideo.fileName;
+};