power-bi 2.4.0 → 2.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 90d415a89ed5860c82824f7e152e4c58d42054bc293f5534f6669774682a1ffe
-  data.tar.gz: 536d4c7cea48c859c7b780db751da15375b060a45c59946a2335887d8e948494
+  metadata.gz: 98d26348ce002789d1f065e91735ae2d441a64bb32c8c935993961036f2d2379
+  data.tar.gz: e3527606c13a8f62a4918b98e0dfc5ed441f691b836b4f670251e80442fcf104
 SHA512:
-  metadata.gz: d31db71feb4b03d7810d5805b3d48016e89ead1782bfae66269aa2a0c55340e6cf21031525ffa6d5b917cdc36041670ef52c1ccff499ebdfc7d0051d8005b6a8
-  data.tar.gz: 1ea49ae1a15e45b16c21b9da86c8ac20abb9e96a6eb63e6a0ed9a7dfcbca25dfff8431896703f2d55736a148668cf4b85ac591eb20c68bf1b2d9f6baee455e3a
+  metadata.gz: 5e1674d67c5cf55473e0325601b65b532dc374cd9e5a08a2345a282c4b5dd0a984392d1c0196857a649a00ceaf45362d21980e85c9050868b5538b4377f20dd6
+  data.tar.gz: 1c7517a30c1d32ddcc7e6d6883ef6a5b91fcac12500c76b89bdbebb473c405dde6b58f45a2b60dc3c60b7dfac440973ea81daa6a49f308c9d22b23670484293a

data/lib/power-bi/admin.rb

@@ -0,0 +1,80 @@
+module PowerBI
+  class Admin
+
+    def initialize(tenant)
+      @tenant = tenant
+    end
+
+    def get_user_artifact_access(user_id, artifact_types: nil)
+      if artifact_types
+        params = {artifactTypes: artifact_types.join(',')}
+      else
+        params = {}
+      end
+
+      url = "/admin/users/#{user_id}/artifactAccess"
+
+      resp = @tenant.get(url, params)
+      data = resp[:ArtifactAccessEntities]
+
+      continuation_token = resp[:continuationToken]
+
+      while continuation_token
+        params = {continuationToken: "'#{continuation_token}'"}
+        resp = @tenant.get(url, params)
+        data += resp[:ArtifactAccessEntities]
+        continuation_token = resp[:continuationToken] ? URI::decode_uri_component(resp[:continuationToken]) : nil
+      end
+
+      data
+    end
+
+    def get_workspaces(filter: nil, expand: nil)
+      params = {}
+      params[:$filter] = filter if filter
+      params[:$expand] = expand if expand
+
+      url = '/admin/groups'
+
+      nr_records = 5000
+      count = 0
+
+      data = []
+
+      loop do
+        params[:$top] = nr_records
+        params[:$skip] = count
+        resp = @tenant.get(url, params)
+        data += resp[:value]
+        batch_count = resp[:value].size
+        count += batch_count
+        break if batch_count < nr_records
+      end
+
+      data
+    end
+
+    def force_delete_workspace_by_workspace_name(user_email, workspace_name)
+      workspace = get_workspaces(filter: "name eq '#{workspace_name}' and state eq 'Active'").first
+      add_user(user_email, workspace[:id], access_right: "Admin")
+      @tenant.workspace(workspace[:id]).delete
+    end
+
+    def force_delete_workspace_by_workspace_id(user_email, workspace_id)
+      add_user(user_email, workspace_id, access_right: "Admin")
+      @tenant.workspace(workspace_id).delete
+    end
+
+    private
+
+    def add_user(user_email, workspace_id, access_right: "Admin")
+      @tenant.post("/admin/groups/#{workspace_id}/users") do |req|
+        req.body = {
+          emailAddress: user_email,
+          groupUserAccessRight: access_right,
+        }.to_json
+      end
+    end
+
+  end
+end

data/lib/power-bi/dataset.rb

@@ -1,6 +1,6 @@
 module PowerBI
   class Dataset < Object
-    attr_reader :workspace, :datasources, :parameters, :refresh_history
+    attr_reader :workspace, :datasources, :parameters
 
     def initialize(tenant, parent, id = nil)
       super(tenant, id)
@@ -38,6 +38,11 @@ module PowerBI
       true
     end
 
+    def refresh_history(entries_to_load = 1)
+      @refresh_history.entries_to_load = entries_to_load
+      @refresh_history
+    end
+
     def last_refresh
       @refresh_history.first
     end

data/lib/power-bi/refresh.rb

@@ -22,18 +22,27 @@ module PowerBI
   end
 
   class RefreshArray < Array
+    attr_reader :entries_to_load
 
     def initialize(tenant, dataset)
       super(tenant, dataset)
       @dataset = dataset
+      @entries_to_load = 1
     end
 
     def self.get_class
       Refresh
     end
 
+    def entries_to_load=(entries_to_load)
+      if entries_to_load > @entries_to_load
+        @entries_to_load = entries_to_load
+        reload
+      end
+    end
+
     def get_data
-      @tenant.get("/groups/#{@dataset.workspace.id}/datasets/#{@dataset.id}/refreshes", {'$top': '1'})[:value]
+      @tenant.get("/groups/#{@dataset.workspace.id}/datasets/#{@dataset.id}/refreshes", {'$top': @entries_to_load.to_s})[:value]
     end
   end
 end

data/lib/power-bi/tenant.rb

@@ -1,6 +1,6 @@
 module PowerBI
   class Tenant
-    attr_reader :workspaces, :gateways, :capacities, :profiles, :profile_id
+    attr_reader :workspaces, :gateways, :capacities, :profiles, :profile_id, :admin
 
     def initialize(token_generator, retries: 5, logger: nil)
       @token_generator = token_generator
@@ -10,6 +10,7 @@ module PowerBI
       @profiles = ProfileArray.new(self)
       @logger = logger
       @profile_id = nil
+      @admin = Admin.new(self)
 
       ## WHY RETRIES? ##
       # It is noticed that once in a while (~0.1% API calls), the Power BI server returns a 500 (internal server error) without apparent reason, just retrying works :-)

data/lib/power-bi.rb

@@ -26,4 +26,5 @@ require_relative "power-bi/gateway_datasource_user"
 require_relative "power-bi/page"
 require_relative "power-bi/user"
 require_relative "power-bi/capacity"
-require_relative "power-bi/profile"
+require_relative "power-bi/profile"
+require_relative "power-bi/admin"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: power-bi
 version: !ruby/object:Gem::Version
-  version: 2.4.0
+  version: 2.5.0
 platform: ruby
 authors:
 - Lode Cools
@@ -89,6 +89,7 @@ files:
 - LICENSE
 - README.md
 - lib/power-bi.rb
+- lib/power-bi/admin.rb
 - lib/power-bi/array.rb
 - lib/power-bi/capacity.rb
 - lib/power-bi/dataset.rb