Added preparing actions for custom build

classes/db.rb

@@ -42,6 +42,9 @@ end
 class RpmUploaded < Sequel::Model(:rpm_uploaded)
 end
 
+class CustomBuildScript < Sequel::Model(:custom_build_script)
+end
+
 class DBase
   attr :error, :last_id, :cfg
 
@@ -79,6 +82,7 @@ class DBase
     rep_id = Repos.where(reponame: repo_name).first
     unless rep_id[:id].nil?
       id = rep_id[:id]
+      CustomBuildScript.where(repo_id: id).delete
       RepocRecips.where(repo_id: id).delete
       ReposProjects.where(repo_id: id).delete
       Repos.where(reponame: repo_name).delete
@@ -369,6 +373,7 @@ class DBase
       count = count + 1
     end
     return 1 if count > 0
+    CustomBuildScript.where(proj_id: proj_id.to_i).delete
     RpmUploaded.where(proj_id: prj_id.to_i).delete
     ReposProjects.where(proj_id: prj_id.to_i).delete
     ProjectsReposSpec.where(proj_id: prj_id.to_i).delete
@@ -431,4 +436,22 @@ class DBase
   def get_project_uploaded_rpms(proj_id)
     RpmUploaded.where(proj_id: proj_id.to_i)
   end
+
+  def get_project_custom_build(proj_id, git_id)
+    CustomBuildScript.where(proj_id: proj_id.to_i, repo_id: git_id.to_i)
+  end
+
+  def set_project_custom_build(id, repo_id, script_name, descr, content)
+    result = get_project_custom_build(id, repo_id).first
+    unless result.nil?
+      CustomBuildScript.where(id: result[:id]).update(:content => content, :filepath => script_name, :descr => descr)
+    else
+      id = CustomBuildScript.insert(:content => content, :filepath => script_name, :descr => descr, :proj_id => id.to_i, :repo_id => repo_id.to_i)
+      @last_id = id
+    end
+  end
+
+  def del_project_custom_build(id, repo_id)
+    CustomBuildScript.where(proj_id: id.to_i, repo_id: repo_id.to_i).delete
+  end
 end