-
-
Notifications
You must be signed in to change notification settings - Fork 1
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
RaaG : Ability to forward text and images to OpenAI API
- Loading branch information
Showing
3 changed files
with
88 additions
and
7 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,75 @@ | ||
| package ai.dragon.util; | ||
|
|
||
| import java.util.Base64; | ||
|
|
||
| public class DataUrlUtil { | ||
| public static String getMimeType(String dataUrl) { | ||
| String[] parts = dataUrl.split(";"); | ||
| if (parts.length < 2) { | ||
| return "octet/stream"; | ||
| } | ||
| String mimeType = parts[0].split(":")[1]; | ||
| return mimeType; | ||
| } | ||
|
|
||
| public static String getDataBytesString(String dataUrl) { | ||
| return dataUrl.substring(dataUrl.indexOf(",") + 1); | ||
| } | ||
|
|
||
| public static String getImageType(String base64String) { | ||
| return getImageType(getDataBytes(getDataBytesString(base64String))); | ||
| } | ||
|
|
||
| private static byte[] getDataBytes(String base64String) { | ||
| return Base64.getDecoder().decode(base64String); | ||
| } | ||
|
|
||
| // Forked from : | ||
| // https://nsclass.github.io/2017/03/13/java-an-example-code-to-extract-mime-type-from-base64-string-of-an-image/ | ||
| private static String getImageType(byte[] data) { | ||
| // filetype magic number(hex) | ||
| // jpg FF D8 FF | ||
| // gif 47 49 46 38 | ||
| // png 89 50 4E 47 0D 0A 1A 0A | ||
| // bmp 42 4D | ||
| // tiff(LE) 49 49 2A 00 | ||
| // tiff(BE) 4D 4D 00 2A | ||
| final byte[] pngPattern = new byte[] { (byte) 0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A }; | ||
| final byte[] jpgPattern = new byte[] { (byte) 0xFF, (byte) 0xD8, (byte) 0xFF }; | ||
| final byte[] gifPattern = new byte[] { 0x47, 0x49, 0x46, 0x38 }; | ||
| final byte[] bmpPattern = new byte[] { 0x42, 0x4D }; | ||
| final byte[] tiffLEPattern = new byte[] { 0x49, 0x49, 0x2A, 0x00 }; | ||
| final byte[] tiffBEPattern = new byte[] { 0x4D, 0x4D, 0x00, 0x2A }; | ||
| if (isMatch(pngPattern, data)) { | ||
| return "image/png"; | ||
| } | ||
| if (isMatch(jpgPattern, data)) { | ||
| return "image/jpg"; | ||
| } | ||
| if (isMatch(gifPattern, data)) { | ||
| return "image/gif"; | ||
| } | ||
| if (isMatch(bmpPattern, data)) { | ||
| return "image/bmp"; | ||
| } | ||
| if (isMatch(tiffLEPattern, data)) { | ||
| return "image/tif"; | ||
| } | ||
| if (isMatch(tiffBEPattern, data)) { | ||
| return "image/tif"; | ||
| } | ||
| return "image/png"; | ||
| } | ||
|
|
||
| private static boolean isMatch(byte[] pattern, byte[] data) { | ||
| if (pattern.length <= data.length) { | ||
| for (int idx = 0; idx < pattern.length; ++idx) { | ||
| if (pattern[idx] != data[idx]) { | ||
| return false; | ||
| } | ||
| } | ||
| return true; | ||
| } | ||
| return false; | ||
| } | ||
| } |