@hiiretail/gcp-infra-generators 1.0.0 → 1.0.1

package/dist/generators/common-resources/pubsub/append.js

@@ -0,0 +1,130 @@
+const yaml = require('js-yaml');
+const fs = require('fs');
+
+const addDLQ = async (yamlArray, env, dlqTopic) => {
+  if (env === 'prod') {
+    yamlArray[0].dead_letter_topic = dlqTopic;
+  }
+};
+
+const appendNotIncludePull = async (inputs, subscriptionFilePath, dlqTopic) => {
+  const pullArray = [];
+
+  pullArray.push({
+    name: `${inputs.existingTopic}+${inputs.subscriberName}`,
+    ack_deadline_seconds: '60',
+    expiration_policy: '',
+  });
+
+  await addDLQ(pullArray, inputs.env, dlqTopic);
+  const yamlPullArray = yaml.dump(pullArray);
+  fs.appendFileSync(
+    subscriptionFilePath,
+    `pull_subscriptions:\n${yamlPullArray}`,
+  );
+};
+
+const appendIncludePull = async (
+  fileContent,
+  originalContentYaml,
+  subscriptionFilePath,
+  input,
+  dlqTopic,
+) => {
+  if (fileContent.includes('pull_subscriptions')) {
+    const pullArray = Object.values(originalContentYaml.pull_subscriptions);
+    const yamlPullArray = yaml.dump(pullArray);
+    fs.writeFileSync(
+      subscriptionFilePath,
+      `pull_subscriptions:\n${yamlPullArray}`,
+    );
+  }
+
+  const newPullArray = [];
+
+  newPullArray.push({
+    name: `${input.existingTopic}+${input.subscriberName}`,
+    ack_deadline_seconds: '60',
+    expiration_policy: '',
+  });
+
+  await addDLQ(newPullArray, input.env, dlqTopic);
+  const finalYamlPullArray = yaml.dump(newPullArray);
+  fs.appendFileSync(subscriptionFilePath, finalYamlPullArray);
+
+  if (fileContent.includes('push_subscriptions')) {
+    const pushArray = Object.values(originalContentYaml.push_subscriptions);
+    const yamlPushArray = yaml.dump(pushArray);
+    fs.appendFileSync(
+      subscriptionFilePath,
+      `push_subscriptions:\n${yamlPushArray}`,
+    );
+  }
+};
+
+const appendNotIncludePush = async (inputs, subscriptionFilePath, dlqTopic) => {
+  const pushArray = [];
+
+  pushArray.push({
+    name: `${inputs.existingTopic}+${inputs.subscriberName}`,
+    push_endpoint: inputs.pushEndpoint,
+    oidc_service_account_email: inputs.oidcEmail,
+    audience: inputs.audience,
+    expiration_policy: '',
+    filter: `attributes.Consumer = "${inputs.subscriberName}" OR NOT attributes:Consumer`,
+  });
+
+  await addDLQ(pushArray, inputs.env, dlqTopic);
+  const yamlPushArray = yaml.dump(pushArray);
+  fs.appendFileSync(
+    subscriptionFilePath,
+    `push_subscriptions:\n${yamlPushArray}`,
+  );
+};
+
+const appendIncludePush = async (
+  fileContent,
+  originalContentYaml,
+  subscriptionFilePath,
+  input,
+  dlqTopic,
+) => {
+  if (fileContent.includes('push_subscriptions')) {
+    const pushArray = Object.values(originalContentYaml.push_subscriptions);
+    const yamlPushArray = yaml.dump(pushArray);
+    fs.writeFileSync(
+      subscriptionFilePath,
+      `push_subscriptions:\n${yamlPushArray}`,
+    );
+  }
+
+  const newPushArray = [];
+
+  newPushArray.push({
+    name: `${input.existingTopic}+${input.subscriberName}`,
+    push_endpoint: input.pushEndpoint,
+    oidc_service_account_email: input.oidcEmail,
+    audience: input.audience,
+    expiration_policy: '',
+    filter: `attributes.Consumer = "${input.subscriberName}" OR NOT attributes:Consumer`,
+  });
+  await addDLQ(newPushArray, input.env, dlqTopic);
+  const yamlPushArray = yaml.dump(newPushArray);
+  fs.appendFileSync(subscriptionFilePath, yamlPushArray);
+
+  if (fileContent.includes('pull_subscriptions')) {
+    const pullArray = Object.values(originalContentYaml.pull_subscriptions);
+    const yamlPullArray = yaml.dump(pullArray);
+    fs.appendFileSync(
+      subscriptionFilePath,
+      `pull_subscriptions:\n${yamlPullArray}`,
+    );
+  }
+};
+
+module.exports = {
+  appendNotIncludePull,
+  appendIncludePull,
+  appendNotIncludePush,
+  appendIncludePush,
+};

package/dist/generators/common-resources/pubsub/get-gcp-projects.js

@@ -0,0 +1,34 @@
+const path = require('path');
+const fs = require('fs');
+
+const getKeyValue = (content, key) => {
+  let value = '';
+  content.split(/\r?\n/).forEach((line) => {
+    if (line.startsWith(key)) {
+      value = line.split('=')[1].replace('"', '').replace('"', '').trim();
+    }
+  });
+  return value;
+};
+
+const getProjectId = (env) => {
+  let projectId = '';
+  const projectHCL = path.join('infra', env, 'project.hcl');
+  if (fs.existsSync(projectHCL)) {
+    const projectHCLContent = fs.readFileSync(projectHCL, 'utf8');
+    projectId = getKeyValue(projectHCLContent, '  project_id');
+  }
+
+  return projectId;
+};
+
+const getProdPushEndopint = (pushEndpoint) => {
+  let prodPushEndopint = '';
+  prodPushEndopint = pushEndpoint.replace('.dev', '.com');
+  return prodPushEndopint;
+};
+
+module.exports = {
+  getProjectId,
+  getProdPushEndopint,
+};

package/dist/generators/common-resources/pubsub/handle-subscribers.js

@@ -0,0 +1,68 @@
+const fs = require('fs');
+const yaml = require('js-yaml');
+const {
+  appendNotIncludePull,
+  appendIncludePull,
+  appendNotIncludePush,
+  appendIncludePush,
+} = require('./append');
+
+const handleSubscribers = async (
+  env,
+  answers,
+  oidcEmail,
+  pushEndpoint,
+  subscriptionFilePath,
+  dlqTopic,
+) => {
+  const { subscriberName, existingTopic, pushOrPull, audience } = answers;
+
+  const subscriptionFileContent = fs.readFileSync(subscriptionFilePath, 'utf8');
+
+  const inputs = {
+    ...this.answers,
+    env,
+    existingTopic,
+    subscriberName,
+    audience,
+    oidcEmail,
+    pushEndpoint,
+    dlqTopic,
+  };
+
+  if (pushOrPull === 'pull') {
+    if (
+      subscriptionFileContent.length === 0 ||
+      !subscriptionFileContent.includes('pull_subscriptions')
+    ) {
+      await appendNotIncludePull(inputs, subscriptionFilePath, dlqTopic);
+    } else {
+      const originalContentYaml = yaml.load(subscriptionFileContent);
+      const fileContent = subscriptionFileContent;
+      await appendIncludePull(
+        fileContent,
+        originalContentYaml,
+        subscriptionFilePath,
+        inputs,
+        dlqTopic,
+      );
+    }
+  } else if (
+    subscriptionFileContent.length === 0 ||
+    !subscriptionFileContent.includes('push_subscriptions')
+  ) {
+    await appendNotIncludePush(inputs, subscriptionFilePath, dlqTopic);
+  } else {
+    const originalContentYaml = yaml.load(subscriptionFileContent);
+    const fileContent = subscriptionFileContent;
+    await appendIncludePush(
+      fileContent,
+      originalContentYaml,
+      subscriptionFilePath,
+      inputs,
+      dlqTopic,
+    );
+  }
+};
+
+module.exports = handleSubscribers;