unipod 0.1.0

data/lib/unipod/config.rb

@@ -0,0 +1,393 @@
+# frozen_string_literal: true
+
+require 'yaml'
+require 'fileutils'
+require 'securerandom'
+require 'json'
+
+module UniPod
+  class Config
+    DEFAULT_CONFIG_PATH = File.join(Dir.home, '.unipod', 'config.yml')
+    
+    # 定义在类外部，可以被类方法使用
+    def self.detect_default_cache_dir
+      # 统一使用 macOS 风格的缓存目录，避免使用 ~/.unipod/cache
+      if RUBY_PLATFORM =~ /darwin/
+        # Mac OS - 标准缓存目录
+        File.expand_path('~/Library/Caches/UniPod')
+      elsif RUBY_PLATFORM =~ /mswin|mingw|cygwin/
+        # Windows
+        win_cache_dir = ENV['LOCALAPPDATA'] ? File.join(ENV['LOCALAPPDATA'], 'UniPod', 'Cache') : File.expand_path('~/AppData/Local/UniPod/Cache')
+        File.expand_path(win_cache_dir)
+      else
+        # Linux/其他平台 - 遵循 XDG 规范
+        xdg_cache_dir = ENV['XDG_CACHE_HOME'] || File.expand_path('~/.cache')
+        File.expand_path(File.join(xdg_cache_dir, 'unipod'))
+      end
+    end
+    
+    # 从Git URL中提取仓库名称，此方法需要与CacheManager中的同名方法保持一致
+    def self.extract_repo_name_from_url(url)
+      if url.nil? || url.empty?
+        return 'default_repo'
+      end
+      
+      # 移除尾部的.git（如果有）
+      url = url.sub(/\.git\z/, '')
+      
+      # 处理不同格式的URL
+      if url.include?('/')
+        # 处理https://格式或git://格式
+        repo_name = url.split('/').last
+      elsif url.include?(':')
+        # 处理git@格式
+        repo_name = url.split(':').last.split('/').last
+      else
+        # 无法解析时使用默认名称
+        repo_name = 'default_repo'
+      end
+      
+      # 确保名称不包含特殊字符
+      repo_name.gsub(/[^a-zA-Z0-9_-]/, '_')
+    end
+    
+    DEFAULT_CONFIG = {
+      server: {
+        host: 'localhost',
+        port: 7748,
+        timeout: 120,  # 增加默认超时时间
+        max_clients: 10,  # 降低最大客户端数量
+        log_level: 'info',
+        session_id: SecureRandom.hex(8)  # 为每次启动生成唯一会话ID
+      },
+      cache: {
+        dir: detect_default_cache_dir,
+        max_size: 1024, # MB
+        expiry: 7 # days
+      },
+      repositories: {
+        index_repos: [
+          {
+            name: extract_repo_name_from_url('https://gitee.com/goodunitylib/JoyUnityLibIndex.git'),
+            url: 'https://gitee.com/goodunitylib/JoyUnityLibIndex.git',
+            enabled: true
+          }
+        ],
+        custom_repos: [],
+        scope_mappings: {}
+      },
+      unity: {
+        package_format: 'upm',  # 'upm' 或 'tgz'
+        preload_packages: true,  # 预加载包信息
+        compatibility: '2019.4'  # 最低兼容版本
+      }
+    }
+
+    class << self
+      def load_config
+        # 初始化默认配置
+        @config = DEFAULT_CONFIG.dup
+        
+        # 如果配置文件存在，尝试读取并合并
+        if File.exist?(config_path)
+          begin
+            user_config = YAML.load_file(config_path)
+            # 使用深度合并来确保用户配置覆盖默认配置，但保留未指定的默认值
+            deep_merge!(@config, user_config)
+          rescue => e
+            # 如果配置文件读取失败，使用默认配置，不终止程序
+            puts "注意: 配置文件读取失败，使用默认配置: #{e.message}" if ENV['UNIPOD_VERBOSE'] == 'true'
+          end
+        end
+        
+        # 检查环境变量中的配置覆盖
+        apply_env_overrides
+        
+        @config
+      end
+
+      # 从环境变量中读取配置
+      def apply_env_overrides
+        # 服务器配置
+        @config[:server][:host] = ENV['UNIPOD_SERVER_HOST'] if ENV['UNIPOD_SERVER_HOST']
+        @config[:server][:port] = ENV['UNIPOD_SERVER_PORT'].to_i if ENV['UNIPOD_SERVER_PORT']
+        
+        # 缓存配置
+        @config[:cache][:dir] = ENV['UNIPOD_CACHE_DIR'] if ENV['UNIPOD_CACHE_DIR']
+        
+        # 其他环境变量...
+      end
+
+      def save_config
+        # 确保配置目录存在
+        FileUtils.mkdir_p(File.dirname(config_path))
+        
+        # 保存配置到文件
+        File.open(config_path, 'w') do |f|
+          f.write(YAML.dump(@config))
+        end
+        
+        puts "配置已保存到: #{config_path}" if ENV['UNIPOD_VERBOSE'] == 'true'
+      end
+
+      def config_path
+        @config_path ||= ENV['UNIPOD_CONFIG'] || DEFAULT_CONFIG_PATH
+      end
+
+      def config
+        @config ||= load_config
+      end
+
+      def get(key, default = nil)
+        keys = key.to_s.split('.')
+        value = config
+        
+        keys.each do |k|
+          if value.is_a?(Hash) && value.key?(k.to_sym)
+            value = value[k.to_sym]
+          elsif value.is_a?(Hash) && value.key?(k)
+            value = value[k]
+          else
+            return default
+          end
+        end
+        
+        value
+      end
+
+      def set(key, value)
+        keys = key.to_s.split('.')
+        last_key = keys.pop
+        
+        # 导航到最后一个键的父节点
+        node = config
+        keys.each do |k|
+          sym_key = k.to_sym
+          str_key = k.to_s
+          
+          key_to_use = node.key?(sym_key) ? sym_key : (node.key?(str_key) ? str_key : sym_key)
+          
+          node[key_to_use] ||= {}
+          node = node[key_to_use]
+        end
+        
+        # 设置值
+        last_sym_key = last_key.to_sym
+        last_str_key = last_key.to_s
+        
+        key_to_use = node.key?(last_sym_key) ? last_sym_key : (node.key?(last_str_key) ? last_str_key : last_sym_key)
+        
+        node[key_to_use] = value
+        
+        # 保存配置 - 仅当用户显式调用save_config时才写入文件
+        
+        value
+      end
+
+      def server_host
+        get('server.host', 'localhost')
+      end
+
+      def server_port
+        get('server.port', 7748)
+      end
+      
+      def server_timeout
+        get('server.timeout', 120)
+      end
+      
+      def server_max_clients
+        get('server.max_clients', 10)
+      end
+      
+      def server_log_level
+        get('server.log_level', 'info')
+      end
+      
+      def server_session_id
+        get('server.session_id', SecureRandom.hex(8))
+      end
+
+      def cache_dir
+        # 确保使用默认缓存目录，而不是 ~/.unipod/cache
+        get('cache.dir', detect_default_cache_dir)
+      end
+
+      def cache_max_size
+        get('cache.max_size', 1024) # MB
+      end
+
+      def cache_expiry
+        get('cache.expiry', 7) # days
+      end
+
+      def index_repos
+        config = load_config
+        repos = config['index_repos'] || []
+        
+        # 查找Unity项目的manifest.json文件，加载其中的package_index_url
+        manifest_repos = load_unity_manifest_repos
+        
+        # 合并从manifest中读取的仓库信息
+        manifest_repos.each do |manifest_repo|
+          # 检查是否已存在同名仓库
+          existing_repo = repos.find { |r| r[:name] == manifest_repo[:name] }
+          if existing_repo
+            # 更新现有仓库信息
+            existing_repo[:package_index_url] = manifest_repo[:package_index_url] if manifest_repo[:package_index_url]
+          else
+            # 添加新仓库
+            repos << manifest_repo
+          end
+        end
+        
+        repos
+      end
+
+      def custom_repos
+        get('repositories.custom_repos', [])
+      end
+      
+      def unity_package_format
+        get('unity.package_format', 'upm')
+      end
+      
+      def unity_preload_packages
+        get('unity.preload_packages', true)
+      end
+      
+      def unity_compatibility
+        get('unity.compatibility', '2019.4')
+      end
+      
+      def add_index_repo(name, url, enabled = true)
+        repos = get('repositories.index_repos', [])
+        
+        # 检查是否已存在同名仓库
+        existing_repo = repos.find { |r| r[:name] == name }
+        if existing_repo
+          # 更新现有仓库信息
+          existing_repo[:url] = url
+          existing_repo[:enabled] = enabled
+        else
+          # 添加新仓库
+          repos << {
+            name: name,
+            url: url,
+            enabled: enabled
+          }
+        end
+        
+        set('repositories.index_repos', repos)
+      end
+      
+      def add_scope_mapping(scope, repo_name)
+        mappings = get('repositories.scope_mappings', {})
+        mappings[scope] = repo_name
+        set('repositories.scope_mappings', mappings)
+      end
+      
+      def get_repo_for_scope(scope)
+        mappings = get('repositories.scope_mappings', {})
+        mappings[scope]
+      end
+      
+      def remove_index_repo(name)
+        repos = index_repos.dup
+        repos.reject! { |r| r[:name] == name }
+        set('repositories.index_repos', repos)
+      end
+      
+      def add_custom_repo(name, url, type = 'git', enabled = true)
+        repos = custom_repos.dup
+        
+        # 检查是否已存在
+        existing = repos.find { |r| r[:name] == name }
+        if existing
+          existing[:url] = url
+          existing[:type] = type
+          existing[:enabled] = enabled
+        else
+          repos << { name: name, url: url, type: type, enabled: enabled }
+        end
+        
+        set('repositories.custom_repos', repos)
+      end
+      
+      # 从Unity的manifest.json文件加载仓库信息
+      def load_unity_manifest_repos
+        repos = []
+        
+        # 查找当前目录及其父目录中的manifest.json文件
+        manifest_path = find_unity_manifest
+        return repos unless manifest_path
+        
+        begin
+          manifest = JSON.parse(File.read(manifest_path))
+          
+          # 检查是否有scopedRegistries配置
+          if manifest['scopedRegistries'] && manifest['scopedRegistries'].is_a?(Array)
+            manifest['scopedRegistries'].each do |registry|
+              next unless registry['name'] && registry['url']
+              
+              repo = {
+                name: registry['name'].downcase.gsub(/[^a-z0-9_-]/, '_'),
+                url: registry['url'],
+                package_index_url: registry['package_index_url'] || registry['index_url']  # 支持两种字段名称
+              }
+              
+              # 如果有package_index_url或index_url，才添加仓库
+              if repo[:package_index_url]
+                repos << repo
+              end
+            end
+          end
+        rescue => e
+          # 忽略解析错误
+          puts "无法解析Unity的manifest.json文件: #{e.message}" if ENV['UNIPOD_VERBOSE'] == 'true'
+        end
+        
+        repos
+      end
+      
+      # 在当前目录及其父目录中查找Unity的manifest.json文件
+      def find_unity_manifest
+        current_dir = Dir.pwd
+        max_depth = 10  # 限制向上查找的层数
+        
+        depth = 0
+        while depth < max_depth
+          # 检查当前目录是否有manifest.json文件
+          manifest_path = File.join(current_dir, 'Packages', 'manifest.json')
+          return manifest_path if File.exist?(manifest_path)
+          
+          # 检查是否已到达根目录
+          parent_dir = File.dirname(current_dir)
+          break if parent_dir == current_dir  # 已到达根目录
+          
+          current_dir = parent_dir
+          depth += 1
+        end
+        
+        nil  # 未找到manifest.json文件
+      end
+      
+      private
+      
+      # 深度合并哈希
+      def deep_merge!(target, source)
+        source.each do |key, value|
+          target_value = target[key]
+          
+          if target_value.is_a?(Hash) && value.is_a?(Hash)
+            deep_merge!(target_value, value)
+          else
+            target[key] = value
+          end
+        end
+        
+        target
+      end
+    end
+  end
+end