nifi_sdk_ruby 0.0.1 → 0.0.2

data/examples/basic.rb

@@ -6,16 +6,41 @@ nifi_client = Nifi.new()
 nifi_client.set_debug(true)
 
 # Get the ID of Root Process Group
-pg_root_id = nifi_client.get_process_group(:attr => 'id')
+pg_root = nifi_client.get_process_group()
 puts "\n"
 puts "PG Root's ID"
 puts "\n"
-puts pg_root_id
+puts pg_root['id']
 puts "\n"
 
-# Get all attrs of the Process Group with ID 9c3ebb60-015b-1000-1027-b27d47832152 (PG Root's child)
-some_pg = nifi_client.get_process_group(:pg_id => '9c3ebb60-015b-1000-1027-b27d47832152')
-puts "PG 9c3ebb60-015b-1000-1027-b27d47832152`s attrs"
+# Create new Process Group
+puts "Create new PG"
 puts "\n"
-puts some_pg
+new_pg = nifi_client.create_process_group(:name => 'test')
 puts "\n"
+puts new_pg
+puts "\n"
+
+# Get all attrs of the Process Group by ID (PG Root's child)
+puts 'PG ' + new_pg['id'] + ' attrs.'
+puts "\n"
+puts nifi_client.get_process_group(:id => new_pg['id'])
+puts "\n"
+
+# Get a Process Group ID
+puts "PG ID"
+puts "\n"
+pg = nifi_client.get_process_group(:id => new_pg['id'])
+puts pg['id']
+puts "\n"
+
+# Delete some Process Group
+puts 'Delete PG with id ' + new_pg['id']
+puts "\n"
+puts nifi_client.delete_process_group(new_pg['id'])
+puts "\n"
+
+# Upload a template to Root Process Group 
+puts 'Upload templete to Root PG'
+puts nifi_client.upload_template(:path => 'IN.hmStaff.taskStatus.xml')
+

data/lib/nifi_sdk_ruby/version.rb

@@ -1,3 +1,3 @@
 module NifiSdkRuby
-  VERSION = "0.0.1"
+  VERSION = "0.0.2"
 end

data/lib/nifi_sdk_ruby.rb

@@ -5,6 +5,8 @@ require 'pp' if ENV['DEBUG']
 require 'httparty'
 require 'curb'
 require 'json'
+require 'securerandom'
+require 'active_support/all'
 
 class Nifi
 
@@ -23,6 +25,7 @@ class Nifi
   @@async
   @@sdk_name
   @@sdk_version
+  @@client_id
 
   def initialize(*args)
 
@@ -36,6 +39,7 @@ class Nifi
     @@async       = DEFAULT_ACYNC
     @@sdk_name    = 'ruby'
     @@sdk_version = NifiSdkRuby::VERSION
+    @@client_id   = SecureRandom.uuid
   end
 
   def set_debug(debug = nil)
@@ -90,16 +94,57 @@ class Nifi
 
     args = args.reduce Hash.new, :merge
 
-    process_group = args[:pg_id] ? args[:pg_id] : 'root'
+    process_group = args[:id] ? args[:id] : 'root'
 
     base_url = @@base_url + "/process-groups/#{process_group}"
-    res = self.class.http_client(base_url)
+    self.class.http_client(base_url)
+  end
+
+  def create_process_group(*args)
+    args = args.reduce Hash.new, :merge
+
+    if args[:name].nil?
+      abort 'name params is mandatory.'
+    end
 
-    if args[:attr]
-      return res[args[:attr]]
+    params = '{"revision":{"clientId":"' + @@client_id + '","version":0},"component":{"name":"' + args[:name] + '","position":{"x":274.54776144527517,"y":-28.886681059739686}}}'
+
+    process_group = args[:id] ? args[:id] : 'root'
+    base_url = @@base_url + "/process-groups/#{process_group}/process-groups"
+    self.class.http_client(base_url, 'POSTRAW', params)
+  end
+
+  def delete_process_group(id = nil)
+
+    if id.nil?
+      abort 'id is mandatory.'
     end
 
-    return res
+    base_url = @@base_url + '/process-groups/' + id + '?clientId=' + @@client_id + '&version=1'
+    self.class.http_client(base_url, 'DELETE')
+  end
+
+  def upload_template(*args)
+
+    args = args.reduce Hash.new, :merge
+    
+    if args[:path].nil?
+      abort 'path params is mandatory.'
+    end
+    path = args[:path]
+
+    if not File.file? path or not File.readable? path
+      abort "Access to #{path} failed"
+    end
+    params = Array.new
+    params << Curl::PostField.file('template', path)
+
+    process_group = args[:id] ? args[:id] : 'root'
+
+    base_url = @@base_url + "/process-groups/#{process_group}/templates/upload"
+    res = self.class.http_client(base_url, 'POST', params)
+
+    return res['templateEntity']['template']
   end
 
   private
@@ -120,6 +165,7 @@ class Nifi
       when 'GET'
         c.get
       when 'POSTRAW'
+        c.headers['Content-Type'] = 'application/json'
         c.post(params)
       when 'POST'
         c.multipart_form_post = true
@@ -133,7 +179,13 @@ class Nifi
     end
 
     if c.response_code.to_s.match(/20./) and not c.body_str.empty?
-      JSON.parse(c.body_str)
+      begin
+        JSON.parse(c.body_str)
+      rescue
+        if c.content_type == 'application/xml'
+          JSON.parse(Hash.from_xml(c.body_str).to_json)
+        end
+      end
     else
       puts c.body_str
     end

data/nifi_sdk_ruby.gemspec

@@ -23,6 +23,7 @@ Gem::Specification.new do |gem|
   gem.add_dependency 'httparty', '>= 0.14.0'
   gem.add_dependency 'curb', '>= 0.9.3'
   gem.add_dependency 'json', '>= 1.8.3'
+  gem.add_dependency 'activesupport', '>= 5.0.2'
 
   gem.add_development_dependency "bundler", "~> 1.14"
   gem.add_development_dependency "rake", "~> 10.0"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nifi_sdk_ruby
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Israel Calvete
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-04-24 00:00:00.000000000 Z
+date: 2017-04-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -52,6 +52,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 1.8.3
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.0.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 5.0.2
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement
@@ -101,6 +115,7 @@ executables: []
 extensions: []
 extra_rdoc_files: []
 files:
+- ".gitignore"
 - CODE_OF_CONDUCT.md
 - Gemfile
 - LICENSE.txt
@@ -108,10 +123,12 @@ files:
 - Rakefile
 - bin/console
 - bin/setup
+- examples/IN.hmStaff.taskStatus.xml
 - examples/basic.rb
 - lib/nifi_sdk_ruby.rb
 - lib/nifi_sdk_ruby/version.rb
 - nifi_sdk_ruby.gemspec
+- pkg/.gitkeep
 - release.md
 homepage: http://rubygems.org/gems/nifi-sdk-ruby
 licenses: