Add extension to the saved blob

kiennq · Feb 3, 2024 · 570b440 · 570b440
1 parent 03bce61
commit 570b440
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 23 deletions.
diff --git a/csclip/Program.cs b/csclip/Program.cs
@@ -240,44 +240,45 @@ async Task<string> GetDataToFileAsync(SaveDataToFileOptions options)
 
             var stdFormat = CfToStandardFormat(options.cf);
 
-            try
+            if (data.Contains(stdFormat))
             {
-                if (data.Contains(stdFormat))
+                if (options.store_path == null)
                 {
-                    if (options.store_path == null)
-                    {
-                        return (string)await data.GetDataAsync(stdFormat); ;
-                    }
-                    else
+                    return (string)await data.GetDataAsync(stdFormat); ;
+                }
+                else
+                {
+                    var blob = await data.GetDataAsync(stdFormat);
+                    if (blob is RandomAccessStreamReference)
                     {
-                        var rawPayload = await data.GetDataAsync(stdFormat);
-                        if (rawPayload is RandomAccessStreamReference)
+                        var sblob = await (blob as RandomAccessStreamReference).OpenReadAsync();
+                        var ext = MimeTypes.MimeTypeMap.GetExtension(sblob.ContentType);
+                        var fileName = $"{Guid.NewGuid().ToString()}{ext}";
+                        _ = Task.Run(async () =>
                         {
-                            var fileName = Guid.NewGuid().ToString();
-                            _ = Task.Run(async () =>
+                            try
                             {
-                                var sin = await (rawPayload as RandomAccessStreamReference).OpenReadAsync();
                                 var tempFile = await StorageFile.CreateStreamedFileAsync("temp", async (sout) =>
                                 {
-                                    await RandomAccessStream.CopyAndCloseAsync(sin, sout);
+                                    await RandomAccessStream.CopyAndCloseAsync(sblob, sout);
                                 }, null);
 
-                                var path = Path.GetFullPath(options.store_path);
+                                var path = Path.GetFullPath($"{options.store_path}/");
                                 (new FileInfo(path)).Directory.Create();
                                 var folder = await StorageFolder.GetFolderFromPathAsync(path);
-                                _ = tempFile.CopyAsync(folder, fileName, NameCollisionOption.ReplaceExisting);
-                            });
+                                await tempFile.CopyAsync(folder, fileName, NameCollisionOption.ReplaceExisting);
+                            }
+                            catch { }
+                        });
 
-                            return fileName;
-                        }
-                        else if (rawPayload is string)
-                        {
-                            return rawPayload as string;
-                        }
+                        return fileName;
+                    }
+                    else if (blob is string)
+                    {
+                        return blob as string;
                     }
                 }
             }
-            catch {}
 
             return "";
         }

diff --git a/csclip/csclip.csproj b/csclip/csclip.csproj
@@ -57,6 +57,9 @@
       <IncludeAssets>runtime; build; native; contentfiles; analyzers; buildtransitive</IncludeAssets>
       <PrivateAssets>all</PrivateAssets>
     </PackageReference>
+    <PackageReference Include="MediaTypeMap.Core">
+      <Version>2.3.3</Version>
+    </PackageReference>
     <PackageReference Include="Microsoft.VisualStudio.Threading">
       <Version>17.8.14</Version>
     </PackageReference>