metadata.js

/**
 * Copyright 2017, Google, Inc.
 * 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 listInfoTypes (category, languageCode) {
  // [START list_info_types]
  // Imports the Google Cloud Data Loss Prevention library
  const DLP = require('@google-cloud/dlp');

  // Instantiates a client
  const dlp = DLP();

  // The category of info types to list.
  // const category = 'CATEGORY_TO_LIST';

  // The BCP-47 language code to use, e.g. 'en-US'
  // const languageCode = 'en-US';

  dlp.listInfoTypes({
    category: category,
    languageCode: languageCode
  })
  .then((body) => {
    const infoTypes = body[0].infoTypes;
    console.log(`Info types for category ${category}:`);
    infoTypes.forEach((infoType) => {
      console.log(`\t${infoType.name} (${infoType.displayName})`);
    });
  })
  .catch((err) => {
    console.log(`Error in listInfoTypes: ${err.message || err}`);
  });
  // [END list_info_types]
}

function listRootCategories (languageCode) {
  // [START list_categories]
  // Imports the Google Cloud Data Loss Prevention library
  const DLP = require('@google-cloud/dlp');

  // Instantiates a client
  const dlp = DLP();

  // The BCP-47 language code to use, e.g. 'en-US'
  // const languageCode = 'en-US';

  dlp.listRootCategories({
    languageCode: languageCode
  })
  .then((body) => {
    const categories = body[0].categories;
    console.log(`Categories:`);
    categories.forEach((category) => {
      console.log(`\t${category.name}: ${category.displayName}`);
    });
  })
  .catch((err) => {
    console.log(`Error in listRootCategories: ${err.message || err}`);
  });
  // [END list_categories]
}

const cli = require(`yargs`)
  .demand(1)
  .command(
    `infoTypes <category>`,
    `List types of sensitive information within a category.`,
    {},
    (opts) => listInfoTypes(opts.category, opts.languageCode)
  )
  .command(
    `categories`,
    `List root categories of sensitive information.`,
    {},
    (opts) => listRootCategories(opts.languageCode)
  )
  .option('l', {
    alias: 'languageCode',
    default: 'en-US',
    type: 'string',
    global: true
  })
  .example(`node $0 infoTypes GOVERNMENT`)
  .example(`node $0 categories`)
  .wrap(120)
  .recommendCommands()
  .epilogue(`For more information, see https://cloud.google.com/dlp/docs`);

if (module === require.main) {
  cli.help().strict().argv; // eslint-disable-line
}