add config.set(flatKey) method

sematext · May 16, 2017 · f769b4b · f769b4b
1 parent 6f9cc82
commit f769b4b
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/lib/util/spmconfig.js b/lib/util/spmconfig.js
@@ -10,6 +10,7 @@
 
 'use strict'
 var flatten = require('flat')
+var unflatten = require('flat').unflatten
 var util = require('util')
 var RC = require('rc-yaml-2')
 var spmDefaultConfig = {
@@ -44,8 +45,20 @@ var SpmConfig = function (appType) {
   return this
 }
 
-SpmConfig.prototype.get = function (key) {
-  return this.rcFlat[key]
+SpmConfig.prototype.get = function (flatKey) {
+  return this.rcFlat[flatKey]
+}
+
+SpmConfig.prototype.set = function (flatKey, value) {
+  var kv = {}
+  kv[flatKey] = value
+  kv.object = true
+  var result = unflatten(kv)
+  delete result.object
+  delete this.rcFlat
+  util._extend(this, result)
+  this.rcFlat = flatten(this)
+  return this
 }
 
 module.exports = new SpmConfig(process.env.SPM_AGENT_APP_TYPE || 'spmagent')
diff --git a/test/test.js b/test/test.js
@@ -73,4 +73,16 @@ describe('SPM for NodeJS tests', function () {
       done(err)
     }
   })
+  it('SPM Config "set" accepts flat key', function (done) {
+    try {
+      config.set('a.b.0.name', 'test')
+      if (config.a.b[0].name === 'test') {
+        done()
+      } else {
+        done(new Error('set flat key: a.b.0.name != test'))
+      }
+    } catch (err) {
+      done(err)
+    }
+  })
 })