n2b 0.3.1 → 0.5.0

data/bin/n2b-test-jira

@@ -0,0 +1,273 @@
+#!/usr/bin/env ruby
+
+require 'net/http'
+require 'uri'
+require 'json'
+require 'base64'
+require 'yaml'
+
+# Jira API connection tester for N2B
+# Tests authentication, permissions, and specific ticket access
+class JiraConnectionTester
+  def initialize
+    @config_file = File.expand_path('~/.n2b/config.yml')
+    load_config
+  end
+
+  def load_config
+    unless File.exist?(@config_file)
+      puts "❌ Config file not found: #{@config_file}"
+      puts "Please run 'n2b --advanced-config' to set up Jira integration first."
+      exit 1
+    end
+
+    @config = YAML.load_file(@config_file)
+    @jira_config = @config['jira']
+
+    unless @jira_config && @jira_config['domain'] && @jira_config['email'] && @jira_config['api_key']
+      puts "❌ Jira configuration incomplete in #{@config_file}"
+      puts "Missing: domain, email, or api_key"
+      puts "Please run 'n2b --advanced-config' to configure Jira."
+      exit 1
+    end
+
+    # Handle domain that may or may not include protocol
+    domain = @jira_config['domain'].to_s.strip
+    if domain.start_with?('http://') || domain.start_with?('https://')
+      @base_url = "#{domain.chomp('/')}/rest/api/3"
+    else
+      @base_url = "https://#{domain.chomp('/')}/rest/api/3"
+    end
+
+    puts "✅ Config loaded successfully"
+    puts "   Domain: #{@jira_config['domain']}"
+    puts "   Email: #{@jira_config['email']}"
+    puts "   API Key: #{@jira_config['api_key'][0..10]}..." # Show only first part for security
+    puts "   Base URL: #{@base_url}"
+    puts
+  end
+
+  def test_connection
+    puts "🔍 Testing Jira API connection..."
+    puts "=" * 50
+
+    # Test 1: Basic API connectivity
+    test_basic_connectivity
+
+    # Test 2: Authentication
+    test_authentication
+
+    # Test 3: Permissions
+    test_permissions
+
+    # Test 4: Specific ticket access (if provided)
+    if ARGV[0]
+      test_ticket_access(ARGV[0])
+    else
+      puts "💡 To test specific ticket access, run: ruby test_jira_connection.rb TICKET-123"
+    end
+  end
+
+  private
+
+  def test_basic_connectivity
+    puts "1️⃣ Testing basic connectivity to Jira..."
+    
+    begin
+      uri = URI.parse(@base_url)
+      puts "   Connecting to: #{uri.host}:#{uri.port}"
+      
+      http = Net::HTTP.new(uri.host, uri.port)
+      http.use_ssl = (uri.scheme == 'https')
+      http.open_timeout = 10
+      http.read_timeout = 10
+      
+      # Simple GET to the API root
+      request = Net::HTTP::Get.new('/rest/api/3/')
+      response = http.request(request)
+      
+      puts "   ✅ Connection successful (HTTP #{response.code})"
+      puts
+    rescue => e
+      puts "   ❌ Connection failed: #{e.message}"
+      puts "   Check your network connection and domain configuration."
+      exit 1
+    end
+  end
+
+  def test_authentication
+    puts "2️⃣ Testing authentication..."
+    
+    begin
+      response = make_api_request('GET', '/myself')
+      
+      if response['accountId']
+        puts "   ✅ Authentication successful"
+        puts "   Account ID: #{response['accountId']}"
+        puts "   Display Name: #{response['displayName']}"
+        puts "   Email: #{response['emailAddress']}"
+        puts
+      else
+        puts "   ⚠️  Authentication response unexpected:"
+        puts "   #{response}"
+        puts
+      end
+    rescue => e
+      puts "   ❌ Authentication failed: #{e.message}"
+      puts "   Check your email and API token."
+      puts "   API token should be generated from: https://id.atlassian.com/manage-profile/security/api-tokens"
+      exit 1
+    end
+  end
+
+  def test_permissions
+    puts "3️⃣ Testing required permissions..."
+
+    begin
+      # Test 1: Project access (Browse Projects permission)
+      response = make_api_request('GET', '/project')
+
+      if response.is_a?(Array) && response.length > 0
+        puts "   ✅ Browse Projects: Can access #{response.length} projects"
+        response.first(3).each do |project|
+          puts "     - #{project['key']}: #{project['name']}"
+        end
+        puts "     ... (showing first 3)" if response.length > 3
+
+        # Test 2: Issue access (Browse Issues permission)
+        test_project = response.first
+        puts "   🔍 Testing issue access in project: #{test_project['key']}"
+
+        begin
+          issues_response = make_api_request('GET', "/search?jql=project=#{test_project['key']}&maxResults=1")
+          if issues_response['issues'] && issues_response['issues'].length > 0
+            puts "   ✅ Browse Issues: Can access issues in #{test_project['key']}"
+
+            # Test 3: Comment access (if we found an issue)
+            test_issue = issues_response['issues'].first
+            begin
+              comments_response = make_api_request('GET', "/issue/#{test_issue['key']}/comment")
+              puts "   ✅ View Comments: Can access comments on #{test_issue['key']}"
+            rescue => e
+              puts "   ❌ View Comments: Cannot access comments (#{e.message})"
+            end
+
+            # Test 4: Add comment permission (we won't actually add, just check the endpoint)
+            puts "   ℹ️  Add Comments: Will be tested when actually posting comments"
+          else
+            puts "   ⚠️  No issues found in #{test_project['key']} to test comment permissions"
+          end
+        rescue => e
+          puts "   ❌ Browse Issues: Cannot search issues (#{e.message})"
+        end
+
+        puts
+      else
+        puts "   ❌ Browse Projects: No projects accessible"
+        puts "   Your API token needs 'Browse Projects' permission"
+        puts
+      end
+    rescue => e
+      puts "   ❌ Permission test failed: #{e.message}"
+      puts "   Your API token might not have sufficient permissions."
+      puts
+    end
+
+    puts "   📋 Required Jira Permissions for N2B:"
+    puts "   • Browse Projects - to access project list"
+    puts "   • Browse Issues - to read ticket details"
+    puts "   • View Comments - to read ticket comments"
+    puts "   • Add Comments - to post analysis results"
+    puts
+  end
+
+  def test_ticket_access(ticket_key)
+    puts "4️⃣ Testing access to specific ticket: #{ticket_key}..."
+    
+    begin
+      # Test ticket access
+      response = make_api_request('GET', "/issue/#{ticket_key}")
+      
+      puts "   ✅ Ticket access successful"
+      puts "   Key: #{response['key']}"
+      puts "   Summary: #{response.dig('fields', 'summary')}"
+      puts "   Status: #{response.dig('fields', 'status', 'name')}"
+      puts "   Assignee: #{response.dig('fields', 'assignee', 'displayName') || 'Unassigned'}"
+      puts
+      
+      # Test comments access
+      comments_response = make_api_request('GET', "/issue/#{ticket_key}/comment")
+      comment_count = comments_response.dig('comments')&.length || 0
+      puts "   ✅ Comments access successful (#{comment_count} comments)"
+      puts
+      
+    rescue => e
+      puts "   ❌ Ticket access failed: #{e.message}"
+      
+      if e.message.include?('404')
+        puts "   Possible causes:"
+        puts "   - Ticket doesn't exist"
+        puts "   - You don't have permission to view this ticket"
+        puts "   - Ticket is in a project you can't access"
+      elsif e.message.include?('401')
+        puts "   Authentication issue - check your API token"
+      elsif e.message.include?('403')
+        puts "   Permission denied - your account can't access this ticket"
+      end
+      puts
+    end
+  end
+
+  def make_api_request(method, path, body = nil)
+    full_url = "#{@base_url}#{path}"
+    uri = URI.parse(full_url)
+    
+    http = Net::HTTP.new(uri.host, uri.port)
+    http.use_ssl = (uri.scheme == 'https')
+    http.read_timeout = 30
+    http.open_timeout = 10
+
+    request = case method.upcase
+              when 'GET'
+                Net::HTTP::Get.new(uri.request_uri)
+              when 'POST'
+                req = Net::HTTP::Post.new(uri.request_uri)
+                req.body = body.to_json if body
+                req
+              else
+                raise "Unsupported HTTP method: #{method}"
+              end
+
+    request['Authorization'] = "Basic #{Base64.strict_encode64("#{@jira_config['email']}:#{@jira_config['api_key']}")}"
+    request['Content-Type'] = 'application/json'
+    request['Accept'] = 'application/json'
+
+    response = http.request(request)
+
+    unless response.is_a?(Net::HTTPSuccess)
+      error_message = "Jira API Error: #{response.code} #{response.message}"
+      error_message += " - #{response.body}" if response.body && !response.body.empty?
+      raise error_message
+    end
+
+    response.body.empty? ? {} : JSON.parse(response.body)
+  rescue Timeout::Error, Errno::EINVAL, Errno::ECONNRESET, EOFError,
+         Net::HTTPBadResponse, Net::HTTPHeaderSyntaxError, Net::ProtocolError, Errno::ECONNREFUSED => e
+    raise "Jira API request failed: #{e.class} - #{e.message}"
+  end
+end
+
+# Run the test
+if __FILE__ == $0
+  puts "🧪 Jira API Connection Tester"
+  puts "=" * 50
+  puts
+
+  tester = JiraConnectionTester.new
+  tester.test_connection
+
+  puts "🎉 Test completed!"
+  puts
+  puts "If all tests passed, your Jira integration should work correctly."
+  puts "If any tests failed, check the error messages above for guidance."
+end

data/lib/n2b/base.rb

@@ -1,3 +1,5 @@
+require_relative 'model_config'
+
 module N2B
   class Base
 
@@ -12,55 +14,216 @@ module N2B
       end
     end
     
-    def get_config( reconfigure: false)
+    def get_config(reconfigure: false, advanced_flow: false)
       config = load_config
-      api_key = ENV['CLAUDE_API_KEY'] || config['access_key']
-      model = config['model'] || 'sonnet35'
-  
-      if api_key.nil? || api_key == '' ||  reconfigure
-        print "choose a language model to use (1:claude, 2:openai, 3:gemini) #{ config['llm'] }: "
-        llm = $stdin.gets.chomp
-        llm = config['llm'] if llm.empty?
-        unless ['claude', 'openai', 'gemini', '1', '2', '3'].include?(llm)
-          puts "Invalid language model. Choose from: claude, openai, gemini"
-          exit 1
+      api_key = ENV['CLAUDE_API_KEY'] || config['access_key'] # This will be used as default or for existing configs
+      _model = config['model'] # Unused but kept for potential future use # Model will be handled per LLM
+
+      # Determine if it's effectively a first-time setup for core LLM details
+      is_first_time_core_setup = config['llm'].nil?
+
+      if api_key.nil? || api_key == '' || config['llm'].nil? || reconfigure
+        puts "--- N2B Core LLM Configuration ---"
+        print "Choose a language model to use (1:claude, 2:openai, 3:gemini, 4:openrouter, 5:ollama) [current: #{config['llm']}]: "
+        llm_choice = $stdin.gets.chomp
+        llm_choice = config['llm'] if llm_choice.empty? && !config['llm'].nil? # Keep current if input is empty
+
+        unless ['claude', 'openai', 'gemini', 'openrouter', 'ollama', '1', '2', '3', '4', '5'].include?(llm_choice)
+          puts "Invalid language model selection. Defaulting to 'claude' or previous if available."
+          llm_choice = config['llm'] || 'claude' # Fallback
+        end
+
+        selected_llm = case llm_choice
+                       when '1', 'claude' then 'claude'
+                       when '2', 'openai' then 'openai'
+                       when '3', 'gemini' then 'gemini'
+                       when '4', 'openrouter' then 'openrouter'
+                       when '5', 'ollama' then 'ollama'
+                       else config['llm'] || 'claude' # Should not happen due to validation
+                       end
+        config['llm'] = selected_llm
+
+        if selected_llm == 'ollama'
+          puts "\n--- Ollama Specific Configuration ---"
+          puts "Ollama typically runs locally and doesn't require an API key."
+
+          # Use new model configuration system
+          current_model = config['model']
+          model_choice = N2B::ModelConfig.get_model_choice(selected_llm, current_model) # Renamed to model_choice to avoid conflict
+          config['model'] = model_choice
+
+          current_ollama_api_url = config['ollama_api_url'] || N2M::Llm::Ollama::DEFAULT_OLLAMA_API_URI
+          print "Enter Ollama API base URL [current: #{current_ollama_api_url}]: "
+          ollama_api_url_input = $stdin.gets.chomp
+          config['ollama_api_url'] = ollama_api_url_input.empty? ? current_ollama_api_url : ollama_api_url_input
+          config.delete('access_key') # Remove access_key if switching to ollama
+        else
+          # Configuration for API key based LLMs
+          puts "\n--- #{selected_llm.capitalize} Specific Configuration ---"
+          current_api_key = config['access_key']
+          print "Enter your #{selected_llm} API key #{ current_api_key.nil? || current_api_key.empty? ? '' : '[leave blank to keep current]' }: "
+          api_key_input = $stdin.gets.chomp
+          config['access_key'] = api_key_input if !api_key_input.empty?
+          config['access_key'] = current_api_key if api_key_input.empty? && !current_api_key.nil? && !current_api_key.empty?
+
+          if config['access_key'].nil? || config['access_key'].empty?
+            puts "API key is required for #{selected_llm}."
+            exit 1
+          end
+
+          current_model = config['model']
+          model_choice = N2B::ModelConfig.get_model_choice(selected_llm, current_model) # Renamed
+          config['model'] = model_choice
+
+          if selected_llm == 'openrouter'
+            current_site_url = config['openrouter_site_url'] || ""
+            print "Enter your OpenRouter Site URL (optional, for HTTP-Referer) [current: #{current_site_url}]: "
+            openrouter_site_url_input = $stdin.gets.chomp
+            config['openrouter_site_url'] = openrouter_site_url_input.empty? ? current_site_url : openrouter_site_url_input
+
+            current_site_name = config['openrouter_site_name'] || ""
+            print "Enter your OpenRouter Site Name (optional, for X-Title) [current: #{current_site_name}]: "
+            openrouter_site_name_input = $stdin.gets.chomp
+            config['openrouter_site_name'] = openrouter_site_name_input.empty? ? current_site_name : openrouter_site_name_input
+          end
+        end
+
+        # --- Advanced Configuration Flow ---
+        # Prompt for advanced settings if advanced_flow is true or if it's the first time setting up core LLM.
+        prompt_for_advanced = advanced_flow || is_first_time_core_setup
+
+        if prompt_for_advanced
+          puts "\n--- Advanced Settings ---"
+          print "Would you like to configure advanced settings (e.g., Jira integration, privacy)? (y/n) [default: n]: "
+          choice = $stdin.gets.chomp.downcase
+
+          if choice == 'y'
+            # Jira Configuration
+            puts "\n--- Jira Integration ---"
+            puts "You can generate a Jira API token here: https://support.atlassian.com/atlassian-account/docs/manage-api-tokens-for-your-atlassian-account/"
+            config['jira'] ||= {}
+            print "Jira Domain (e.g., your-company.atlassian.net) [current: #{config['jira']['domain']}]: "
+            config['jira']['domain'] = $stdin.gets.chomp.then { |val| val.empty? ? config['jira']['domain'] : val }
+
+            print "Jira Email Address [current: #{config['jira']['email']}]: "
+            config['jira']['email'] = $stdin.gets.chomp.then { |val| val.empty? ? config['jira']['email'] : val }
+
+            print "Jira API Token #{config['jira']['api_key'] ? '[leave blank to keep current]' : ''}: "
+            api_token_input = $stdin.gets.chomp
+            config['jira']['api_key'] = api_token_input if !api_token_input.empty?
+
+            print "Default Jira Project Key (optional, e.g., MYPROJ) [current: #{config['jira']['default_project']}]: "
+            config['jira']['default_project'] = $stdin.gets.chomp.then { |val| val.empty? ? config['jira']['default_project'] : val }
+
+            # Privacy Settings
+            puts "\n--- Privacy Settings ---"
+            config['privacy'] ||= {}
+            puts "Allow sending shell command history to the LLM? (true/false) [current: #{config['privacy']['send_shell_history']}]"
+            config['privacy']['send_shell_history'] = $stdin.gets.chomp.then { |val| val.empty? ? config['privacy']['send_shell_history'] : (val.downcase == 'true') }
+
+            puts "Allow sending LLM interaction history (your prompts and LLM responses) to the LLM? (true/false) [current: #{config['privacy']['send_llm_history']}]"
+            config['privacy']['send_llm_history'] = $stdin.gets.chomp.then { |val| val.empty? ? config['privacy']['send_llm_history'] : (val.downcase == 'true') }
+
+            puts "Allow sending current directory to the LLM? (true/false) [current: #{config['privacy']['send_current_directory']}]"
+            config['privacy']['send_current_directory'] = $stdin.gets.chomp.then { |val| val.empty? ? config['privacy']['send_current_directory'] : (val.downcase == 'true') }
+
+            puts "Append n2b generated commands to your shell history file? (true/false) [current: #{config['append_to_shell_history']}]"
+            # Note: append_to_shell_history was previously outside 'privacy' hash. Standardizing it inside.
+            config['append_to_shell_history'] = $stdin.gets.chomp.then { |val| val.empty? ? config['append_to_shell_history'] : (val.downcase == 'true') }
+            config['privacy']['append_to_shell_history'] = config['append_to_shell_history'] # Also place under privacy for consistency
+
+          else # User chose 'n' for advanced settings
+            # Ensure Jira config is empty or defaults are cleared if user opts out of advanced
+            config['jira'] ||= {} # Ensure it exists
+            # If they opt out, we don't clear existing, just don't prompt.
+            # If it's a fresh setup and they opt out, these will remain empty/nil.
+
+            # Ensure privacy settings are initialized to defaults if not already set by advanced flow
+            config['privacy'] ||= {}
+            config['privacy']['send_shell_history'] = config['privacy']['send_shell_history'] || false
+            config['privacy']['send_llm_history'] = config['privacy']['send_llm_history'] || true # Default true
+            config['privacy']['send_current_directory'] = config['privacy']['send_current_directory'] || true # Default true
+            config['append_to_shell_history'] = config['append_to_shell_history'] || false
+            config['privacy']['append_to_shell_history'] = config['append_to_shell_history']
+          end
+        else # Not prompting for advanced (neither advanced_flow nor first_time_core_setup)
+          # Ensure defaults for privacy if they don't exist from a previous config
+          config['jira'] ||= {} # Ensure Jira key exists
+          config['privacy'] ||= {}
+          config['privacy']['send_shell_history'] = config['privacy']['send_shell_history'] || false
+          config['privacy']['send_llm_history'] = config['privacy']['send_llm_history'] || true
+          config['privacy']['send_current_directory'] = config['privacy']['send_current_directory'] || true
+          config['append_to_shell_history'] = config['append_to_shell_history'] || false
+          config['privacy']['append_to_shell_history'] = config['append_to_shell_history']
         end
-        llm = 'claude' if llm == '1'
-        llm = 'openai' if llm == '2'
-        llm = 'gemini' if llm == '3'
-        llm_class = case llm
-                   when 'openai'
-                     N2M::Llm::OpenAi
-                   when 'gemini'
-                     N2M::Llm::Gemini
-                   else
-                     N2M::Llm::Claude
-                   end
-
-        print "Enter your #{llm} API key: #{ api_key.nil? || api_key.empty? ? '' : '(leave blank to keep the current key '+api_key[0..10]+'...)' }"
-        api_key = $stdin.gets.chomp 
-        api_key = config['access_key'] if api_key.empty?
-        print "Choose a model (#{ llm_class::MODELS.keys }, #{ llm_class::MODELS.keys.first } default): "
-        model = $stdin.gets.chomp 
-        model = llm_class::MODELS.keys.first if model.empty?
-        config['llm'] = llm
-        config['access_key'] = api_key
-        config['model'] = model
-        unless llm_class::MODELS.keys.include?(model)
-          puts "Invalid model. Choose from: #{llm_class::MODELS.keys.join(', ')}"
-          exit 1
+
+        # Validate configuration before saving
+        validation_errors = validate_config(config)
+        if validation_errors.any?
+          puts "\n⚠️  Configuration validation warnings:"
+          validation_errors.each { |error| puts "  - #{error}" }
+          puts "Configuration saved with warnings."
         end
-        puts "configure privacy settings directly in the config file #{CONFIG_FILE}"
-        config['privacy'] ||= {} 
-        config['privacy']['send_shell_history'] = false
-        config['privacy']['send_llm_history'] = true
-        config['privacy']['send_current_directory'] =true
-        config['append_to_shell_history'] = false 
-        puts "Current configuration: #{config['privacy']}"
+
+        puts "\nConfiguration saved to #{CONFIG_FILE}"
         FileUtils.mkdir_p(File.dirname(CONFIG_FILE)) unless File.exist?(File.dirname(CONFIG_FILE))
         File.write(CONFIG_FILE, config.to_yaml)
+      else
+        # If not reconfiguring, still ensure privacy and jira keys exist with defaults if missing
+        # This handles configs from before these settings were introduced
+        config['jira'] ||= {}
+        config['privacy'] ||= {}
+        config['privacy']['send_shell_history'] = config['privacy']['send_shell_history'] || false
+        config['privacy']['send_llm_history'] = config['privacy']['send_llm_history'] || true
+        config['privacy']['send_current_directory'] = config['privacy']['send_current_directory'] || true
+        # append_to_shell_history was outside 'privacy' before, ensure it's correctly defaulted
+        # and also placed under 'privacy' for future consistency.
+        current_append_setting = config.key?('append_to_shell_history') ? config['append_to_shell_history'] : false
+        config['append_to_shell_history'] = current_append_setting
+        config['privacy']['append_to_shell_history'] = config['privacy']['append_to_shell_history'] || current_append_setting
       end
       config
     end
+
+    private
+
+    def validate_config(config)
+      errors = []
+
+      # Validate LLM configuration
+      if config['llm'].nil? || config['llm'].empty?
+        errors << "LLM provider not specified"
+      end
+
+      # Validate model name
+      if config['model'].nil? || config['model'].empty?
+        errors << "Model not specified"
+      elsif config['model'].length < 3
+        errors << "Model name '#{config['model']}' seems too short - might be invalid"
+      elsif %w[y n yes no true false].include?(config['model'].downcase)
+        errors << "Model name '#{config['model']}' appears to be a boolean response rather than a model name"
+      end
+
+      # Validate API key for non-Ollama providers
+      if config['llm'] != 'ollama' && (config['access_key'].nil? || config['access_key'].empty?)
+        errors << "API key missing for #{config['llm']} provider"
+      end
+
+      # Validate Jira configuration if present
+      if config['jira'] && !config['jira'].empty?
+        jira_config = config['jira']
+        if jira_config['domain'] && !jira_config['domain'].empty?
+          # If domain is set, email and api_key should also be set
+          if jira_config['email'].nil? || jira_config['email'].empty?
+            errors << "Jira email missing when domain is configured"
+          end
+          if jira_config['api_key'].nil? || jira_config['api_key'].empty?
+            errors << "Jira API key missing when domain is configured"
+          end
+        end
+      end
+
+      errors
+    end
   end
 end