chef_fixie 0.1.0 → 0.5.0

data/lib/chef_fixie/console.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2015 Chef Software Inc. 
+# Copyright (c) 2015 Chef Software Inc.
 # License :: Apache License, Version 2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -18,20 +18,21 @@
 #
 # Much of this code was orginally derived from the orgmapper tool, which had many varied authors.
 
-require 'optparse'
-require 'pp'
-require 'pry'
+require "optparse"
+require "pp"
+require "pry"
 
-require 'chef_fixie'
-require 'chef_fixie/context'
+require_relative "../chef_fixie"
+require_relative "context"
 
 module ChefFixie
   module Console
     extend self
 
     def start
+      @@started_from_command_line = true
       configure
-      Fixie.setup
+      ChefFixie.setup
       configure_pry
       Pry.start
     end
@@ -41,14 +42,14 @@ module ChefFixie
       if ARGV.first && ARGV[0].chars.first != "-" && config_file = ARGV.shift
         config_file = File.expand_path(config_file)
       end
-      Fixie.load_config(config_file)
+      ChefFixie.load_config(config_file)
 
       options = {}
       OptionParser.new do |opt|
         opt.banner = "Usage: fixie [config] [options]"
-        opt.on('--authz_uri AUTH_URI', "The URI of the opscode authz service") { |v| options[:authz_uri] =v }
-        opt.on("--sql_database DATABASE", 'The URI of the opscode_chef database') { |v| options[:sql_database] = v }
-        opt.on_tail('-h', '--help', 'Show this message') do
+        opt.on("--authz_uri AUTH_URI", "The URI of the opscode authz service") { |v| options[:authz_uri] = v }
+        opt.on("--sql_database DATABASE", "The URI of the opscode_chef database") { |v| options[:sql_database] = v }
+        opt.on_tail("-h", "--help", "Show this message") do
           puts opt
           puts "\nExample configuration file:\n\n"
           puts ChefFixie::Config.instance.example_config
@@ -67,7 +68,7 @@ module ChefFixie
       Pry.config.history.file = "~/.fixie_history"
       Pry.config.prompt_name = "fixie"
       Pry::Commands.block_command("fixie-help", "Show fixie's help") do
-      output.puts(<<-HALP)
+        output.puts(<<-HALP)
 ** ORGS **
 * access with ORGS or ORGS
 * access a specific org: ORGS['orgname']
@@ -87,5 +88,9 @@ HALP
       end
     end
 
+    def started_from_command_line?
+      @@started_from_command_line == true
+    end
+
   end
 end

data/lib/chef_fixie/context.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2015 Chef Software Inc. 
+# Copyright (c) 2015 Chef Software Inc.
 # License :: Apache License, Version 2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -18,9 +18,8 @@
 #
 # Much of this code was orginally derived from the orgmapper tool, which had many varied authors.
 
-
 module ChefFixie
-    module Context
+  module Context
 
     def describe_orgs
       OrgMetrics.org_stats(orgs)
@@ -67,6 +66,5 @@ module ChefFixie
 
       ChefFixie::Dissociator.dissociate_user(org, user)
     end
-
   end
 end

data/lib/chef_fixie/sql.rb

@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2014-2015 Chef Software Inc. 
+# Copyright (c) 2014-2015 Chef Software Inc.
 # License :: Apache License, Version 2.0
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
@@ -16,20 +16,20 @@
 #
 # Author: Mark Anderson <mark@chef.io>
 
-require 'ffi_yajl'
-require 'uuidtools'
-require 'sequel'
+require "ffi_yajl"
+require "uuidtools"
+require "sequel"
 
-require 'chef_fixie/config'
+require_relative "config"
 
 Sequel.default_timezone = :utc
 
 module ChefFixie
   module Sql
-    
+
     class InvalidConfig < StandardError
     end
-    
+
     # A connection string passed to Sequel.connect()
     #
     # Examples:
@@ -46,25 +46,25 @@ module ChefFixie
 
     # Returns the connection string or raises an error if you didn't set one.
     def self.connection_string
-      @connection_string ||= ChefFixie.configure {|x| x.sql_database }
+      @connection_string ||= ChefFixie.configure { |x| x.sql_database }
     end
-    
+
     # Returns a Sequel::Data baseobject, which wraps access to the database.
     def self.default_connection
       @database ||= Sequel.connect(connection_string, :max_connections => 2)
       #      @database.loggers << Logger.new($stdout)
     end
-    
+
     # Generate a new UUID. Currently uses the v1 UUID scheme.
     def new_uuid
       UUIDTools::UUID.timestamp_create.hexdigest
     end
-    
+
     # Parse the portion of the object that's stored as a blob o' JSON
     def from_json(serialized_data)
       FFI_Yajl::Parser.parse(serialized_data, :symbolize_keys => true)
     end
-    
+
     # Encode the portion of the object that's stored as a blob o' JSON
     def as_json(data)
       FFI_Yajl::Encoder.encode(data)

data/lib/chef_fixie/sql_objects.rb

@@ -17,12 +17,12 @@
 # Author: Mark Anderson <mark@chef.io>
 #
 
-require 'pp'
-require 'sequel'
+require "pp"
+require "sequel"
 
-require 'chef_fixie/config'
-require 'chef_fixie/authz_objects'
-require 'chef_fixie/authz_mapper'
+require_relative "config"
+require_relative "authz_objects"
+require_relative "authz_mapper"
 
 Sequel.extension :inflector
 
@@ -39,18 +39,40 @@ module ChefFixie
         "ChefFixie::Sql" + "::" # this should be autogenerated not hardcoded
       end
 
+      def self.to_name(class_or_name)
+        name =
+          case
+          when class_or_name.is_a?(Symbol)
+            class_or_name.to_s
+          when class_or_name.is_a?(Class)
+            class_or_name.name
+          when class_or_name.is_a?(String)
+            class_or_name
+          else
+            class_or_name.class.to_s
+          end
+        name.split("::")[-1]
+      end
+
       # The class for the table, e.g. Orgs
       def self.table_class(name)
+        name = to_name(name)
         (base + name.to_s.pluralize.camelize).constantize
       end
+
       # The class for one instance of the object, e.g. Org
       def self.object_class(name)
+        name = to_name(name)
         (base + name.to_s.singularize.camelize).constantize
       end
+
       def self.singular(name)
+        name = to_name(name)
         name.to_s.singularize
       end
+
       def self.plural(name)
+        name = to_name(name)
         name.to_s.pluralize
       end
     end
@@ -60,38 +82,51 @@ module ChefFixie
       def initialize(data)
         @data = data
       end
+
       def data
         @data
       end
 
+      def table
+        Relationships.table_class(self).new
+      end
+
       # TODO rework this to use better style
       def self.ro_access(*args)
         args.each do |field|
           fundef = "def #{field}; @data.#{field}; end"
-          self.class_eval(fundef)
+          class_eval(fundef)
         end
       end
       # TODO figure out model for write access
 
       def self.name_field(field)
         fundef = "def name; @data.#{field}; end"
-        self.class_eval(fundef)
+        class_eval(fundef)
       end
 
       def self.std_timestamp
         [:created_at, :updated_at].each do |i|
-          self.ro_access(i)
+          ro_access(i)
         end
       end
+
       # Pretty much any object with an authz id has these fields
       def self.std_authz
-        self.std_timestamp
+        std_timestamp
         [:authz_id, :last_updated_by].each do |i|
-          self.ro_access(i)
+          ro_access(i)
         end
       end
 
-
+      def delete
+        rows = table.by_id(id)
+        raise "id #{id} matches more than one object" if rows.all.count != 1
+        rows.inner.delete
+        if respond_to?(:authz_delete)
+          authz_delete
+        end
+      end
     end
 
     class Org < SqlObject
@@ -102,13 +137,14 @@ module ChefFixie
           funname = Relationships.plural(object)
           # defer evaluation of mapper to make sure we have a chance for everyone to initialize
           fundef = "def #{funname}; Relationships.table_class(:#{object}).new.by_org_id(org_id); end"
-          self.class_eval(fundef)
+          class_eval(fundef)
         end
       end
 
       def initialize(data)
         super(data)
       end
+
       def org_id
         data[:id]
       end
@@ -116,7 +152,47 @@ module ChefFixie
       def global_admins
         name = self.name
         global_admins_name = "#{name}_global_admins"
-        ChefFixie::Sql::Groups.new["#{name}_global_admins"]
+        read_access_name = "#{name}_read_access_group"
+        ChefFixie::Sql::Groups.new[global_admins_name] || \
+          ChefFixie::Sql::Groups.new[read_access_name]
+      end
+
+      alias read_access_group global_admins
+
+      # Iterators for objects in authz; using containers to enumerate things
+      # It might be better to metaprogram this up instead,
+      #
+      # TODO Write some tests to validate that this stuff
+      # works, since it depends on a lot of name magic...
+
+      NAME_FIXUP = { "data" => "data_bags", "sandboxes" => nil }
+      def objects_by_container_type(container)
+        name = NAME_FIXUP.has_key?(container) ? NAME_FIXUP[container] : container
+        return [] if name.nil?
+
+        object_type = name.to_sym
+        #        raise Exception "No such object_type #{object_type}" unless respond_to?(object_type)
+        send(object_type).all(:all)
+      end
+
+      def each_authz_object_by_class
+        containers = self.containers.all(:all)
+        containers.each do |container|
+          objects = objects_by_container_type(container.name)
+          if block_given?
+            yield objects
+          end
+        end
+        nil
+      end
+
+      def each_authz_object
+        each_authz_object_by_class do |objectlist|
+          objectlist.each do |object|
+            yield object
+          end
+        end
+        nil
       end
 
       scoped_type :container, :group, :client,
@@ -188,7 +264,6 @@ module ChefFixie
     # org_migration_state_id_seq policy_revisions
     # policy_revisions_policy_groups_association sandboxed_checksums
 
-
     class CookbookArtifact < SqlObject
       include AuthzObjectMixin
       def initialize(data)
@@ -269,35 +344,37 @@ module ChefFixie
       def self.max_count_default
         50
       end
-      
+
       def get_table
         :unknown_table
       end
+
       def mk_element(x)
         x
       end
 
       def initialize(tablespec = nil)
         ChefFixie::Sql.default_connection
-        @inner = tablespec || Sequel::Model(self.get_table)
+        @inner = tablespec || Sequel::Model(get_table)
       end
+
       def inner
         # Make sure we have init
         @inner
       end
 
       def filter_core(field, exp)
-        self.class.new(inner.filter(field=>exp))
+        self.class.new(inner.filter(field => exp))
       end
 
-      def all(max_count=:default)
+      def all(max_count = :default)
         if max_count == :default
           max_count = ChefFixie::Sql::SqlTable.max_count_default
         end
         if max_count != :all
-          return :too_many_results if (inner.count > max_count)
+          return :too_many_results if inner.count > max_count
         end
-        elements = inner.all.map {|org| mk_element(org) }
+        elements = inner.all.map { |org| mk_element(org) }
       end
 
       #
@@ -306,7 +383,7 @@ module ChefFixie
       #     https://stackoverflow.com/questions/9658724/ruby-metaprogramming-class-eval/9658775#9658775
       def self.primary(arg)
         name = :"by_#{arg}"
-        self.class_eval("def [](arg); #{name}(arg).all(1).first; end")
+        class_eval("def [](arg); #{name}(arg).all(1).first; end")
 
         listfun = <<EOLF
 def list(max_count=:default)
@@ -318,26 +395,27 @@ def list(max_count=:default)
   end
 end
 EOLF
-        self.class_eval(listfun)
+        class_eval(listfun)
       end
 
       def self.filter_by(*args)
         args.each do |field|
           name = "by_#{field}"
           fundef = "def #{name}(exp); filter_core(:#{field},exp); end"
-          self.class_eval(fundef)
+          class_eval(fundef)
         end
       end
 
       def self.table(name)
         fundef = "def get_table; :#{name}; end"
-        self.class_eval(fundef)
+        class_eval(fundef)
       end
+
       # doesn't work yet
       # element Org in class Orgs will fail because it can't find Org (undefined)
       def self.element(name)
         fundef = "ElementType = name; def mk_element(x); #{name}.new(x); end"
-        self.class_eval(fundef)
+        class_eval(fundef)
       end
     end
 
@@ -349,7 +427,7 @@ EOLF
       primary :name
       filter_by :name, :id, :full_name, :authz_id
 
-      GlobalOrg = "0"*32
+      GlobalOrg = "0" * 32
 
       def self.org_guid_to_name(guid)
         "global" if guid == GlobalOrg
@@ -370,7 +448,7 @@ EOLF
 
       def by_org_id_user_id(org_id, user_id)
         # db table constraint guarantees that this is unique
-        inner.filter(:org_id=>org_id, :user_id=>user_id).all.first
+        inner.filter(:org_id => org_id, :user_id => user_id).all.first
       end
 
     end
@@ -380,7 +458,7 @@ EOLF
 
       def by_org_id_user_id(org_id, user_id)
         # db table constraint guarantees that this is unique
-        inner.filter(:org_id=>org_id, :user_id=>user_id).all.first
+        inner.filter(:org_id => org_id, :user_id => user_id).all.first
       end
     end
     class Users < SqlTable
@@ -482,7 +560,7 @@ EOLF
       filter_by :name, :id, :org_id, :authz_id
     end
 
-    class Roles  < SqlTable
+    class Roles < SqlTable
       table :roles
       element Sql::Role
       register_authz :role, :object
@@ -491,7 +569,5 @@ EOLF
       filter_by :name, :id, :org_id, :authz_id, :last_updated_by
     end
 
-
-
   end
 end