Product search rebased

samples/productSearch/importProductSets.v1p3beta1.js

@@ -0,0 +1,92 @@
+/**
+ * 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.
+ */
+
+'use strict';
+// [START vision_product_search_import_product_images]
+function importProductSets(projectId, location, gcsUri) {
+  // Imports the Google Cloud client library
+  const vision = require('@google-cloud/vision').v1p3beta1;
+  // Creates a client
+  const client = new vision.ProductSearchClient();
+
+  /**
+   * TODO(developer): Uncomment the following line before running the sample.
+   */
+  // const projectId = 'Your Google Cloud project Id';
+  // const location = 'A compute region name';
+  // const gcsUri = 'Google Cloud Storage path of the input image'';
+
+  // A resource that represents Google Cloud Platform location.
+  const projectLocation = client.locationPath(projectId, location);
+
+  // Set the input configuration along with Google Cloud Storage URI
+  const inputConfig = {
+    gcsSource: {
+      csvFileUri: gcsUri,
+    },
+  };
+
+  // Import the product sets from the input URI.
+  client
+    .importProductSets({parent: projectLocation, inputConfig: inputConfig})
+    .then(responses => {
+      const response = responses[0];
+      const operation = responses[1];
+      console.log('Processing operation name: ', operation.name);
+
+      // synchronous check of operation status
+      return response.promise();
+    })
+    .then(responses => {
+      console.log('Processing done.');
+      console.log('Results of the processing:');
+
+      for (let i in responses[0].statuses) {
+        console.log(
+          'Status of processing ',
+          i,
+          'of the csv:',
+          responses[0].statuses[i]
+        );
+
+        // Check the status of reference image
+        if (responses[0].statuses[i].code === 0) {
+          console.log(responses[0].referenceImages[i]);
+        } else {
+          console.log('No reference image.');
+        }
+      }
+    })
+    .catch(err => {
+      console.error(err);
+    });
+  // [END vision_product_search_import_product_images]
+}
+// [END vision_product_search_import_product_set]
+
+require(`yargs`) // eslint-disable-line
+  .demand(1)
+  .command(
+    `importProductSets <projectId> <location> <gcsUri>`,
+    `Import a Product Set`,
+    {},
+    opts => importProductSets(opts.projectId, opts.location, opts.gcsUri)
+  )
+  .example(`node $0 COMMAND ARG`)
+  .wrap(120)
+  .recommendCommands()
+  .epilogue(`For more information, see https://cloud.google.com/vision/docs`)
+  .help()
+  .strict().argv;

samples/productSearch/productSearch.v1p3beta1.js

@@ -0,0 +1,165 @@
+/**
+ * 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.
+ */
+
+'use strict';
+
+function addProductToProductSet(projectId, location, productId, productSetId) {
+  // [START vision_product_search_add_product_to_product_set]
+
+  const vision = require('@google-cloud/vision').v1p3beta1;
+
+  const client = new vision.ProductSearchClient();
+
+  /**
+   * TODO(developer): Uncomment the following line before running the sample.
+   */
+  // const projectId = 'Your Google Cloud project Id';
+  // const location = 'A compute region name';
+  // const productId = 'Id of the product';
+  // const productSetId = 'Id of the product set';
+
+  const productPath = client.productPath(projectId, location, productId);
+  const productSetPath = client.productSetPath(
+    projectId,
+    location,
+    productSetId
+  );
+
+  const request = {
+    name: productSetPath,
+    product: productPath,
+  };
+
+  client
+    .addProductToProductSet(request)
+    .then(() => {
+      console.log(`Product added to product set.`);
+    })
+    .catch(err => {
+      console.error(err);
+    });
+  // [END vision_product_search_add_product_to_product_set]
+}
+
+function listProductsInProductSet(projectId, location, productSetId) {
+  // [START vision_product_search_list_products_in_product_set]
+
+  const vision = require('@google-cloud/vision').v1p3beta1;
+
+  const client = new vision.ProductSearchClient();
+
+  /**
+   * TODO(developer): Uncomment the following line before running the sample.
+   */
+  // const projectId = 'Your Google Cloud project Id';
+  // const location = 'A compute region name';
+  // const productSetId = 'Id of the product set';
+  const productSetPath = client.productSetPath(
+    projectId,
+    location,
+    productSetId
+  );
+  const request = {
+    name: productSetPath,
+  };
+
+  client.listProductsInProductSet(request).then(results => {
+    const products = results[0];
+    products.forEach(product => {
+      console.log(`Product name: ${product.name}`);
+      console.log(`Product display name: ${product.displayName}`);
+    });
+  });
+  // [END vision_product_search_list_products_in_product_set]
+}
+
+function removeProductFromProductSet(
+  projectId,
+  location,
+  productId,
+  productSetId
+) {
+  // [START vision_product_search_remove_product_from_product_set]
+
+  const vision = require('@google-cloud/vision').v1p3beta1;
+
+  const client = new vision.ProductSearchClient();
+
+  /**
+   * TODO(developer): Uncomment the following line before running the sample.
+   */
+  const productSetPath = client.productSetPath(
+    projectId,
+    location,
+    productSetId
+  );
+
+  const productPath = client.productPath(projectId, location, productId);
+
+  const request = {
+    name: productSetPath,
+    product: productPath,
+  };
+
+  client
+    .removeProductFromProductSet(request)
+    .then(() => {
+      console.log(`Product removed from product set.`);
+    })
+    .catch(err => {
+      console.error(err);
+    });
+  // [END vision_product_search_remove_product_from_product_set]
+}
+
+require(`yargs`) // eslint-disable-line
+  .demand(1)
+  .command(
+    `addProductToProductSet <projectId> <location> <productId> <productSetId>`,
+    `Add a product to product set`,
+    {},
+    opts =>
+      addProductToProductSet(
+        opts.projectId,
+        opts.location,
+        opts.productId,
+        opts.productSetId
+      )
+  )
+  .command(
+    `listProductsInProductSet <projectId> <location> <productSetId>`,
+    `List all products in a product set`,
+    {},
+    opts =>
+      listProductsInProductSet(opts.projectId, opts.location, opts.productSetId)
+  )
+  .command(
+    `removeProductFromProductSet <projectId> <location> <productId> <productSetId>`,
+    `Remove a products from a product set`,
+    {},
+    opts =>
+      removeProductFromProductSet(
+        opts.projectId,
+        opts.location,
+        opts.productId,
+        opts.productSetId
+      )
+  )
+  .example(`node $0 COMMAND ARG`)
+  .wrap(120)
+  .recommendCommands()
+  .epilogue(`For more information, see https://cloud.google.com/vision/docs`)
+  .help()
+  .strict().argv;