Updated Redact sample to new format.

Author: kurtisvg

dlp/src/main/java/dlp/snippets/RedactImageFile.java

@@ -0,0 +1,86 @@
+/*
+ * Copyright 2019 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package dlp.snippets;
+
+import com.google.cloud.dlp.v2.DlpServiceClient;
+import com.google.privacy.dlp.v2.ByteContentItem;
+import com.google.privacy.dlp.v2.ByteContentItem.BytesType;
+import com.google.privacy.dlp.v2.InfoType;
+import com.google.privacy.dlp.v2.InspectConfig;
+import com.google.privacy.dlp.v2.Likelihood;
+import com.google.privacy.dlp.v2.ProjectName;
+import com.google.privacy.dlp.v2.RedactImageRequest;
+import com.google.privacy.dlp.v2.RedactImageResponse;
+import com.google.protobuf.ByteString;
+import java.io.FileInputStream;
+import java.io.FileOutputStream;
+import java.util.ArrayList;
+import java.util.List;
+
+class RedactImageFile {
+
+  static void redactImageFile(String projectId, String filePath) {
+    // String projectId = "my-project-id";
+    // String filePath = "path/to/image.png";
+
+    // Initialize client that will be used to send requests. This client only needs to be created
+    // once, and can be reused for multiple requests. After completing all of your requests, call
+    // the "close" method on the client to safely clean up any remaining background resources.
+    try (DlpServiceClient dlp = DlpServiceClient.create()) {
+      // Specify the project used for request.
+      ProjectName project = ProjectName.of(projectId);
+
+      // Specify the content to be inspected.
+      ByteString fileBytes = ByteString.readFrom(new FileInputStream(filePath));
+      ByteContentItem byteItem =
+          ByteContentItem.newBuilder().setType(BytesType.IMAGE).setData(fileBytes).build();
+
+      // Specify the type of info and likelihood necessary to redact.
+      List<InfoType> infoTypes = new ArrayList<>();
+      // See https://cloud.google.com/dlp/docs/infotypes-reference for complete list of info types
+      for (String typeName : new String[] {"PHONE_NUMBER", "EMAIL_ADDRESS", "CREDIT_CARD_NUMBER"}) {
+        infoTypes.add(InfoType.newBuilder().setName(typeName).build());
+      }
+      InspectConfig config =
+          InspectConfig.newBuilder()
+              .addAllInfoTypes(infoTypes)
+              .setMinLikelihood(Likelihood.LIKELY)
+              .build();
+
+      // Construct the Redact request to be sent by the client.
+      RedactImageRequest request =
+          RedactImageRequest.newBuilder()
+              .setParent(project.toString())
+              .setByteItem(byteItem)
+              .setInspectConfig(config)
+              .build();
+
+      // Use the client to send the API request.
+      RedactImageResponse response = dlp.redactImage(request);
+
+      // Parse the response and process results.
+      String outputPath = "redacted.png";
+      FileOutputStream redacted = new FileOutputStream(outputPath);
+      redacted.write(response.getRedactedImage().toByteArray());
+      redacted.close();
+      System.out.println("Redacted image written to " + outputPath);
+
+    } catch (Exception e) {
+      System.out.println("Error during inspectFile: \n" + e.toString());
+    }
+  }
+}

dlp/src/test/java/dlp/snippets/RedactTests.java

@@ -0,0 +1,91 @@
+/*
+ * Copyright 2018 Google LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package dlp.snippets;
+
+import static junit.framework.TestCase.assertNotNull;
+import static org.junit.Assert.assertThat;
+
+import java.io.ByteArrayOutputStream;
+import java.io.PrintStream;
+import org.hamcrest.CoreMatchers;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.BeforeClass;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+@RunWith(JUnit4.class)
+public class RedactTests {
+
+  private ByteArrayOutputStream bout;
+
+  private static final String PROJECT_ID = System.getenv("GOOGLE_CLOUD_PROJECT");
+
+  private static void requireEnvVar(String varName) {
+    assertNotNull(
+        System.getenv(varName),
+        "Environment variable '%s' is required to perform these tests.".format(varName)
+    );
+  }
+
+  @BeforeClass
+  public static void checkRequirements() {
+    requireEnvVar("GOOGLE_APPLICATION_CREDENTIALS");
+    requireEnvVar("GOOGLE_CLOUD_PROJECT");
+  }
+
+  @Before
+  public void beforeTest() {
+    bout = new ByteArrayOutputStream();
+    System.setOut(new PrintStream(bout));
+  }
+
+  @After
+  public void tearDown() {
+    System.setOut(null);
+    bout.reset();
+  }
+
+  @Test
+  public void testInspectString() {
+    InspectString.inspectString(PROJECT_ID,  "I'm Gary and my email is [email protected]");
+
+    String output = bout.toString();
+    assertThat(output, CoreMatchers.containsString("Info type: EMAIL_ADDRESS"));
+  }
+
+  @Test
+  public void textInspectTestFile() {
+    InspectTextFile.inspectTextFile(PROJECT_ID,  "src/test/resources/test.txt");
+
+    String output = bout.toString();
+    assertThat(output, CoreMatchers.containsString("Info type: PHONE_NUMBER"));
+    assertThat(output, CoreMatchers.containsString("Info type: EMAIL_ADDRESS"));
+  }
+
+
+  @Test
+  public void testInspectImageFile() {
+    InspectImageFile.inspectImageFile(PROJECT_ID,  "src/test/resources/test.png");
+
+    String output = bout.toString();
+    assertThat(output, CoreMatchers.containsString("Info type: PHONE_NUMBER"));
+    assertThat(output, CoreMatchers.containsString("Info type: EMAIL_ADDRESS"));
+  }
+
+}