nifi_sdk_ruby 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 5bb97fd650f0de430ee1b893f5abff6bf28c59f1
-  data.tar.gz: 049c9dc2772ee4c9acd83b1ae8f69e686f522a0a
+  metadata.gz: be48d97b56f690c502c1b8723c4867461ec0798c
+  data.tar.gz: 6a77e0569fb024980fb3fca6c00eb53f04060266
 SHA512:
-  metadata.gz: 284592edc6779800145e9150086f513210c7e67bb96ab5c985afe075321d9b40e650f57120cc1ae15a0a646b51f10291deed51fc0588f62b3774b15993922609
-  data.tar.gz: 28e7e0e03ded209b4458a65e69ab0df00cb686f384e0e2f8387a1f1b347281b8f3bbb1e1722416a7536cbd1eb64114e2a9f7b45b475740aa5de4df97fd4975c6
+  metadata.gz: a9edf3dbce0917300ed3fbb1e44e7274656d1623ecca087d7f55a917c49676c5f7757f1e24ee05b10a37a0ab3b76c41f989ee71d6299219e1d76114107bd8793
+  data.tar.gz: 1fd9b45e1e7dea96ece0f062cacd638bf46cab9a319ec46c14e214cd3a9eec92c67881491f4533966e02c65ffd1d891acf20e7a357b3477eba65879c1a2e7f2e

data/README.md

@@ -30,44 +30,83 @@ Or install it yourself as:
 require 'nifi_sdk_ruby'
 
 nifi_client = Nifi.new()
-nifi_client.set_debug(true)
+nifi_client.set_debug true
 
 ```
 
-### Get root id
+### Get Root's Process Group id
 
 ```ruby
 # Get Root Process Group
-pg_root = nifi_client.get_process_group()
+pg_root = nifi_client.get_process_group
 puts "PG Root's ID"
 puts pg_root['id']
 ```
 
-### Create new PG
+### Create new Process Group
 
 ```ruby
 new_pg = nifi_client.create_process_group(:name => 'test')
 puts new_pg
 ```
 
-### Get all attrs of the Process Group with ID 9c3ebb60-015b-1000-1027-b27d47832152 (PG Root's child)
+### Check if Process Group exists
+
+```ruby
+puts nifi_client.process_group_by_name? test
+```
+
+### Get all attrs a the Process Group by ID (PG Root's child)
 
 ```ruby
 puts  nifi_client.get_process_group(:id => new_pg['id'])
 ```
 
-### Delete some PG
+### Get all attrs a the Process Group by Name (Root's childs)
+
+```ruby
+puts nifi_client.get_process_group_by_name 'test'
+```
+
+### Delete some Process Group
 
 ```ruby
 puts nifi_client.delete_process_group(new_pg['id'])
 ```
 
-# Upload a template to Root PG
+### Upload a template to Root Process Group
+
+From file ...
 
 ```ruby
 puts nifi_client.upload_template(:path => 'IN.hmStaff.taskStatus.xml')
 ```
 
+From url ...
+
+```ruby
+puts nifi_client.upload_template(:path => 'https://your.domain.net/IN.hmStaff.taskStatus.xml')
+```
+
+### Check if Process Group exists
+
+```ruby
+puts nifi_client.template_by_name? test
+```
+
+
+### Get all attrs of a template by Name (Root's childs)
+
+```ruby
+t = nifi_client.get_template_group_by_name 'test'
+puts t
+```
+
+### Delete template
+```ruby
+puts nifi_client.delete_template t['id']
+```
+
 ## Contributing
 
 Bug reports and pull requests are welcome on GitHub at https://github.com/icalvete/nifi_sdk_ruby. This project is intended to be a safe, welcoming space for collaboration, and contributors are expected to adhere to the [Contributor Covenant](http://contributor-covenant.org) code of conduct.

data/examples/basic.rb

@@ -3,10 +3,10 @@ require 'pp'
 require '../lib/nifi_sdk_ruby'
 
 nifi_client = Nifi.new()
-nifi_client.set_debug(true)
+nifi_client.set_debug true
 
 # Get the ID of Root Process Group
-pg_root = nifi_client.get_process_group()
+pg_root = nifi_client.get_process_group
 puts "\n"
 puts "PG Root's ID"
 puts "\n"
@@ -37,10 +37,9 @@ 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 nifi_client.delete_process_group new_pg['id']
 puts "\n"
 
 # Upload a template to Root Process Group 
-puts 'Upload templete to Root PG'
+puts 'Upload template to Root PG'
 puts nifi_client.upload_template(:path => 'IN.hmStaff.taskStatus.xml')
-

data/lib/nifi_sdk_ruby.rb

@@ -7,6 +7,8 @@ require 'curb'
 require 'json'
 require 'securerandom'
 require 'active_support/all'
+require 'open-uri'
+require 'nokogiri'
 
 class Nifi
 
@@ -107,6 +109,10 @@ class Nifi
       abort 'name params is mandatory.'
     end
 
+    if self.process_group_by_name? args[:name]
+      abort 'The process group ' + args[:name] + ' already exists'
+    end
+
     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'
@@ -124,18 +130,68 @@ class Nifi
     self.class.http_client(base_url, 'DELETE')
   end
 
+  def get_process_group_by_name(name = nil)
+
+    if name.nil?
+      abort 'name is mandatory.'
+    end
+
+    res = self.class.exists
+
+    pg = res.select do |r|
+      r['name'] == name and r['identifier'] =~ /process-groups/
+    end
+
+    if pg.count == 1
+      self.class.get pg[0]['identifier']
+    else
+      abort 'Unable to locate group with name ' + name
+    end
+  end
+
+  def process_group_by_name?(name = nil)
+
+    if name.nil?
+      abort 'name is mandatory.'
+    end
+
+    res = self.class.exists
+
+    pg = res.select do |r|
+      r['name'] == name and r['identifier'] =~ /process-groups/
+    end
+
+    pg.count == 1 ? true : false
+  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 path =~ URI::regexp
+
+      download_s = open(path)
+      download_t = "/tmp/#{download_s.base_uri.to_s.split('/')[-1]}"
+      IO.copy_stream(download_s, download_t)
+      path = download_t
+    end
+
     if not File.file? path or not File.readable? path
       abort "Access to #{path} failed"
     end
+
+    t = File.open(path) { |f| Nokogiri::XML(f) }
+    name = t.xpath('//template/name').text
+
+    if self.template_by_name? name
+      abort 'The template ' + name + ' already exists'
+    end
+
     params = Array.new
     params << Curl::PostField.file('template', path)
 
@@ -147,8 +203,66 @@ class Nifi
     return res['templateEntity']['template']
   end
 
+  def delete_template(id = nil)
+
+    if id.nil?
+      abort 'id is mandatory.'
+    end
+
+    base_url = @@base_url + '/templates/' + id
+    self.class.http_client(base_url, 'DELETE')
+  end
+
+  def get_template_by_name(name = nil)
+
+    if name.nil?
+      abort 'name is mandatory.'
+    end
+
+    res = self.class.exists
+
+    t = res.select do |r|
+      r['name'] == name and r['identifier'] =~ /templates/
+    end
+
+
+    if t.count == 1
+      t[0]['identifier'].scan(/\/templates\/(.*)/)
+    else
+      abort 'Unable to locate template with name ' + name
+    end
+
+  end
+
+  def template_by_name?(name = nil)
+
+    if name.nil?
+      abort 'name is mandatory.'
+    end
+
+    res = self.class.exists
+
+    pg = res.select do |r|
+      r['name'] == name and r['identifier'] =~ /templates/
+    end
+
+    pg.count == 1 ? true : false
+  end
+
   private
 
+  def self.exists
+    base_url = @@base_url + "/resources"
+    res = self.http_client(base_url)
+
+    return res['resources']
+  end
+
+  def self.get(resource)
+    base_url = @@base_url + resource
+    self.http_client(base_url)
+  end
+
   def self.http_client(url, method = 'GET', params = nil, filename = nil)
     c = Curl::Easy.new
     #c.http_auth_types              = :basic

data/lib/nifi_sdk_ruby/version.rb

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

data/nifi_sdk_ruby.gemspec

@@ -24,6 +24,7 @@ Gem::Specification.new do |gem|
   gem.add_dependency 'curb', '>= 0.9.3'
   gem.add_dependency 'json', '>= 1.8.3'
   gem.add_dependency 'activesupport', '>= 5.0.2'
+  gem.add_dependency 'nokogiri', '>= 1.6.7.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.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Israel Calvete
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2017-04-25 00:00:00.000000000 Z
+date: 2017-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -66,6 +66,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 5.0.2
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.6.7.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.6.7.2
 - !ruby/object:Gem::Dependency
   name: bundler
   requirement: !ruby/object:Gem::Requirement