baid 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 83bacc2e8f9264fa7203aa6b9c9d1ded8a18974c63c72fbd3ef4f0e0ae6f7a76
-  data.tar.gz: '094836c43eab0251dc04fc8738b3b885b57cda69edb7197704c123a40d0450c1'
+  metadata.gz: 221fba49d1c1620e0e5423517dc5b0c63f1d6d71898377b3df90671ae5732a2a
+  data.tar.gz: 499df0b58c473f06738f1e725e5c9b023a7b1b481c3192fbdc6494e2c24e5626
 SHA512:
-  metadata.gz: 2f39c86e23698bd6883c23521050f745ffbea60547c235a5d0bcb7083f575b8ef77361faac640169354403fdfbdcb77071352ce5330773b892faccc355420d68
-  data.tar.gz: b392ac3a82514828f91f3eb59cc77699b8c3842cbdefca33969c9763ac528180e3396037420c390d50915936af9fa6f46c59fcdf9b47a0d7105854206b838111
+  metadata.gz: 62e89d04b3c8cbb5ed97a4a4b20bf16b3044cddf4eb37863304fcd7222a7973b3987465fa69a18ae84f40b7e205cb2d69ad41dfc25022657629b12edd2eea243
+  data.tar.gz: bea46903bc722717c6ca60506a13d2124c58edd277e16f0e390fce4f282c6783904557b8b9cb20d427426a91a9e54aa2838382958677b3e3c86b38c30e162f64

data/lib/baid/commands/init.rb

@@ -20,13 +20,85 @@ module Baid
           puts "Configured MCP for #{agent[:name]}"
         end
 
+        init_result = smart_init(agents)
+
         config_path = File.join(@project_dir, ".baid", "config.yml")
         FileUtils.mkdir_p(File.dirname(config_path))
-        config = { "agents" => agents.map { |a| a[:name] }, "installed_skills" => [] }
+        config = {
+          "agents" => agents.map { |a| a[:name] },
+          "installed_skills" => init_result[:skills]
+        }
+        config["tech_stack"] = init_result[:tech_stack] if init_result[:tech_stack]&.any?
         File.write(config_path, config.to_yaml)
 
         puts "Baid initialized with #{agents.length} agent(s)."
       end
+
+      private
+
+      def smart_init(agents)
+        token = Config.load_token
+        workspace = Config.load_active_workspace
+        return { skills: [], tech_stack: [] } unless token && workspace
+
+        puts "Analyzing project..."
+        scan = ProjectScanner.scan(@project_dir)
+
+        client = ApiClient.new(token: token)
+        response = client.post("/init/analyze", {
+          workspace_slug: workspace,
+          file_tree: scan[:file_tree],
+          package_files: scan[:package_files]
+        })
+
+        return { skills: [], tech_stack: [] } unless response.code == 200
+
+        data = JSON.parse(response.body)
+        tech_stack = data["tech_stack"] || []
+        answers = present_questions(data["questions"] || [])
+
+        setup_response = client.post("/init/setup", {
+          workspace_slug: workspace,
+          tech_stack: tech_stack,
+          answers: answers,
+          agents: agents.map { |a| a[:name] },
+          recommended_skills: data["recommended_skills"] || []
+        })
+
+        return { skills: [], tech_stack: tech_stack } unless setup_response.code == 200
+
+        setup_data = JSON.parse(setup_response.body)
+        skills = setup_data["skills"] || []
+
+        skills.each do |skill|
+          SkillWriter.write(skill, agents, @project_dir)
+          puts "Installed skill: #{skill["name"]}"
+        end
+
+        { skills: skills.map { |s| s["name"] }, tech_stack: tech_stack }
+      rescue StandardError
+        { skills: [], tech_stack: [] }
+      end
+
+      def present_questions(questions)
+        return {} if questions.empty?
+
+        prompt = TTY::Prompt.new
+        answers = {}
+
+        questions.each do |q|
+          case q["type"]
+          when "confirm"
+            answers[q["id"]] = prompt.yes?(q["text"], default: q["default"] != false)
+          when "select"
+            answers[q["id"]] = prompt.select(q["text"], q["options"] || [])
+          when "multi_select"
+            answers[q["id"]] = prompt.multi_select(q["text"], q["options"] || [], default: q["defaults"] || [])
+          end
+        end
+
+        answers
+      end
     end
   end
 end

data/lib/baid/commands/skills.rb

@@ -5,28 +5,31 @@ module Baid
     class Skills
       def list(workspace_slug = nil)
         workspace_slug = resolve_workspace(workspace_slug)
-        return unless workspace_slug
 
-        client = ApiClient.new
-        response = client.get("/workspaces/#{workspace_slug}/skills")
+        workspace_skills = []
+        if workspace_slug
+          client = ApiClient.new
+          response = client.get("/workspaces/#{workspace_slug}/skills")
 
-        unless response.code == 200
-          puts "Failed to fetch skills (#{response.code}). Check your token with `baid whoami`."
-          return
+          if response.code == 200
+            data = JSON.parse(response.body)
+            workspace_skills = data["skills"] || []
+          end
         end
 
-        data = JSON.parse(response.body)
-        skills = data["skills"] || []
+        skills_sh = SkillsShScraper.new.fetch
+
+        all_skills = workspace_skills + skills_sh
 
-        if skills.empty?
-          puts "No skills in workspace."
+        if all_skills.empty?
+          puts "No skills found."
           return
         end
 
-        puts format("%-25s %-15s %s", "NAME", "SOURCE", "VERSION")
-        puts "-" * 50
-        skills.each do |skill|
-          puts format("%-25s %-15s %s", skill["name"], skill["source"], skill["latest_version"])
+        puts format("%-35s %-15s %s", "NAME", "SOURCE", "VERSION/REPO")
+        puts "-" * 70
+        all_skills.each do |skill|
+          puts format("%-35s %-15s %s", skill["name"], skill["source"], skill["latest_version"])
         end
       end
 

data/lib/baid/mcp_configurator.rb

@@ -2,11 +2,21 @@
 
 module Baid
   class McpConfigurator
-    BAID_MCP_ENTRY = {
-      "command" => "npx",
-      "args" => ["-y", "@baid/mcp-server"],
-      "env" => {}
-    }.freeze
+    def self.build_mcp_entry
+      token = Config.load_token
+      workspace = Config.load_active_workspace
+      api_url = ENV.fetch("BAID_API_URL", "https://baid.dev")
+
+      env = { "BAID_API_URL" => api_url }
+      env["BAID_TOKEN"] = token if token
+      env["BAID_WORKSPACE"] = workspace if workspace
+
+      {
+        "command" => "npx",
+        "args" => ["-y", "@baid/mcp-server"],
+        "env" => env
+      }
+    end
 
     def self.configure(agent, project_dir = ".")
       config_path = File.join(project_dir, agent[:config_path])
@@ -14,7 +24,7 @@ module Baid
 
       config = load_config(config_path)
       config["mcpServers"] ||= {}
-      config["mcpServers"]["baid"] = BAID_MCP_ENTRY.dup
+      config["mcpServers"]["baid"] = build_mcp_entry
 
       File.write(config_path, JSON.pretty_generate(config))
     end

data/lib/baid/project_scanner.rb

@@ -0,0 +1,63 @@
+# frozen_string_literal: true
+
+module Baid
+  class ProjectScanner
+    PACKAGE_FILES = %w[
+      Gemfile
+      package.json
+      requirements.txt
+      pom.xml
+      go.mod
+      Cargo.toml
+      composer.json
+      pubspec.yaml
+      build.gradle
+      pyproject.toml
+    ].freeze
+
+    IGNORED = %w[. .. node_modules vendor .git .svn .hg __pycache__ .DS_Store].freeze
+
+    MAX_PACKAGE_SIZE = 10_000
+
+    def self.scan(project_dir)
+      file_tree = collect_tree(project_dir, depth: 0)
+      package_files = collect_packages(project_dir)
+
+      { file_tree: file_tree, package_files: package_files }
+    end
+
+    def self.collect_tree(dir, depth:)
+      entries = []
+      Dir.children(dir).sort.each do |name|
+        next if IGNORED.include?(name) || name.start_with?(".")
+
+        full = File.join(dir, name)
+        if File.directory?(full)
+          entries << "#{name}/"
+          if depth < 1
+            collect_tree(full, depth: depth + 1).each do |sub|
+              entries << "#{name}/#{sub}"
+            end
+          end
+        else
+          entries << name
+        end
+      end
+      entries
+    end
+
+    def self.collect_packages(project_dir)
+      result = {}
+      PACKAGE_FILES.each do |filename|
+        path = File.join(project_dir, filename)
+        next unless File.exist?(path)
+
+        content = File.read(path)
+        result[filename] = content.length > MAX_PACKAGE_SIZE ? content[0...MAX_PACKAGE_SIZE] : content
+      end
+      result
+    end
+
+    private_class_method :collect_tree, :collect_packages
+  end
+end

data/lib/baid/skills_sh_scraper.rb

@@ -0,0 +1,38 @@
+# frozen_string_literal: true
+
+module Baid
+  class SkillsShScraper
+    SKILLS_SH_URL = "https://skills.sh/"
+    MAX_SKILLS = 50
+
+    def fetch
+      response = HTTParty.get(SKILLS_SH_URL, timeout: 10)
+      return [] unless response.code == 200
+
+      parse_skills(response.body)
+    rescue StandardError
+      []
+    end
+
+    private
+
+    def parse_skills(html)
+      # skills.sh embeds skill data as JSON objects in the Next.js RSC payload
+      # The payload uses backslash-escaped quotes, so unescape before matching
+      unescaped = html.gsub('\"', '"')
+
+      skills = []
+
+      unescaped.scan(/"source":"([^"]+)","skillId":"([^"]+)","name":"([^"]+)","installs":(\d+)/) do |source, skill_id, name, installs|
+        skills << {
+          "name" => name,
+          "source" => "skills.sh",
+          "latest_version" => source,
+          "installs" => installs.to_i
+        }
+      end
+
+      skills.first(MAX_SKILLS)
+    end
+  end
+end

data/lib/baid.rb

@@ -5,6 +5,7 @@ require "httparty"
 require "json"
 require "yaml"
 require "fileutils"
+require "tty-prompt"
 
 module Baid
   VERSION = "0.1.0"
@@ -14,6 +15,7 @@ require_relative "baid/config"
 require_relative "baid/api_client"
 require_relative "baid/agent_detector"
 require_relative "baid/mcp_configurator"
+require_relative "baid/project_scanner"
 require_relative "baid/commands/login"
 require_relative "baid/commands/logout"
 require_relative "baid/commands/whoami"
@@ -23,6 +25,7 @@ require_relative "baid/commands/reconfigure"
 require_relative "baid/commands/install"
 require_relative "baid/commands/search"
 require_relative "baid/commands/update"
+require_relative "baid/skills_sh_scraper"
 require_relative "baid/commands/skills"
 require_relative "baid/commands/workspace"
 require_relative "baid/cli"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: baid
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Baid Team
@@ -115,7 +115,9 @@ files:
 - lib/baid/commands/workspace.rb
 - lib/baid/config.rb
 - lib/baid/mcp_configurator.rb
+- lib/baid/project_scanner.rb
 - lib/baid/skill_writer.rb
+- lib/baid/skills_sh_scraper.rb
 homepage: https://baid.dev
 licenses:
 - MIT