Merge pull request #52 from openhacku-team-a/feture/upload-img-method

[Add] image uploader
a-company-jp · Mar 15, 2024 · a67ce40 · a67ce40
2 parents 86c23d6 + 2b84d01
commit a67ce40
Show file tree

Hide file tree

Showing 2 changed files with 41 additions and 14 deletions.
diff --git a/src/hooks/useApiPBClient.ts b/src/hooks/useApiPBClient.ts
@@ -81,7 +81,44 @@ const useApiPBClient = () => {
         return { data: null, unauthorized: false, error: resp.statusText };
       });
   };
-  return { get, put, post };
+
+  const imageUpload = async (
+    userId: string,
+    file: File,
+  ): Promise<BackendResponse> => {
+    const contentType = getFileContentType(file);
+
+    headers["Content-Type"] = contentType;
+
+    return await axios
+      .put(`/user/${userId}/icon`, file, { headers })
+      .then((resp) => {
+        if (resp.status < 210) {
+          return { data: resp.data, unauthorized: false, error: null };
+        }
+        if (resp.status === 401) {
+          return { data: null, unauthorized: true, error: resp.status };
+        }
+        return { data: null, unauthorized: false, error: resp.statusText };
+      });
+  };
+  return { get, put, post, imageUpload };
 };
 
 export default useApiPBClient;
+
+function getFileContentType(file: File): string {
+  const extension = file.name.split(".").pop();
+  switch (extension) {
+    case "png":
+      return "image/png";
+    case "jpg":
+    case "jpeg":
+      return "image/jpeg";
+    case "gif":
+      return "image/gif";
+    // 追加の拡張子に対する処理を追加できます
+    default:
+      return "application/octet-stream"; // デフォルトのContent-Type
+  }
+}
diff --git a/src/hooks/useFetchUser.ts b/src/hooks/useFetchUser.ts
@@ -60,19 +60,9 @@ const useFetchUser = () => {
     });
   };
 
-  //   const icon = async (userId: string, file: File) => {
-  //     return await client.put(
-  //         `/user/${userId}/icon`,
-  //         ).then((resp) => {
-  //         if (resp.unauthorized) {
-  //             throw new Error("unauthorized");
-  //           }
-  //           if (resp.error) {
-  //             throw new Error(resp.error);
-  //           }
-  //           return UserInfoResponse.fromBinary(resp.data)
-  //     })
-  //   }
+  const iconUpload = async (userId: string, file: File) => {
+    return await client.imageUpload(userId, file);
+  };
 
   return { me, get, list, create };
 };