riak-admin/src/index.coffee

Programmatically use the riak-admin tool for Basho Riak 2.0

###*
@author Isaac Johnston <isaac.johnston@joukou.com>
@copright (c) 2014 Joukou Ltd. All rights reserved.

This Source Code Form is subject to the terms of the Mozilla Public License,
v. 2.0. If a copy of the MPL was not distributed with this file, You can obtain
one at http://mozilla.org/MPL/2.0/.
###


_ = require( 'lodash' )
Q = require( 'q' )
{ spawn } = require( 'child_process' )

module.exports = ( { cmd } ) ->

  bucketType:

    list: ->
      deferred = Q.defer()

      child = spawn( cmd, [ 'bucket-type', 'list' ] )

      stdout = ''
      stderr = ''

      child.stdout.on( 'data', ( chunk ) ->
        stdout += chunk
      )

      child.stderr.on( 'data', ( chunk ) ->
        stderr += chunk
      )

      child.on( 'close', ( code ) ->
        unless code is 0
          deferred.reject( new Error( stderr ) )
          return

        matches = stdout.match( /^[\w_]+ \((active|not active)\)$/gm )
        bucketTypes = _.map( matches, ( match ) ->
          [ match, name, status ] =
            match.match( /^([\w_]+) \((active|not active)\)$/m )
          name: name
          active: status is 'active'
        )

        deferred.resolve( bucketTypes )
      )

      deferred.promise

    status: ( name ) ->
      deferred = Q.defer()

      child = spawn( cmd, [ 'bucket-type', 'status', name ] )

      stdout = ''
      stderr = ''

      child.stdout.on( 'data', ( chunk ) ->
        stdout += chunk
      )

      child.stderr.on( 'data', ( chunk ) ->
        stderr += chunk
      )
      
      child.on( 'close', ( code ) ->
        unless code is 0 or code is 1 # TODO exit code is incorrect on status
          deferred.reject( new Error( stderr ) )
          return

        props = stdout.match( /^[\w_]+: [\w\[\]\{\},_@'\.]+$/gm )
        props = _.reduce( props, ( memo, prop, i ) ->
          [ prop, name, value ] =
            prop.match( /^([\w_]+): ([\w\[\]\{\},_@'\.]+)$/ )
    
          if value is 'true'
            value = true
          else if value is 'false'
            value = false
          else if /^\d+$/.test( value )
            value = parseInt( value, 10 )
          memo[ name ] = value
          memo
        , {} )

        deferred.resolve( props )
      )

      deferred.promise

    activate: ( name ) ->
      deferred = Q.defer()

      child = spawn( cmd, [ 'bucket-type', 'activate', name ] )

      stdout = ''
      stderr = ''

      child.stdout.on( 'data', ( chunk ) ->
        stdout += chunk
      )

      child.stderr.on( 'data', ( chunk ) ->
        stderr += chunk
      )
      
      child.on( 'close', ( code ) ->
        unless code is 0
          deferred.reject( new Error( stderr ) )
          return

        deferred.resolve()
      )

      deferred.promise
           

    create: ( name, options ) ->
      deferred = Q.defer()

      child = spawn( cmd, [
        'bucket-type', 'create', name, "'#{JSON.stringify( options )}'"
      ] )
      
      stdout = ''
      stderr = ''

      child.stdout.on( 'data', ( chunk ) ->
        stdout += chunk
      )

      child.stderr.on( 'data', ( chunk ) ->
        stderr += chunk
      )

      child.on( 'close', ( code ) ->
        unless code is 0
          deferred.reject( new Error( stderr ) )
          return

        deferred.resolve()
      )

      deferred.promise

    update: ( name, options ) ->
      deferred = Q.defer()

      child = spawn( cmd, [
        'bucket-type', 'update', name, "'#{JSON.stringify( options )}'"
      ] )
      
      stdout = ''
      stderr = ''

      child.stdout.on( 'data', ( chunk ) ->
        stdout += chunk
      )

      child.stderr.on( 'data', ( chunk ) ->
        stderr += chunk
      )

      child.on( 'close', ( code ) ->
        unless code is 0
          deferred.reject( new Error( stderr ) )
          return

        deferred.resolve()
      )

      deferred.promise