antigravity-sdk 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5fc78b32cbc7cf1f01aa9dde4e53637d3de9ed140337ed461fde3639b8e73559
-  data.tar.gz: 78a973c5dff6151ccd0485b75556b6e984c980676e9702a044b7df42221824ef
+  metadata.gz: afe9c665bf383ac5f2b9e0cf07ea94f085373c0c2a0d0b10c135a56a639e0eac
+  data.tar.gz: d2cf2720c77a9d6d854c8415c2113c13f3ca91c5aabd86d413cb39422e47cf31
 SHA512:
-  metadata.gz: c853db9ef0827378e9c9e37482df51e815814fd257c03b5b0449f322056a3c07c281e3fa57d984cf61abfa773fa9e06b3624f8615ec85de9b6597f11c7c1a593
-  data.tar.gz: 3a5fb1474f5d45351f73321eae75ee1e239bf7e10cf85d5e192fa451b0298a1be38a3609b808420dd7294fb6fa771eb2fe33227b5833d66fbb5e9cb26189955a
+  metadata.gz: b730d5312a05e57473a765d8398cc3e2b80edcbcf931bf802541ca72f0962b7600c6dbeac7761b4815da34caae823e1b1f6dacd993eb9550d156173aac797a39
+  data.tar.gz: 7cedb00e25e9e8d17d00a6e94588301b75f126c107e4a06fd145db7c87c08fb5d2ea8cc0f78aa3a39577263b5067e0bc7e84e0f4176ec35931a95d7bdbba36d0

data/lib/antigravity/agent.rb

@@ -35,6 +35,12 @@ module Antigravity
       sidecar
     end
 
+    def attach_logger(log_target = "log/antigravity.log", level: ::Logger::INFO)
+      logger_guard = Guards::AgentLogger.new(log_target, level: level)
+      logger_guard.attach_to(self)
+      logger_guard
+    end
+
     def load_skill(skill_path)
       skill = Skill.load(skill_path)
       @skills << skill

data/lib/antigravity/guards.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+require "logger"
+require "fileutils"
+
+module Antigravity
+  module Guards
+    # Configurable opt-in logger guard supporting file loggers and Rails.logger
+    class AgentLogger
+      attr_reader :logger
+
+      def initialize(log_target = "log/antigravity.log", level: Logger::INFO)
+        if log_target.is_a?(String)
+          FileUtils.mkdir_p(File.dirname(log_target)) rescue nil
+          @logger = ::Logger.new(log_target)
+        elsif log_target.respond_to?(:info)
+          @logger = log_target
+        else
+          @logger = ::Logger.new($stdout)
+        end
+        @logger.level = level if @logger.respond_to?(:level=)
+      end
+
+      def before_prompt(prompt_text)
+        @logger.info("[Antigravity::Prompt] User: '#{prompt_text}'")
+      end
+
+      def after_response(response)
+        @logger.info("[Antigravity::Response] Assistant (#{response.model_id}): #{response.content.strip}")
+      end
+
+      def before_tool_call(tool_name, params)
+        @logger.info("[Antigravity::Tool] Executing '#{tool_name}' with params: #{params.inspect}")
+      end
+
+      def after_tool_call(tool_name, params, result)
+        @logger.info("[Antigravity::Tool] Result for '#{tool_name}': #{result}")
+        result
+      end
+
+      def attach_to(agent)
+        logger_guard = self
+        agent.before_prompt { |p| logger_guard.before_prompt(p) }
+        agent.after_response { |r| logger_guard.after_response(r) }
+        agent.before_tool_call { |t, p| logger_guard.before_tool_call(t, p) }
+        agent.after_tool_call { |t, p, r| logger_guard.after_tool_call(t, p, r) }
+      end
+    end
+
+    # Configurable opt-in file protection guard
+    class FileProtection
+      DEFAULT_FILES = [
+        /\.env(\..*)?$/i,
+        /Gemfile(\.lock)?$/i,
+        /config\/secrets\.yml$/i,
+        /config\/database\.yml$/i
+      ].freeze
+
+      attr_reader :protected_files
+
+      def initialize(files: nil, add: [], remove: [])
+        initial_patterns = files ? Array(files) : DEFAULT_FILES
+        patterns = initial_patterns.map { |f| f.is_a?(Regexp) ? f : /#{Regexp.escape(f.to_s)}$/i }
+        patterns += Array(add).map { |f| f.is_a?(Regexp) ? f : /#{Regexp.escape(f.to_s)}$/i }
+        patterns -= Array(remove).map { |f| f.is_a?(Regexp) ? f : /#{Regexp.escape(f.to_s)}$/i }
+        @protected_files = patterns.freeze
+      end
+
+      def call(_tool_name, params)
+        path = extract_path(params)
+        return :allow unless path
+
+        if protected_path?(path)
+          { status: :deny, reason: "FileProtection Guard: Modifications to '#{path}' are restricted." }
+        else
+          :allow
+        end
+      end
+
+      def to_proc
+        method(:call).to_proc
+      end
+
+      private
+
+      def extract_path(params)
+        return nil unless params.is_a?(Hash)
+        params[:path] || params["path"] || params[:file] || params["file"] || params[:target]
+      end
+
+      def protected_path?(path)
+        filename = File.basename(path)
+        @protected_files.any? { |pattern| path =~ pattern || filename =~ pattern }
+      end
+    end
+
+    # Configurable opt-in secret masking filter
+    class SecretMasker
+      DEFAULT_PATTERNS = [
+        /(AIzaSy[A-Za-z0-9_-]{25,45})/,           # Google API Keys
+        /(sk-[A-Za-z0-9]{25,50})/,                # OpenAI API Keys
+        /(ghp_[A-Za-z0-9]{25,45})/,               # GitHub Tokens
+        /(bearer\s+[A-Za-z0-9\._-]{20,})/i       # Bearer Tokens
+      ].freeze
+
+      attr_reader :patterns, :replacement
+
+      def initialize(patterns: nil, replacement: "[REDACTED_SECRET]")
+        @patterns = (patterns ? Array(patterns) : DEFAULT_PATTERNS).freeze
+        @replacement = replacement
+      end
+
+      def call(_tool_name, _params, result)
+        return result unless result.is_a?(String)
+
+        sanitized = result.dup
+        @patterns.each do |pattern|
+          sanitized.gsub!(pattern, @replacement)
+        end
+        sanitized
+      end
+
+      def to_proc
+        method(:call).to_proc
+      end
+    end
+  end
+end

data/lib/antigravity.rb

@@ -5,6 +5,7 @@ require_relative "antigravity/config"
 require_relative "antigravity/message"
 require_relative "antigravity/harness"
 require_relative "antigravity/hooks"
+require_relative "antigravity/guards"
 require_relative "antigravity/sidecar"
 require_relative "antigravity/tool"
 require_relative "antigravity/skill"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: antigravity-sdk
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Riccardo Carlesso
@@ -37,6 +37,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '2.6'
+- !ruby/object:Gem::Dependency
+  name: logger
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '1.5'
 description: An elegant, Ruby-like SDK for building autonomous AI agents with Google
   Antigravity.
 email:
@@ -49,6 +63,7 @@ files:
 - lib/antigravity/agent.rb
 - lib/antigravity/client.rb
 - lib/antigravity/config.rb
+- lib/antigravity/guards.rb
 - lib/antigravity/harness.rb
 - lib/antigravity/hooks.rb
 - lib/antigravity/message.rb