mock build.Part 4

classes/db.rb

@@ -270,4 +270,40 @@ class DBase
   def get_rpms_for_git(git_id)
     $DDB["select t2.id as rpmid, t2.rpmname, t1.reponame as repoid, t4.id as builid, t4.proj_id as prjid, t4.create_at from repos as t1 join rpms as t2 on t2.repo_id = t1.id join build_rpm as t3 on t3.rpm_id = t2.id join buildtask as t4 on t4.id = t3.build_id where t1.id = ? and t2.savepath not like '%.src.rpm' order by t4.create_at, t2.rpmname", git_id.to_i].all
   end
+
+  def get_rpm_info(rpm_id)
+    Rpms[rpm_id.to_i]
+  end
+
+  def get_rpm_build(rpm_id)
+    bld_info = BuildRpms.where(rpm_id: rpm_id.to_i).first
+    if bld_info.nil?
+      nil
+    else
+      bld_info[:build_id]
+    end
+  end
+
+  def get_rpm_srpms(rpm_id)
+    bld_info = BuildRpms.where(rpm_id: rpm_id.to_i).first
+    if bld_info.nil?
+      nil
+    else
+      bld_info[:build_id]
+      result = BuildRpms.where(build_id: bld_info[:build_id].to_i)
+      if result.nil?
+        nil
+      else
+        result.each do |item|
+          rpm_p = Rpms[item[:rpm_id].to_i]
+          unless rpm_p.nil?
+            if rpm_p[:savepath] =~ /\.src\.rpm$/
+              return rpm_p
+            end
+          end
+        end
+      end
+      nil
+    end
+  end
 end