Merge pull request #32 from vania-pooh/master

Ability to get and set image from the clipboard

Author: vania-pooh

pom.xml

@@ -17,7 +17,7 @@
         <maven.compiler.target>11</maven.compiler.target>
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
         <swagger.version>1.6.4</swagger.version>
-        <openapi-spec.version>1.0.14</openapi-spec.version>
+        <openapi-spec.version>1.0.15</openapi-spec.version>
         <openapi-plugin.version>5.1.0</openapi-plugin.version>
         <jackson.version>2.13.1</jackson.version>
         <junit-jupiter.version>5.8.2</junit-jupiter.version>

src/main/java/com/aerokube/lightning/extensions/MoonCommands.java

@@ -1,11 +1,15 @@
 package com.aerokube.lightning.extensions;
 
+import com.aerokube.lightning.FileUtils;
 import com.aerokube.lightning.WebDriver;
+import com.aerokube.lightning.WebDriverException;
 import com.aerokube.lightning.WebDriverExtension;
 import com.aerokube.lightning.api.AerokubeApi;
-import com.aerokube.lightning.model.StringResponse;
+import com.aerokube.lightning.model.ClipboardData;
 
 import javax.annotation.Nonnull;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Path;
 import java.util.Base64;
 import java.util.List;
 
@@ -19,20 +23,48 @@ public MoonCommands(WebDriver webDriver) {
     }
 
     @Nonnull
-    public MoonCommands updateClipboard(@Nonnull byte[] content) {
+    public MoonCommands updateClipboardText(@Nonnull String text) {
         execute(() -> {
-            String encodedContent = Base64.getEncoder().encodeToString(content);
-            aerokubeApi.updateClipboard(getSessionId(), new StringResponse().value(encodedContent));
+            aerokubeApi.updateClipboard(getSessionId(), new ClipboardData().value(text));
             return null;
         });
         return this;
     }
 
     @Nonnull
-    public byte[] getClipboard() {
+    public MoonCommands updateClipboardImage(@Nonnull Path image) {
+        execute(() -> {
+            String encodedImage = FileUtils.encodeFileToBase64(image);
+            aerokubeApi.updateClipboard(
+                    getSessionId(),
+                    new ClipboardData()
+                            .value(encodedImage)
+                            .media(ClipboardData.MediaEnum.IMAGE_PNG)
+            );
+            return null;
+        });
+        return this;
+    }
+
+    @Nonnull
+    public String getClipboardText() {
         return execute(() -> {
-            String encodedBytes = aerokubeApi.getClipboard(getSessionId()).getValue();
-            return Base64.getDecoder().decode(encodedBytes);
+            ClipboardData clipboard = aerokubeApi.getClipboard(getSessionId());
+            if (clipboard.getMedia() != null) {
+                throw new WebDriverException(String.format("Received non-text data from the clipboard: %s", clipboard.getMedia()));
+            }
+            return clipboard.getValue();
+        });
+    }
+
+    @Nonnull
+    public byte[] getClipboardImage() {
+        return execute(() -> {
+            ClipboardData clipboard = aerokubeApi.getClipboard(getSessionId());
+            if (clipboard.getMedia() == null || !ClipboardData.MediaEnum.IMAGE_PNG.equals(clipboard.getMedia())) {
+                throw new WebDriverException("Received non-image data from the clipboard");
+            }
+            return Base64.getDecoder().decode(clipboard.getValue().getBytes(StandardCharsets.UTF_8));
         });
     }