saving state

Author: raykao

.gitignore

@@ -3,4 +3,7 @@ node_modules
 
 aks.json
 test_output.json
-test.json
+test.json
+
+terraform/
+output/

index.js

@@ -1,12 +1,16 @@
 const AzureCompute = require('@azure/arm-compute');
 const AzureResources = require('@azure/arm-resources')
 const MsRest = require('@azure/ms-rest-nodeauth');
+
+const FS = require('fs');
+
 const ClientID = process.env.ARM_CLIENT_ID;
 const ClientSecret = process.env.ARM_CLIENT_SECRET;
 const SubscriptionID = process.env.ARM_SUBSCRIPTION_ID;
 const TenantID = process.env.ARM_TENANT_ID;
-const debug = process.env.DEBUG || false;
-const FS = require('fs');
+
+
+const TFAzureResource = require('./lib/tfobj');
 
 function getCredentials() { 
   return new Promise((resolve, reject) => {
@@ -21,49 +25,21 @@ function getCredentials() {
   })
 }
 
-function toTfObject(azObject){
-  const startTF = `resource "${azObject.armResourceType}" "${azObject.name}" {`
-  const endTF = `}`
-  let tfObject = startTF + "\n";
-
-  delete azObject.id;
-  delete azObject.armResourceType;
-  delete azObject.managedBy;
-  
-  for (var property in azObject) {   
-    const type = azObject[property].constructor;
-
-    if(azObject.hasOwnProperty(property)){
-      switch(type) {
-        case Object:
-          break;
-        default:
-          tfObject += objectify(property, azObject[property])
-      }
-    }
-  }
-
-  tfObject += endTF;
-
-  if(debug){
-    console.log(tfObject);
-  }
-
-  return tfObject;
-}
-
-function objectify(property, value) {
-  return "\t" + `${property} = "${value}"` + "\n";
-}
-
 getCredentials().then(credentials => {
   const client = new AzureResources.ResourceManagementClient(credentials, SubscriptionID);
 
   client.resourceGroups.list()
   .then((groups)=>{
-    groups.forEach((group, index) => {
-      group.armResourceType = 'azurerm_resource_group';
-      toTfObject(group);
+    groups.forEach(group => {
+      let thing = new TFAzureResource('azurerm_resource_group', group);
+    
+      thing.saveToFile()
+        .then(obj => {console.log(obj)})
+        .catch()
     })
+    // groups.forEach((group, index) => {
+    //   group.armResourceType = 'azurerm_resource_group';
+    //   toTfObject(group);
+    // })
   })
 })

lib/tfobj.js

@@ -1,12 +1,62 @@
+const fs = require('fs');
+
 class TFAzureResource {
-  constructor(json) {
-    this.jsonObject = json;
+  constructor(type, jsonObject) {
+    this.uuid = jsonObject.id;
+    this.name = jsonObject.name;
+    this.type = type;
+    this.terraformId = `${this.type}.${this.name}`;
+    this.originalJSON = jsonObject;
   }
 
+  toTF(){
+    const azObject = this.originalJSON;
+    const startTF = `resource "${this.type}" "${azObject.name}" {`
+    const endTF = `}`
+    let tfObject = startTF + "\n";
+  
+    delete azObject.id;
+    delete azObject.properties;
+    delete azObject.managedBy;
+    
+    for (var property in azObject) {   
+      const type = azObject[property].constructor;
+  
+      if(azObject.hasOwnProperty(property)){
+        switch(type) {
+          case Object:
+            break;
+          default:
+            tfObject += this.objectify(property, azObject[property])
+        }
+      }
+    }
+  
+    tfObject += endTF;
+    
+    return tfObject;
+  }  
+
   objectify(property, value) {
     return "\t" + `${property} = "${value}"` + "\n";
   }
+
+  saveToFile(){
+    return new Promise((resolve, reject) => {
+      let opts = 'utf8';
+      const dir = `./output`
+      const data = this.toTF();
+
+      if(!fs.existsSync(dir)) fs.mkdirSync(dir);
+
+      
+      fs.writeFile(dir + `/${this.type}.${this.name}.tf`, data, opts, (err) => {
+        if(err) reject(err)
+        else resolve(data)
+      })
+    });
+  }
 }
 
 
-module.exports.TFAzureResource = TFAzureResource;
+module.exports = TFAzureResource;