gazer 0.2.9

data/lib/gzr/commands/space/create.rb

@@ -0,0 +1,50 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/space'
+
+module Gzr
+  module Commands
+    class Space
+      class Create < Gzr::Command
+        include Gzr::Space
+        def initialize(name,parent_space, options)
+          super()
+          @name = name
+          @parent_space = parent_space
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          space = nil
+          with_session do
+            space = create_space(@name, @parent_space)
+            output.puts "Created space #{space.id}" unless @options[:plain] 
+            output.puts space.id if @options[:plain] 
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/space/export.rb

@@ -0,0 +1,117 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/space'
+require_relative '../../modules/look'
+require_relative '../../modules/dashboard'
+require_relative '../../modules/filehelper'
+require 'pathname'
+require 'stringio'
+
+module Gzr
+  module Commands
+    class Space
+      class Export < Gzr::Command
+        include Gzr::Space
+        include Gzr::Look
+        include Gzr::Dashboard
+        include Gzr::FileHelper
+        def initialize(space_id, options)
+          super()
+          @space_id = space_id
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning("options: #{@options.inspect}") if @options[:debug]
+          with_session("3.1") do
+            if @options[:tar] || @options[:tgz] then
+              arc_path = Pathname.new(@options[:tgz] || @options[:tar])
+              arc_path = Pathname.new(File.expand_path(@options[:dir])) + arc_path unless arc_path.absolute?
+              f = File.open(arc_path.to_path, "wb")
+              tarfile = StringIO.new(String.new,"w")
+              begin
+                tw = Gem::Package::TarWriter.new(tarfile)
+                process_space(@space_id, tw)
+                tw.flush
+                tarfile.rewind
+                if @options[:tgz]
+                  gzw = Zlib::GzipWriter.new(f)
+                  gzw.write tarfile.string
+                  gzw.close
+                else
+                  f.write tarfile.string
+                end
+              ensure
+                f.close
+                tarfile.close
+              end
+            else
+              process_space(@space_id, @options[:dir])
+            end
+          end
+        end
+
+        def process_space(space_id, base, rel_path = nil)
+          space = query_space(space_id)
+          path = Pathname.new(space.name.gsub('/',"\u{2215}"))
+          path = rel_path + path if rel_path
+
+          write_file("Space_#{space.id}_#{space.name}.json", base, path) do |f|
+            f.write JSON.pretty_generate(space.to_attrs.reject do |k,v|
+              [:looks, :dashboards].include?(k)
+            end)
+          end
+          space.looks.each do |l|
+            look = query_look(l.id)
+            write_file("Look_#{look.id}_#{look.title}.json", base, path) do |f|
+              f.write JSON.pretty_generate(look.to_attrs) 
+            end
+          end
+          space.dashboards.each do |d|
+            data = query_dashboard(d.id)
+            data.to_attrs()[:dashboard_elements].each_index do |i|
+              element = data[:dashboard_elements][i]
+              if element[:merge_result_id]
+                merge_result = merge_query(element[:merge_result_id])
+                merge_result[:source_queries].each_index do |j|
+                  source_query = merge_result[:source_queries][j]
+                  merge_result[:source_queries][j][:query] = query(source_query[:query_id])
+                end
+                data[:dashboard_elements][i][:merge_result] = merge_result
+              end
+            end
+            write_file("Dashboard_#{data.id}_#{data.title}.json", base, path) do |f|
+              f.write JSON.pretty_generate(data.to_attrs)
+            end
+          end
+          space_children = query_space_children(space_id)
+          space_children.each do |child_space|
+            process_space(child_space.id, base, path)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/space/ls.rb

@@ -0,0 +1,97 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/space'
+require 'tty-table'
+
+module Gzr
+  module Commands
+    class Space
+      class Ls < Gzr::Command
+        include Gzr::Space
+        def initialize(filter_spec, options)
+          super()
+          @filter_spec = filter_spec
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning("options: #{@options.inspect}") if @options[:debug]
+          with_session do
+            space_ids = process_args([@filter_spec])
+            begin
+              puts "No spaces match #{@filter_spec}"
+              return nil
+            end unless space_ids && space_ids.length > 0
+
+            data = space_ids.map do |space_id|
+              query_space(space_id, @options[:fields])
+            end.compact
+
+            begin
+              puts "No data returned for spaces #{space_ids.inspect}"
+              return nil
+            end unless data && data.length > 0
+
+            @options[:fields] = 'dashboards(id,title)' if @filter_spec == 'lookml'
+            table_hash = Hash.new
+            fields = field_names(@options[:fields])
+            table_hash[:header] = field_names(@options[:fields]) unless @options[:plain]
+            rows = []
+            data.each do |r|
+              h = r.to_attrs
+              if @filter_spec != 'lookml' then
+                rows << [h[:parent_id],h[:id],h[:name], nil, nil, nil, nil]
+                subspaces = query_space_children(h[:id], "id,name,parent_id")
+                rows += subspaces.map do |r|
+                  h1 = r.to_attrs
+                  [h1[:parent_id], h1[:id], h1[:name], nil, nil, nil, nil]
+                end
+              end
+              h[:looks].each do |r|
+                rows << [h[:parent_id],h[:id],h[:name], r[:id], r[:title], nil, nil]
+              end if h[:looks]
+              h[:dashboards].each do |r|
+                rows << [h[:parent_id],h[:id],h[:name], nil, nil, r[:id], r[:title]] unless @filter_spec == 'lookml'
+                rows << [r[:id], r[:title]] if @filter_spec == 'lookml'
+              end if h[:dashboards]
+            end
+            table_hash[:rows] = rows
+            table = TTY::Table.new(table_hash) if data[0]
+            alignments = fields.collect do |k|
+              (k =~ /id\)*$/) ? :right : :left
+            end
+            begin
+              if @options[:csv] then
+                output.puts render_csv(table)
+              else
+                output.puts table.render(if @options[:plain] then :basic else :ascii end, alignments: alignments)
+              end
+            end if table
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/space/rm.rb

@@ -0,0 +1,56 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/space'
+
+module Gzr
+  module Commands
+    class Space
+      class Rm < Gzr::Command
+        include Gzr::Space
+        def initialize(space,options)
+          super()
+          @space = space
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          with_session do
+            space = query_space(@space)
+
+            begin
+              puts "Space #{@space} not found"
+              return nil
+            end unless space
+            children = query_space_children(@space)
+            unless (space.looks.length == 0 && space.dashboards.length == 0 && children.length == 0) || @options[:force] then
+              raise Gzr::CLI::Error, "Space '#{space.name}' is not empty. Space cannot be deleted unless --force is specified"
+            end
+            delete_space(@space)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/space/top.rb

@@ -0,0 +1,62 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/space'
+require 'tty-table'
+
+module Gzr
+  module Commands
+    class Space
+      class Top < Gzr::Command
+        include Gzr::Space
+        def initialize(options)
+          super()
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning("options: #{@options.inspect}") if @options[:debug]
+          with_session do
+            spaces = all_spaces("id,name,is_shared_root,is_users_root,is_root,is_user_root,is_embed_shared_root,is_embed_users_root")
+
+            begin
+              puts "No spaces found"
+              return nil
+            end unless spaces && spaces.length > 0
+
+            table = TTY::Table.new(header: spaces[0].to_attrs.keys) do |t|
+              spaces.each do |h|
+                t << h.to_attrs.values if (
+                  h.is_shared_root || h.is_users_root || h.is_root ||
+                  h.is_user_root || h.is_embed_shared_root || h.is_embed_users_root
+                )
+              end
+            end if spaces[0]
+            output.puts table.render(:ascii, alignments: [:right]) if table
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/space/tree.rb

@@ -0,0 +1,79 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/space'
+require 'tty-tree'
+
+module Gzr
+  module Commands
+    class Space
+      class Tree < Gzr::Command
+        include Gzr::Space
+        def initialize(filter_spec, options)
+          super()
+          @filter_spec = filter_spec
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning("options: #{@options.inspect}") if @options[:debug]
+          with_session do
+            space_ids = process_args([@filter_spec])
+
+            tree_data = Hash.new
+            
+            space_ids.each do |space_id| 
+              s = query_space(space_id, "id,name,parent_id,looks(id,title),dashboards(id,title)")
+              space_name = s.name
+              space_name = "nil (#{s.id})" unless space_name 
+              space_name = "\"#{space_name}\"" if ((space_name != space_name.strip) || (space_name.length == 0))
+              space_name += " (#{space_id})" unless space_ids.length == 1
+              tree_data[space_name] =
+                [ recurse_spaces(s.id) ] +
+                s.looks.map { |l| "(l) #{l.title}" } +
+                  s.dashboards.map { |d| "(d) #{d.title}" }
+            end
+            tree = TTY::Tree.new(tree_data)
+            output.puts tree.render
+          end
+        end
+
+        def recurse_spaces(space_id)
+          data = query_space_children(space_id, "id,name,parent_id,looks(id,title),dashboards(id,title)")
+          tree_branch = Hash.new
+          data.each do |s|
+            space_name = s.name
+            space_name = "nil (#{s.id})" unless space_name 
+            space_name = "\"#{space_name}\"" if ((space_name != space_name.strip) || (space_name.length == 0))
+            tree_branch[space_name] =
+              [ recurse_spaces(s.id) ] +
+              s.looks.map { |l| "(l) #{l.title}" } +
+              s.dashboards.map { |d| "(d) #{d.title}" }
+          end
+          tree_branch
+        end
+      end
+    end
+  end
+end