tng 0.1.4

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: 58907277090e7b363bd0907be694ee5023ec4b18b0d24a4ab0956ba2e29fd648
+  data.tar.gz: 055abaa322144382a523118edbef976b6ded86425b18d5c6b0308c5e282d3d89
+SHA512:
+  metadata.gz: e321fee31b6b91e63c54a318b30f28b66cf05e4e342a2cca2fbc67d1fa76e86ee9102fd788ede777bd6d8f80f32bc7d46bfa985db7b82d4e1b91efe2e32addc3
+  data.tar.gz: eca7b0234cf41ef0466f4016229a9236c577e4b0488a227ffa96cfb2baac4587309e70dc95ef6460bad0b4225f98582f00e1c2a50c310e7259a39e3d97f0c3ea

data/Gemfile

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+source "https://rubygems.org"
+
+# Specify your gem's dependencies in tng.gemspec
+gemspec
+
+gem "irb"
+gem "rake", "~> 13.0"
+
+gem "rake-compiler"
+
+gem "minitest", "~> 5.16"
+
+gem "rubocop", "~> 1.21"

data/LICENSE.md

@@ -0,0 +1,32 @@
+# Commercial License
+
+Copyright (c) 2025 TNG. All rights reserved.
+
+## License Grant
+
+This software is available under a commercial license. You may use this software if you have obtained a valid license from the copyright holder.
+
+## Restrictions
+
+Without a valid commercial license, you may NOT:
+- Use this software in any commercial or production environment
+- Fork, copy, modify, or create derivative works
+- Distribute, redistribute, sublicense, or sell copies
+- Reverse engineer or decompile the software
+- Remove or alter any copyright notices
+
+## Commercial Use
+
+To obtain a commercial license for production use, contact: [claudiu@tng.sh]
+
+## Evaluation Use
+
+You may evaluate this software for a period of 30 days for non-commercial purposes only.
+
+## Termination
+
+Any unauthorized use will result in immediate termination of rights and may result in legal action.
+
+## Disclaimer
+
+THIS SOFTWARE IS PROVIDED "AS IS" WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED.

data/README.md

@@ -0,0 +1,427 @@
+# TNG - AI-Powered Test Generation for Rails
+
+TNG is an AI-powered test generation tool that automatically creates comprehensive tests for your Rails applications. It analyzes your code structure, authentication patterns, and configuration to generate high-quality, contextual tests that follow your project's conventions.
+
+## Installation
+
+### Standard Installation
+
+Add this line to your application's Gemfile:
+
+```ruby
+gem 'tng'
+```
+
+And then execute:
+
+```bash
+bundle install
+```
+
+## Configuration
+
+TNG requires proper configuration to generate high-quality tests. Generate the configuration file by running:
+
+```bash
+rails generate tng:install
+```
+
+This creates `config/initializers/tng.rb` with a configuration template that you'll need to customize for your application. The generator will attempt to detect your existing setup and pre-configure common settings.
+
+### Configuration Options Explained
+
+#### API Configuration
+- **`api_key`**: Your TNG API key from https://app.tng.sh/
+- **`base_url`**: TNG service endpoint (automatically configured, do not modify)
+
+#### Code Reading
+- **`read_source_code`**: Analyzes the specific code file for which tests are being generated
+- **`read_test_code`**: Reads existing tests for the specific file to match your testing patterns
+
+#### Testing Framework Settings
+- **`testing_framework`**: Automatically detected. Possible values: `minitest` or `rspec`
+
+*Note: The following settings depend on your testing framework.*
+
+**For Minitest:**
+- **`test_style`**: Test organization style. Possible options:
+    - `test_block` - `test "does something" do ... end`
+    - `spec` - `it "does something" do ... end`
+    - `unit` - `def test_does_something ... end`
+- **`assertion_style`**: Assertion syntax preference. Possible values:
+    - `assert/refute` - Uses syntax like `assert`, `assert_equal`, `assert_nil`, `refute`, `refute_nil`
+    - `assert/assert_not` - Uses syntax like `assert`, `assert_equal`, `assert_nil`, `assert_not`, `assert_not_equl`, `assert_not_nil`
+    - `must/wont` - `value.must_equal`, `value.must_be_instance_of`, `value.must_raise`, `value.wont_equal`, `value.wont_be_instance_of`, `value.wont_raise`
+
+**For RSpec:**
+- **`describe_style`**: Include describe blocks. Possible values: `true`, `false`
+- **`context_style`**: Context block style. Possible values: `context`, `describe`
+- **`it_style`**: Example block style. Possible values: `it`, `specify`
+- **`before_style`**: Setup block style. Possible values: `before`, `setup`
+- **`after_style`**: Teardown block style. Possible values: `after`, `teardown`
+- **`let_style`**: Use let helpers. Possible values: `true`, `false`
+- **`subject_style`**: Use subject helpers. Possible values: `true`, `false`
+
+#### Authentication Methods
+Supported authentication types:
+- **`session`**: Session-based authentication
+- **`devise`**: Devise gem integration
+- **`jwt`**: JSON Web Token authentication
+- **`token_auth`**: Token-based authentication
+- **`basic_auth`**: HTTP Basic Authentication
+- **`oauth`**: OAuth authentication
+- **`headers`**: Custom header authentication
+- **`custom`**: Custom authentication logic
+
+#### Mock and Factory Libraries
+- **`mock_library`**: Mock library preference
+    - **For Minitest**: `minitest/mock`, `mocha`, `nil`
+    - **For RSpec**: `rspec-mocks`, `mocha`, `flexmock`, `nil`
+- **`http_mock_library`**: HTTP mocking library. Possible values: `webmock`, `vcr`, `httparty`, `nil`
+- **`factory_library`**: Factory library preference. Possible values: `factory_bot`, `factory_girl`, `fabrication`, `fabricator`, `fixtures`, `active_record`
+
+## Configuration Examples
+
+### Minitest with FactoryBot Setup
+
+```ruby
+# config/initializers/tng.rb
+Tng.configure do |config|
+  config.api_key = ENV["TNG_API_KEY"]
+  config.base_url = "https://app.tng.sh/"
+  
+  config.read_source_code = true
+  config.read_test_code = true
+  
+  config.testing_framework = "minitest"
+  config.test_style = "test_block"
+  config.assertion_style = "assert/refute"
+  config.setup_style = true
+  config.teardown_style = false
+  
+  config.mock_library = "minitest/mock"
+  config.http_mock_library = "webmock"
+  config.factory_library = "factory_bot"
+  
+  config.authentication_enabled = true
+  config.authentication_methods = [
+    {
+      method: "authenticate_user!",
+      file_location: "app/controllers/application_controller.rb",
+      auth_type: "session"
+    }
+  ]
+end
+```
+
+### RSpec with FactoryBot Setup
+
+```ruby
+# config/initializers/tng.rb
+Tng.configure do |config|
+  config.api_key = ENV["TNG_API_KEY"]
+  config.base_url = "https://app.tng.sh/"
+  
+  config.read_source_code = true
+  config.read_test_code = true
+  
+  config.testing_framework = "rspec"
+  config.assertion_style = "expect"
+  config.describe_style = true
+  config.context_style = "context"
+  config.it_style = "it"
+  config.before_style = "before"
+  config.after_style = "after"
+  config.let_style = true
+  config.subject_style = true
+  
+  config.mock_library = "rspec-mocks"
+  config.http_mock_library = "webmock"
+  config.factory_library = "factory_bot"
+  
+  config.authentication_enabled = true
+  config.authentication_methods = [
+    {
+      method: "authenticate_user!",
+      file_location: "app/controllers/application_controller.rb",
+      auth_type: "devise"
+    }
+  ]
+end
+```
+
+### API-Only Application with Token Authentication
+
+```ruby
+# config/initializers/tng.rb
+Tng.configure do |config|
+  config.api_key = ENV["TNG_API_KEY"]
+  config.base_url = "https://app.tng.sh/"
+  
+  config.read_source_code = true
+  config.read_test_code = true
+  
+  config.testing_framework = "minitest"
+  config.test_style = "test_block"
+  config.assertion_style = "assert/refute"
+  
+  config.mock_library = "minitest/mock"
+  config.http_mock_library = "webmock"
+  config.factory_library = "fixtures"
+  
+  config.authentication_enabled = true
+  config.authentication_methods = [
+    {
+      method: "authenticate_via_token!",
+      file_location: "app/controllers/application_controller.rb",
+      auth_type: "headers"
+    }
+  ]
+end
+```
+
+### Multiple Authentication Methods
+
+```ruby
+# config/initializers/tng.rb
+Tng.configure do |config|
+  config.api_key = ENV["TNG_API_KEY"]
+  config.base_url = "https://app.tng.sh/"
+  
+  config.read_source_code = true
+  config.read_test_code = true
+  
+  config.testing_framework = "rspec"
+  config.assertion_style = "expect"
+  config.let_style = true
+  
+  config.mock_library = "rspec-mocks"
+  config.http_mock_library = "webmock"
+  config.factory_library = "factory_bot"
+  
+  config.authentication_enabled = true
+  config.authentication_methods = [
+    {
+      method: "authenticate_user!",
+      file_location: "app/controllers/application_controller.rb",
+      auth_type: "session"
+    },
+    {
+      method: "authenticate_api_user!",
+      file_location: "app/controllers/api/base_controller.rb",
+      auth_type: "token_auth"
+    }
+  ]
+  
+  config.authorization_library = "pundit"
+end
+```
+
+## Advanced Configuration
+
+### Custom Authentication Setup
+
+For applications with custom authentication logic that doesn't fit standard patterns:
+
+```ruby
+config.authentication_methods = [
+  {
+    method: "verify_custom_auth!",
+    file_location: "app/controllers/concerns/custom_auth.rb",
+    auth_type: "custom"
+  }
+]
+```
+
+#### Rails API Application
+
+```ruby
+config.authentication_enabled = true
+config.authentication_methods = [
+  {
+    method: "authenticate_with_token",
+    file_location: "app/controllers/application_controller.rb",
+    auth_type: "jwt"
+  }
+]
+config.factory_library = "fixtures"  # Often simpler for API apps
+```
+
+#### Multi-Tenant Application
+
+```ruby
+config.authentication_methods = [
+  {
+    method: "authenticate_tenant!",
+    file_location: "app/controllers/application_controller.rb",
+    auth_type: "custom"
+  },
+  {
+    method: "authenticate_user!",
+    file_location: "app/controllers/application_controller.rb",
+    auth_type: "devise"
+  }
+]
+```
+
+#### Legacy Application with Mixed Authentication
+
+```ruby
+config.authentication_methods = [
+  {
+    method: "authenticate_admin!",
+    file_location: "app/controllers/admin/base_controller.rb",
+    auth_type: "basic_auth"
+  },
+  {
+    method: "authenticate_user!",
+    file_location: "app/controllers/application_controller.rb",
+    auth_type: "session"
+  },
+  {
+    method: "authenticate_api_key!",
+    file_location: "app/controllers/api/v1/base_controller.rb",
+    auth_type: "headers"
+  }
+]
+```
+
+## Usage Patterns
+
+### Interactive Mode
+
+The recommended way to use TNG for beginners and exploratory testing:
+
+```bash
+bundle exec tng
+```
+
+This launches an interactive session where you can:
+- Browse your application structure
+- Select files to generate tests for
+- Preview generated tests before saving
+- Adjust configuration on the fly
+
+### Direct Mode
+
+Generate tests directly from the command line without interactive prompts:
+
+#### Standard Format
+```bash
+bundle exec tng --type=controller --file=users_controller
+bundle exec tng --type=controller --file="admin/users_controller"
+bundle exec tng --type=model --file=user
+bundle exec tng --type=model --file="blog/post"
+```
+
+#### Short Aliases
+```bash
+# Super concise - perfect for quick generation
+bundle exec tng -t c -f ping
+bundle exec tng -t m -f user
+
+# Mixed format also works
+bundle exec tng --type=c --file=ping
+bundle exec tng -t controller -f users_controller
+```
+
+Use `bundle exec tng --help` for more options.
+
+### Debug Mode
+
+For troubleshooting configuration or generation issues:
+
+```bash
+# Interactive mode with debug on
+DEBUG=1 bundle exec tng
+
+# Direct mode with debug
+DEBUG=1 bundle exec tng -t c -f ping
+```
+
+### Help and Options
+
+```bash
+bundle exec tng --help              # Show help message
+bundle exec tng -h                  # Short help
+```
+
+## Troubleshooting
+
+### Configuration Issues
+
+#### API Key Problems
+
+**Error: "Invalid API key"**
+- Verify your API key from https://app.tng.sh/
+- Check environment variable: `echo $TNG_API_KEY`
+- Ensure no extra spaces in `.env` file: `TNG_API_KEY=your_key_here`
+
+**Error: "API key not found"**
+- Set environment variable: `export TNG_API_KEY=your_key`
+- Or add to `.env` file: `TNG_API_KEY=your_key`
+- Restart Rails server after changing environment variables
+
+#### Configuration Validation
+
+**Error: "Invalid configuration"**
+- Run configuration check: `rails console` then `Tng.configuration`
+- Verify all required settings are present
+- Check for typos in configuration file
+
+**Poor Test Quality**
+- Verify authentication methods are correctly configured
+- Check factory library setting matches your setup
+- Ensure `read_source_code` and `read_test_code` are enabled
+- Review testing framework settings match your project
+
+#### Authentication Configuration Issues
+
+**Tests don't include authentication**
+- Verify `authentication_enabled = true`
+- Check authentication methods are correctly specified
+- Ensure file paths exist: `ls app/controllers/application_controller.rb`
+- Verify method names exist in specified files
+
+**Authentication tests fail**
+- Check auth method names match your application
+- Verify file locations are correct
+- Ensure auth_type matches your authentication pattern
+
+### Generation Issues
+
+#### No Tests Generated
+
+**Check these common issues:**
+1. File path is correct: `ls app/controllers/users_controller.rb`
+2. File contains valid Ruby code
+3. Configuration is valid
+4. API key is working
+5. Internet connection is available
+
+#### Generated Tests Are Generic
+
+**Common causes:**
+- Authentication not properly configured
+- Factory library setting doesn't match your setup
+- `read_source_code` or `read_test_code` disabled
+- File analysis failed (check DEBUG mode)
+
+### Debug Mode
+
+Enable debug mode to get detailed information:
+
+```bash
+DEBUG=1 bundle exec tng -t c -f your_controller
+```
+
+Debug output includes:
+- API request/response information
+
+### Getting Help
+
+If you're still experiencing issues: Contact the gem maintainers or open a GitHub issue for support
+
+## License
+
+See [LICENSE.md](LICENSE.md) for details.

data/Rakefile

@@ -0,0 +1,124 @@
+# frozen_string_literal: true
+
+require "bundler/gem_tasks"
+require "minitest/test_task"
+
+Minitest::TestTask.create
+
+require "rubocop/rake_task"
+
+RuboCop::RakeTask.new
+
+require "rb_sys/extensiontask"
+
+task build: :compile
+
+GEMSPEC = Gem::Specification.load("tng.gemspec")
+
+RbSys::ExtensionTask.new("tng", GEMSPEC) do |ext|
+  ext.lib_dir = "lib/tng"
+end
+
+task default: %i[compile test rubocop]
+
+desc "Recompile Rust code and copy the binary for packaging"
+task redo: :compile do
+  puts "Copying compiled binary to binaries/ directory..."
+  binary_name = RUBY_PLATFORM.include?("darwin") ? "tng.bundle" : "tng.so"
+  source_file = File.join("lib", "tng", binary_name)
+  destination_dir = "binaries"
+
+  if File.exist?(source_file)
+    FileUtils.mkdir_p(destination_dir)
+    FileUtils.cp(source_file, File.join(destination_dir, binary_name))
+    puts "✅ Copied #{binary_name} to #{destination_dir}/"
+  else
+    puts "❌ Error: Compiled binary not found at #{source_file}"
+    puts "Make sure 'rake compile' ran successfully."
+  end
+end
+
+desc "Build binaries for both Mac and Linux platforms"
+task :build_both do
+  puts "Building tng native extensions for both platforms..."
+
+  # Clean up and create binaries directory
+  puts "Cleaning up old binaries..."
+  FileUtils.rm_rf("binaries")
+  puts "Creating binaries directory..."
+  FileUtils.mkdir_p("binaries")
+  puts "Binaries directory created: #{Dir.pwd}/binaries/"
+
+  # Build Mac version
+  puts "Building Mac version..."
+  system("bundle exec rake compile") || (puts "❌ Mac build failed"
+                                         exit 1)
+
+  # Copy Mac binary
+  mac_binary = "lib/tng/tng.bundle"
+  if File.exist?(mac_binary)
+    FileUtils.cp(mac_binary, "binaries/tng.bundle")
+    puts "Mac binary copied to binaries/ for packaging"
+  else
+    puts "❌ Mac binary not found"
+    exit 1
+  end
+
+  # Ensure it's in lib/tng/ for gem loading
+  if File.exist?("lib/tng/tng.bundle")
+    puts "Mac binary confirmed in lib/tng/ for gem loading"
+  else
+    puts "Warning: Mac binary not found in lib/tng/ - copying from binaries/"
+    FileUtils.cp("binaries/tng.bundle", "lib/tng/tng.bundle")
+  end
+
+  # Build Linux version
+  puts "Building Linux version..."
+  system("./build_linux_binaries.sh") || (puts "❌ Linux build failed"
+                                          exit 1)
+
+  puts "Build complete!"
+  puts "Binaries for packaging:"
+  system("ls -la binaries/")
+  puts "Binaries for gem loading:"
+  system("ls -la lib/tng/*.bundle lib/tng/*.so 2>/dev/null || echo 'No binaries found in lib/tng/'")
+end
+
+desc "Development build: compile for current OS and build gem"
+task dev: :redo do
+  puts "Building gem..."
+  system("gem build tng.gemspec") || (puts "❌ Gem build failed"
+                                      exit 1)
+  puts "✅ Development build complete!"
+end
+
+# Add this task to suppress the native artifact warning
+task release: :build_both do
+  # Temporarily rename the binary to avoid the warning during gem build
+  puts "📦 Building gem with pre-compiled binaries..." if File.exist?("binaries/tng.bundle")
+end
+
+desc "Clean release: build binaries and release gem"
+task :clean_release do
+  puts "🔨 Running build_both to ensure fresh binaries..."
+  Rake::Task[:build_both].invoke
+
+  puts "🧹 Cleaning development artifacts (keeping binaries/)..."
+  # Clean ext/ compiled files (development artifacts)
+  Dir.glob("ext/**/*.{bundle,so,dylib}").each do |file|
+    File.delete(file)
+    puts "Removed development artifact: #{file}"
+  end
+
+  # Clean lib/tng/ compiled files (development artifacts)
+  Dir.glob("lib/tng/*.{bundle,so,dylib}").each do |file|
+    File.delete(file)
+    puts "Removed development artifact: #{file}"
+  end
+
+  puts "📦 Binaries for distribution:"
+  system("ls -la binaries/")
+
+  puts "🚀 Releasing gem..."
+  Rake::Task["release"].invoke
+end

data/bin/load_dev

@@ -0,0 +1,22 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Development loader for TNG gem
+# This script loads the local development version of the gem
+# Usage: bin/load_dev
+
+puts "🔧 Loading TNG development environment..."
+
+# Add the lib directory to the load path
+lib_path = File.expand_path("../lib", __dir__)
+$LOAD_PATH.unshift(lib_path) unless $LOAD_PATH.include?(lib_path)
+
+# Load the development version
+require_relative "../lib/tng"
+
+puts "✅ TNG development version loaded!"
+puts "📍 Loaded from: #{lib_path}"
+
+# Start IRB with the gem loaded
+require "irb"
+IRB.start