migrate code from googleapis/java-webrisk

webrisk/pom.xml

@@ -0,0 +1,60 @@
+<?xml version='1.0' encoding='UTF-8'?>
+<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+  <modelVersion>4.0.0</modelVersion>
+  <groupId>com.google.cloud</groupId>
+  <artifactId>webrisk-snippets</artifactId>
+  <packaging>jar</packaging>
+  <name>Google Web Risk Snippets</name>
+  <url>https://github.com/googleapis/java-webrisk</url>
+
+  <!--
+    The parent pom defines common style checks and testing strategies for our samples.
+    Removing or replacing it should not affect the execution of the samples in anyway.
+  -->
+  <parent>
+    <groupId>com.google.cloud.samples</groupId>
+    <artifactId>shared-configuration</artifactId>
+    <version>1.2.0</version>
+  </parent>
+
+  <properties>
+    <maven.compiler.target>1.8</maven.compiler.target>
+    <maven.compiler.source>1.8</maven.compiler.source>
+    <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+  </properties>
+
+
+  <!-- [START webrisk_install_with_bom] -->
+  <dependencyManagement>
+    <dependencies>
+      <dependency>
+        <groupId>com.google.cloud</groupId>
+        <artifactId>libraries-bom</artifactId>
+        <version>26.1.4</version>
+        <type>pom</type>
+        <scope>import</scope>
+      </dependency>
+    </dependencies>
+  </dependencyManagement>
+
+  <dependencies>
+    <dependency>
+      <groupId>com.google.cloud</groupId>
+      <artifactId>google-cloud-webrisk</artifactId>
+    </dependency>
+    <!-- [END webrisk_install_with_bom] -->
+
+    <dependency>
+      <groupId>junit</groupId>
+      <artifactId>junit</artifactId>
+      <version>4.13.2</version>
+      <scope>test</scope>
+    </dependency>
+    <dependency>
+      <groupId>com.google.truth</groupId>
+      <artifactId>truth</artifactId>
+      <version>1.1.3</version>
+      <scope>test</scope>
+    </dependency>
+  </dependencies>
+</project>

webrisk/src/main/java/webrisk/SearchUriExample.java

@@ -0,0 +1,53 @@
+/*
+ * Copyright 2020 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 webrisk;
+
+import com.google.cloud.webrisk.v1.WebRiskServiceClient;
+import com.google.webrisk.v1.SearchUrisRequest;
+import com.google.webrisk.v1.SearchUrisResponse;
+import com.google.webrisk.v1.ThreatType;
+import java.io.IOException;
+
+public class SearchUriExample {
+
+  public static void searchUriExample() throws IOException {
+    // The URL to be searched
+    String uri = "http://testsafebrowsing.appspot.com/s/malware.html";
+    SearchUrisResponse response = searchUriExample(uri);
+  }
+
+  // [START webrisk_search_uri]
+  public static SearchUrisResponse searchUriExample(String uri) throws IOException {
+    // create-webrisk-client
+    try (WebRiskServiceClient webRiskServiceClient = WebRiskServiceClient.create()) {
+      // Query the url for a specific threat type
+      SearchUrisRequest searchUrisRequest =
+          SearchUrisRequest.newBuilder().addThreatTypes(ThreatType.MALWARE).setUri(uri).build();
+      SearchUrisResponse searchUrisResponse = webRiskServiceClient.searchUris(searchUrisRequest);
+      webRiskServiceClient.shutdownNow();
+      if (!searchUrisResponse.getThreat().getThreatTypesList().isEmpty()) {
+        System.out.println("The URL has the following threat : ");
+        System.out.println(searchUrisResponse);
+      } else {
+        System.out.println("The URL is safe!");
+      }
+
+      return searchUrisResponse;
+    }
+  }
+  // [END webrisk_search_uri]
+}

webrisk/src/main/java/webrisk/SubmitUriExample.java

@@ -0,0 +1,49 @@
+/*
+ * Copyright 2020 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 webrisk;
+
+import com.google.cloud.webrisk.v1.WebRiskServiceClient;
+import com.google.webrisk.v1.CreateSubmissionRequest;
+import com.google.webrisk.v1.Submission;
+import java.io.IOException;
+
+public class SubmitUriExample {
+
+  public static void submitUriExample() throws IOException {
+    // The URL to be submitted
+    String uri = "http://testsafebrowsing.appspot.com/s/malware.html";
+    Submission response = submitUriExample(uri);
+  }
+
+  // [START webrisk_submit_uri]
+  public static Submission submitUriExample(String uri) throws IOException {
+    // create-webrisk-client
+    try (WebRiskServiceClient webRiskServiceClient = WebRiskServiceClient.create()) {
+      Submission submission = Submission.newBuilder().setUri(uri).build();
+      CreateSubmissionRequest submissionRequest =
+          CreateSubmissionRequest.newBuilder()
+              .setParent("projects/your-project-id")
+              .setSubmission(submission)
+              .build();
+      Submission submissionResponse = webRiskServiceClient.createSubmission(submissionRequest);
+      webRiskServiceClient.shutdownNow();
+      System.out.println("The submitted " + submissionResponse);
+      return submissionResponse;
+    }
+  }
+  // [END webrisk_submit_uri]
+}

webrisk/src/test/java/webrisk/SearchUriExampleTest.java

@@ -0,0 +1,47 @@
+/*
+ * Copyright 2020 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 webrisk;
+
+import com.google.common.truth.Truth;
+import com.google.webrisk.v1.SearchUrisResponse;
+import com.google.webrisk.v1.ThreatType;
+import java.io.IOException;
+import java.util.List;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+@RunWith(JUnit4.class)
+public class SearchUriExampleTest {
+  @Test
+  public void testSearchWithThreat() throws IOException {
+    // The URL to be searched
+    String uri = "http://testsafebrowsing.appspot.com/s/malware.html";
+    SearchUrisResponse actualResponse = SearchUriExample.searchUriExample(uri);
+    List<ThreatType> type = actualResponse.getThreat().getThreatTypesList();
+    Truth.assertThat(type).contains(ThreatType.MALWARE);
+  }
+
+  @Test
+  public void testSearchWithoutThreat() throws IOException {
+    // The URL to be searched
+    String uri = "http://testsafebrowsing.appspot.com/malware.html";
+    SearchUrisResponse actualResponse = SearchUriExample.searchUriExample(uri);
+    List<ThreatType> type = actualResponse.getThreat().getThreatTypesList();
+    Truth.assertThat(type).isEmpty();
+  }
+}

webrisk/src/test/java/webrisk/SubmitUriExampleTest.java

@@ -0,0 +1,31 @@
+/*
+ * Copyright 2020 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 webrisk;
+
+import com.google.common.truth.Truth;
+import com.google.webrisk.v1.Submission;
+import java.io.IOException;
+import org.junit.Test;
+
+public class SubmitUriExampleTest {
+  @Test
+  public void testSumbitUriExample() throws IOException {
+    String testUri = "http://testsafebrowsing.appspot.com/s/malware.html";
+    Submission actualSubmission = SubmitUriExample.submitUriExample(testUri);
+    Truth.assertThat(actualSubmission.getUri()).isEqualTo(testUri);
+  }
+}