lg_pod_plugin 1.1.3 → 1.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3edf15fe0a8e9cfcc59a9224cd55aa947dc4c347f3a2dedbbdab3bb07b4b1f65
-  data.tar.gz: '09cbcc226c7176f819d23b950adfaaab4d4cefa918403edac8b2cd8eec9e0cef'
+  metadata.gz: 10e8d5a2e4c3e10541b118c8e93fcb603cca17298f5553191a83f32f59ecbe20
+  data.tar.gz: b4241f999d846e0e51a586333d541f9f17c759f71fc7b7c7ccb3065a822b5b89
 SHA512:
-  metadata.gz: e422b84be0ca83287c305152a8543a195ca27fa742937d81113f30a0e697d97865b114ca71222e5280eb62c196985e2da333a71f07e2b718bf71515b497ff2e5
-  data.tar.gz: 686b7cb744ed6857dee9317bb74eaa3841fd34e75f388be3912ca2ea1dd7ac2eeccbce9fdec1ac563e28ffd34b59f478a68f9cf9d0aa48d2d1517c776e588fc8
+  metadata.gz: 6839fc1d93fc1eb254b0f1a6017438494a0c8d9ece898cbb4ff50d9ad4077e41df23739d80911cbf31f6e181da873eb13a1b09fa79152f7f8500968c0ed1c7f2
+  data.tar.gz: 62271e786fb952bdd95dc2812d0314f54f2b462927dfbae5c8fa1443f0a0d915bf8945c4633ce53926a8fcb6abe7abffc70db99357f5f76cf485a7b405c6d728

data/bin/lg

@@ -1,5 +1,5 @@
 #!/usr/bin/env ruby
 require 'cocoapods'
 require 'lg_pod_plugin'
-LgPodPlugin.log_green "欢迎使用 `lg_pod_plugin`"
-LgPodPlugin::Command.run(ARGV)
+LgPodPlugin.log_blue "欢迎使用 `lg_pod_plugin`"
+LgPodPlugin::Command.run(ARGV)

data/lib/command/command.rb

@@ -6,7 +6,6 @@ module LgPodPlugin
     require_relative 'update'
     require_relative 'init'
     require_relative 'install'
-    require_relative '../lg_pod_plugin/gitlab_api.rb'
     self.command = 'lg'
     self.version = VERSION
     self.abstract_command = true
@@ -26,4 +25,4 @@ module LgPodPlugin
     end
 
   end
-end
+end

data/lib/command/install.rb

@@ -45,8 +45,13 @@ module LgPodPlugin
       end
 
       def run
-        LgPodPlugin::Main.run("install",{ :verbose => self.log, :repo_update => self.repo_update })
+
+        begin_time = Time.now.to_i
+        LgPodPlugin::Main.run("install", { :verbose => self.log, :repo_update => self.repo_update })
+        end_time = Time.now.to_i
+        LgPodPlugin.log_blue "`lg install`安装所需时间: #{end_time - begin_time}"
       end
+
     end
   end
-end
+end

data/lib/command/update.rb

@@ -32,7 +32,10 @@ module LgPodPlugin
       end
 
       def run
-        LgPodPlugin::Main.run("update",{ :verbose => self.log, :repo_update => self.repo_update })
+        begin_time = Time.now.to_i
+        LgPodPlugin::Main.run("update", { :verbose => self.log, :repo_update => self.repo_update})
+        end_time = Time.now.to_i
+        LgPodPlugin.log_blue "`lg update`安装所需时间: #{end_time - begin_time}"
       end
     end
   end

data/lib/lg_pod_plugin/config/l_config.rb

@@ -0,0 +1,59 @@
+require 'json'
+require 'uri'
+require 'io/console'
+
+module LgPodPlugin
+
+  class LConfig
+    attr_accessor :host
+    attr_accessor :base_url
+    attr_accessor :project_name
+    attr_accessor :access_token
+    attr_accessor :refresh_token
+    attr_accessor :project
+
+    def initialize
+    end
+
+    public
+    def self.get_config(git, uri)
+      return nil unless uri&.host
+      return nil unless LUtils.is_gitlab_uri(git, uri.hostname)
+      user_id = LUserAuthInfo.get_user_id(uri.hostname)
+      user_info = LSqliteDb.shared.query_user_info(user_id)
+      # 用户授权 token 不存在, 提示用户输入用户名密码
+      unless user_info
+        user_info = GitLabAPI.get_gitlab_access_token_input(uri, user_id, nil, nil)
+        return nil unless user_info
+      end
+      time_now = Time.now.to_i
+      # 判断 token 是否失效
+      if user_info.expires_in <= time_now
+        # 刷新 token 失败时, 通过已经保存的用户名密码来刷新 token
+        new_user_info = GitLabAPI.refresh_gitlab_access_token uri.hostname, user_info.refresh_token
+        if new_user_info.nil?
+          username = user_info.username
+          password = user_info.password
+          user_info = GitLabAPI.get_gitlab_access_token_input(uri, user_id, username, password)
+          return nil unless user_info
+        else
+          user_info = new_user_info
+        end
+      end
+
+      config = LConfig.new
+      config.host = uri.hostname
+      config.access_token = user_info.access_token
+      config.refresh_token = user_info.refresh_token
+      config.base_url = LUtils.get_gitlab_base_url(git)
+      config.project_name = LUtils.get_git_project_name(git)
+      config.project = LSqliteDb.shared.query_project_info(config.project_name, git)
+      unless config.project
+        config.project = GitLabAPI.request_project_info(config.host, config.project_name, config.access_token, git)
+      end
+      return config
+
+    end
+
+  end
+end

data/lib/lg_pod_plugin/config/lockfile_model.rb

@@ -0,0 +1,59 @@
+require 'cocoapods-core'
+
+module LgPodPlugin
+
+  class LockfileModel
+    attr_accessor :lockfile
+    attr_accessor :release_pods
+    attr_accessor :external_sources_data
+    attr_accessor :checkout_options_data
+
+    def initialize
+    end
+
+    def self.from_file
+      lockfile = LProject.shared.lockfile
+      unless lockfile
+        lockfile_model = LockfileModel.new
+        lockfile_model.lockfile = nil
+        lockfile_model.release_pods = {}
+        lockfile_model.checkout_options_data = {}
+        lockfile_model.external_sources_data = {}
+        return lockfile_model
+      end
+      release_pods = Hash.new
+      pods = lockfile.send(:generate_pod_names_and_versions)
+      pods.each do |element|
+        if LUtils.is_a_string?(element) || element.is_a?(Hash)
+          key = element.is_a?(Hash) ? element.keys.first : element
+          next unless key
+          if key.include?(" ")
+            pod_name = LUtils.pod_real_name(key.split(" ").first)
+          else
+            pod_name = key
+          end
+          tag = key[/(?<=\().*?(?=\))/]
+          release_pods[pod_name] = tag
+        else
+          next
+        end
+      end
+      lockfile_model = LockfileModel.new
+      lockfile_model.lockfile = lockfile
+      lockfile_model.release_pods = release_pods
+      lockfile_model.checkout_options_data = lockfile.send(:checkout_options_data)
+      lockfile_model.checkout_options_data = {} unless lockfile_model.checkout_options_data
+      lockfile_model.external_sources_data = lockfile.send(:external_sources_data)
+      lockfile_model.external_sources_data = {} unless lockfile_model.external_sources_data
+      lockfile_model
+    end
+
+    def checkout_options_for_pod_named(name)
+      return {} unless @lockfile
+      hash = @lockfile.checkout_options_for_pod_named(name)
+      hash ? hash : {}
+    end
+
+
+  end
+end

data/lib/lg_pod_plugin/config/podspec.rb

@@ -0,0 +1,266 @@
+require 'cocoapods'
+require 'cocoapods-core'
+
+module LgPodPlugin
+
+  class PodSpec
+    attr_reader :spec
+    attr_accessor :json_files
+    attr_accessor :source_files
+
+    public
+    def self.form_file(path)
+      begin
+        spec = Pod::Specification.from_file(path)
+        return PodSpec.new(spec)
+      rescue
+        return nil
+      end
+    end
+
+    public
+    def prepare_command
+      json = self.to_pretty_json
+      return nil unless json
+      hash = JSON.parse json
+      hash["prepare_command"]
+    end
+
+    public
+    def to_pretty_json(*a)
+      require 'json'
+      JSON.pretty_generate(self.spec.to_hash, *a) << "\n"
+    end
+
+
+    public
+    def self.form_string(string, path)
+      begin
+        #Work around for Rubinius incomplete encoding in 1.9 mode
+        if string.respond_to?(:encoding) && string.encoding.name != 'UTF-8'
+          string = string.force_encoding("gb2312").force_encoding("utf-8")
+        end
+        spec = Pod::Specification.from_string string, path
+        return PodSpec.new(spec)
+      rescue
+        return nil
+      end
+    end
+
+    def self.form_pod_spec(spec)
+      return PodSpec.new(spec)
+    end
+
+    def self.form_json(object)
+      begin
+        spec = Pod::Specification.from_json(object)
+        return PodSpec.new(spec)
+      rescue
+        return nil
+      end
+    end
+
+    def initialize(spec)
+      if spec == nil
+        @spec = nil
+        @json_files = Hash.new
+        @source_files = Hash.new
+        return
+      end
+      set = Set[]
+      @spec = spec
+      attributes_hash = spec.send(:attributes_hash)
+      return unless attributes_hash.is_a?(Hash)
+      license = attributes_hash["license"]
+      if license.is_a?(Hash)
+        license_file = license["file"]
+        set.add(license_file) if license_file
+      else
+        set.add("LICENSE")
+      end
+      # 解析主模块依赖信息
+      set.merge(parse_subspec_with(attributes_hash))
+      subspecs = spec.subspecs
+      unless subspecs.is_a?(Array)
+        self.source_files = parse_with_set set
+        if self.source_files.has_key?("*")
+          self.source_files.delete("*")
+          self.source_files["All"] = "All"
+        end
+        self.json_files = spec.to_pretty_json
+        return
+      end
+      subspecs.each do |sub_spec|
+        sub_attributes_hash = sub_spec.send(:attributes_hash)
+        next unless sub_attributes_hash && sub_attributes_hash.is_a?(Hash)
+        sub_set = self.parse_subspec_with(sub_attributes_hash)
+        next if sub_set.empty?
+        set.merge(sub_set)
+      end
+      self.source_files = parse_with_set set
+      if self.source_files.has_key?("*")
+        self.source_files.delete("*")
+        self.source_files["All"] = "All"
+      end
+      self.json_files = spec.to_pretty_json
+    end
+
+    public
+    def write_to_file(path)
+      LCache.write_spec(@spec, path)
+    end
+
+    def parse_with_set(set)
+      new_hash = Hash.new
+      set.each do |element|
+        if element.start_with?("**/")
+          new_hash["All"] = "All"
+        elsif element.include?("/")
+          str = element.split("/", 0).first
+          new_hash[str] = str unless str.empty?
+        elsif element.start_with?("**.") || element.start_with?("*.")
+          new_hash["All"] = "All"
+        elsif element.start_with?("*.framework") || element.include?("*.a")
+          new_hash["All"] = "All"
+        else
+          new_hash[element] = element unless element.empty?
+        end
+      end
+      new_hash
+    end
+
+    # 公共解析解析subspec
+    def parse_subspec_with(hash)
+      set = Set[]
+      source_files = self.parse_source_files(hash["source_files"])
+      set.merge(source_files) unless source_files.empty?
+      resources = self.parse_resource_files(hash["resource"] ||= hash["resources"])
+      set.merge(resources) unless resources.empty?
+      resource_bundles = self.parse_resource_bundles(hash["resource_bundle"] ||= hash["resource_bundles"])
+      set.merge(resource_bundles) unless resource_bundles.empty?
+      project_header_files = self.parse_project_header_files(hash["project_header_files"])
+      set.merge(resource_bundles) unless project_header_files.empty?
+      private_header_files = self.parse_private_header_files(hash["private_header_files"])
+      set.merge(private_header_files) unless private_header_files.empty?
+      vendored_frameworks = self.parse_vendored_frameworks(hash["vendored_frameworks"])
+      set.merge(vendored_frameworks) unless vendored_frameworks.empty?
+      vendored_library = self.parse_vendored_library(hash["vendored_library"] ||= hash["vendored_libraries"])
+      set.merge(vendored_library) unless vendored_library.empty?
+      #parse_preserve_path
+      preserve_paths = self.parse_preserve_path(hash["preserve_path"] ||= hash["preserve_paths"])
+      set.merge(preserve_paths) unless preserve_paths.empty?
+      module_map = self.parse_module_map(hash["module_map"])
+      set.merge(module_map) unless module_map.empty?
+      ios_hash = hash["ios"]
+      if ios_hash && ios_hash.is_a?(Hash)
+
+        ios_source_files = self.parse_source_files(ios_hash["source_files"])
+        set.merge(ios_source_files) unless ios_source_files.empty?
+
+        ios_resources = self.parse_resource_files(ios_hash["resource"] ||= ios_hash["resources"])
+        set.merge(ios_resources) unless ios_resources.empty?
+
+        ios_resource_bundles = self.parse_resource_bundles(ios_hash["resource_bundle"] ||= ios_hash["resource_bundles"])
+        set.merge(ios_resource_bundles) unless ios_resource_bundles.empty?
+
+        ios_project_header_files = self.parse_project_header_files(ios_hash["project_header_files"])
+        set.merge(ios_project_header_files) unless ios_project_header_files.empty?
+
+        ios_private_header_files = self.parse_private_header_files(ios_hash["private_header_files"])
+        set.merge(ios_private_header_files) unless ios_private_header_files.empty?
+
+        ios_vendored_frameworks = self.parse_vendored_frameworks(ios_hash["vendored_frameworks"])
+        set.merge(ios_vendored_frameworks) unless ios_vendored_frameworks.empty?
+
+        ios_vendored_library = self.parse_vendored_library(ios_hash["vendored_library"] ||= ios_hash["vendored_libraries"])
+        set.merge(ios_vendored_library) unless ios_vendored_library.empty?
+
+        ios_preserve_paths = self.parse_preserve_path(ios_hash["preserve_path"] ||= ios_hash["preserve_paths"])
+        set.merge(ios_preserve_paths) unless ios_preserve_paths.empty?
+
+        ios_module_map = self.parse_module_map(ios_hash["module_map"])
+        set.merge(ios_module_map) unless ios_module_map.empty?
+
+      end
+      set
+    end
+
+    # 公共解析文件路径的方法
+    def parse_public_source_files(source_files)
+      return Array.new unless source_files
+      array = Array.new
+      if LUtils.is_a_string?(source_files)
+        array.append(source_files) unless source_files.empty?
+      elsif source_files.is_a?(Array)
+        source_files.each do |element|
+          next unless LUtils.is_a_string?(element)
+          array.append(element) unless element.empty?
+        end
+      elsif source_files.is_a?(Hash)
+        source_files.each do |_, val|
+          if LUtils.is_a_string?(val)
+            array.append(val) unless val.empty?
+          elsif val.is_a?(Array)
+            val.each do |element|
+              next unless LUtils.is_a_string?(element)
+              array.append(element) unless element.empty?
+            end
+          end
+        end
+      end
+      array
+    end
+
+    # 解析source_files路径
+    def parse_source_files(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 resource所在路径
+    def parse_resource_files(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析public_header_files字段的值
+    def parse_public_header_files(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 parse_resource_bundles
+    def parse_resource_bundles(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 project_header_files
+    def parse_project_header_files(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 private_header_files
+    def parse_private_header_files(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 vendored_frameworks
+    def parse_vendored_frameworks(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 parse_vendored_library
+    def parse_vendored_library(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 parse_preserve_path
+    def parse_preserve_path(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+    # 解析 module_map
+    def parse_module_map(source_files)
+      self.parse_public_source_files(source_files)
+    end
+
+  end
+
+end

data/lib/lg_pod_plugin/config/target.rb

@@ -0,0 +1,33 @@
+require 'cocoapods'
+
+module LgPodPlugin
+  class LPodTarget
+    attr_reader :name
+    attr_reader :dependencies
+    def initialize(target)
+      internal_hash = target.send(:internal_hash)
+      @name = internal_hash["name"]
+      array = Array.new(internal_hash['dependencies'] ||= [])
+      dependencies = array.reject do |e|
+        if LUtils.is_a_string?(e)
+          true
+        elsif e.is_a?(Hash)
+          key = e.keys.last ||= ""
+          val = e[key].last ||= ""
+          !val.is_a?(Hash)
+        else
+          true
+        end
+      end
+      external_pods = Hash.new
+      dependencies.each do |e|
+        key = e.keys.last ||= ""
+        val = e[key].last ||= {}
+        next unless val.is_a?(Hash)
+        pod = ExternalPod.new(self, key, val)
+        external_pods[pod.name] = pod
+      end
+      @dependencies = external_pods
+    end
+  end
+end

data/lib/lg_pod_plugin/{database.rb → db/database.rb}

@@ -1,7 +1,10 @@
-require "sqlite3"
 require 'singleton'
-require_relative 'l_cache'
-
+arch = %x(ruby -v)
+if arch.include?("arm64")
+  require_relative '../../sqlite3-1.5.3-arm64-darwin/lib/sqlite3'
+else
+  require_relative '../../sqlite3-1.5.3-x86_64-darwin/lib/sqlite3'
+end
 module LgPodPlugin
 
   class ProjectModel
@@ -70,11 +73,12 @@ module LgPodPlugin
 
   class LSqliteDb
     include Singleton
-    REQUIRED_ATTRS ||= %i[db ].freeze
+    REQUIRED_ATTRS ||= %i[db].freeze
     attr_accessor(*REQUIRED_ATTRS)
     K_USER_TABLE = "user_tab"
     K_USER_PROJECTS = "user_projects"
     K_POD_LATEST_REFS = "user_pod_latest_refs"
+    K_POD_SHA_BRANCH = "user_pod_sha_with_branch"
 
     def self.shared
       return LSqliteDb.instance
@@ -82,9 +86,13 @@ module LgPodPlugin
 
     # 初始化 db
     def initialize
+    end
+
+    #初始化database
+    def init_database
       root_path = LFileManager.download_director.join("database")
       db_file_path = root_path.join("my.db")
-      if !root_path.exist? || !db_file_path.exist?
+      unless root_path.exist? && db_file_path.exist?
         FileUtils.mkdir(root_path)
         FileUtils.chdir(root_path)
         FileUtils.touch("my.db")
@@ -127,11 +135,18 @@ module LgPodPlugin
         );"
       self.db.execute(sql3)
 
-      super
+      #添加项目表
+      sql4 = "create table if not exists #{K_POD_SHA_BRANCH}(
+        id varchar(100) primary key not null,
+        name varchar(100),
+        git varchar(100),
+	      branch varchar(100),
+        sha varchar(100)
+        );"
+      self.db.execute(sql4)
     end
 
     public
-
     def insert_user_info(user)
       # pp "user.id => #{user.id}"
       if self.query_user_info(user.id) != nil
@@ -146,6 +161,7 @@ module LgPodPlugin
     end
 
     public
+
     def query_user_info(user_id)
       user_info = nil
       self.db.execute("select * from #{K_USER_TABLE} where id = '#{user_id}';") do |row|
@@ -161,11 +177,17 @@ module LgPodPlugin
       user_info
     end
 
+    #删除用户信息
+    def delete_user_info(id)
+      ps = @db.prepare("DELETE FROM #{K_USER_TABLE} WHERE id = :n")
+      ps.execute('n' => id)
+    end
+
     # 保存项目信息到数据库
     def insert_project(project)
       if self.query_project_info(project.name, project.http_url_to_repo) != nil
         self.db.execute_batch(
-          "UPDATE #{K_USER_PROJECTS} SET name = (:name), desc = (:desc), path = (:path), ssh_url_to_repo = (:ssh_url_to_repo), http_url_to_repo = (:http_url_to_repo), web_url = (:web_url), name_with_namespace = (:name_with_namespace), path_with_namespace = (:path_with_namespace) where (id = :id)", { "name" => project.name, "desc" => project.description, "path" => project.path, "ssh_url_to_repo" => project.ssh_url_to_repo, :http_url_to_repo => project.http_url_to_repo, :web_url => project.web_url, :id => project.id , :path_with_namespace => project.path_with_namespace, :name_with_namespace => project.name_with_namespace}
+          "UPDATE #{K_USER_PROJECTS} SET name = (:name), desc = (:desc), path = (:path), ssh_url_to_repo = (:ssh_url_to_repo), http_url_to_repo = (:http_url_to_repo), web_url = (:web_url), name_with_namespace = (:name_with_namespace), path_with_namespace = (:path_with_namespace) where (id = :id)", { "name" => project.name, "desc" => project.description, "path" => project.path, "ssh_url_to_repo" => project.ssh_url_to_repo, :http_url_to_repo => project.http_url_to_repo, :web_url => project.web_url, :id => project.id, :path_with_namespace => project.path_with_namespace, :name_with_namespace => project.name_with_namespace }
         )
       else
         self.db.execute("INSERT INTO #{K_USER_PROJECTS} (id, name, desc, path, ssh_url_to_repo, http_url_to_repo, web_url,name_with_namespace, path_with_namespace)
@@ -174,7 +196,7 @@ module LgPodPlugin
     end
 
     # 通过名称查询项目信息
-    def query_project_info(name, git = nil)
+    def query_project_info(name, git)
       project_info = nil
       self.db.execute("select * from #{K_USER_PROJECTS} where name = '#{name}' or path = '#{name}' ;") do |row|
         name_with_namespace = row[7]
@@ -186,25 +208,30 @@ module LgPodPlugin
         web_url = row[6]
         description = row[2]
         ssh_url_to_repo = row[4]
-        http_url_to_repo =  row[5]
+        http_url_to_repo = row[5]
         project_info = ProjectModel.new(id, name, description, path, ssh_url_to_repo, http_url_to_repo, web_url, name_with_namespace, path_with_namespace)
         return project_info
       end
-      return project_info
+      project_info
     end
 
-    def insert_pod_refs(name, git, branch, tag , commit)
+    def delete_project_by_id(project_id)
+      ps = @db.prepare("DELETE FROM #{K_USER_PROJECTS} WHERE id = :n")
+      ps.execute('n' => project_id)
+    end
+
+    def insert_pod_refs(name, git, branch, tag, commit)
       id = LPodLatestRefs.get_pod_id(name, git)
       pod = LPodLatestRefs.new(id, name, git, branch, tag, commit)
       if self.query_pod_refs(id) != nil
         self.db.execute_batch(
-          "UPDATE #{K_POD_LATEST_REFS} SET name = (:name), git = (:git), branch = (:branch), tag = (:tag), sha = (:sha) where (id = :id)", { "name" => pod.name, "git" => pod.git, "sha" => pod.commit, "branch" => pod.branch, :tag => pod.tag, :id => pod.id}
+          "UPDATE #{K_POD_LATEST_REFS} SET name = (:name), git = (:git), branch = (:branch), tag = (:tag), sha = (:sha) where (id = :id)", { "name" => pod.name, "git" => pod.git, "sha" => pod.commit, "branch" => pod.branch, :tag => pod.tag, :id => pod.id }
         )
       else
         self.db.execute("INSERT INTO #{K_POD_LATEST_REFS} (id, name, git, branch, tag, sha)
             VALUES (?, ?, ?, ?, ?, ?)", [pod.id, pod.name, pod.git, pod.branch, pod.tag, pod.commit])
       end
-
+      self.insert_pod_sha_with_branch(name, git, commit, branch)
     end
 
     def query_pod_refs(id)
@@ -218,9 +245,42 @@ module LgPodPlugin
         commit = row[5]
         pod_info = LPodLatestRefs.new(id, name, git, branch, tag, commit)
       end
-      return pod_info
+      pod_info
+    end
+
+    # 保存 sha, branch 到数据库
+    def insert_pod_sha_with_branch(name, git, sha, branch)
+      return unless name && git && sha
+      id = Digest::MD5.hexdigest(name + git + sha)
+      if self.query_branch_with_sha(name, git, sha)[:sha] != nil
+        self.db.execute_batch(
+          "UPDATE #{K_POD_SHA_BRANCH} SET name = (:name), git = (:git), branch = (:branch), sha = (:sha) where (id = :id)", { "name" => name, "git" => git, "branch" => branch, :id => id, :sha => sha }
+        )
+      else
+        self.db.execute("INSERT INTO #{K_POD_SHA_BRANCH} (id, name, git, branch, sha)
+            VALUES (?, ?, ?, ?, ?)", [id, name, git, branch, sha])
+      end
+    end
+
+    # 通过 sha 查询对应 branch
+    def query_branch_with_sha(id = nil, name, git, sha)
+      hash_map = Hash.new
+      id = Digest::MD5.hexdigest(name + git + sha) unless id
+      self.db.execute("select * from #{K_POD_SHA_BRANCH} where id = '#{id}' and name = '#{name}' and git = '#{git}' ;") do |row|
+        new_id = row[0]
+        new_sha = row[4]
+        new_git = row[2]
+        new_name = row[1]
+        new_branch = row[3]
+        hash_map[:id] = new_id
+        hash_map[:git] = new_git
+        hash_map[:sha] = new_sha
+        hash_map[:name] = new_name
+        hash_map[:branch] = new_branch
+      end
+      hash_map
     end
 
   end
 
-end
+end