forked from Gitlink/forgeplus
45 lines
1.3 KiB
Ruby
45 lines
1.3 KiB
Ruby
# == Schema Information
|
|
#
|
|
# Table name: project_units
|
|
#
|
|
# id :integer not null, primary key
|
|
# project_id :integer
|
|
# unit_type :integer
|
|
# created_at :datetime not null
|
|
# updated_at :datetime not null
|
|
#
|
|
# Indexes
|
|
#
|
|
# index_project_units_on_project_id (project_id)
|
|
#
|
|
|
|
class ProjectUnit < ApplicationRecord
|
|
belongs_to :project
|
|
|
|
enum unit_type: {code: 1, issues: 2, pulls: 3, wiki:4, devops: 5, versions: 6, resources: 7}
|
|
|
|
validates :unit_type, uniqueness: { scope: :project_id}
|
|
|
|
def self.init_types(project_id, project_type='common')
|
|
unit_types = project_type == 'sync_mirror' ? ProjectUnit::unit_types.except("pulls") : ProjectUnit::unit_types
|
|
unit_types.each do |_, v|
|
|
self.create!(project_id: project_id, unit_type: v)
|
|
end
|
|
end
|
|
|
|
def self.update_by_unit_types!(project, types)
|
|
# 同步镜像项目不能有合并请求模块
|
|
types.delete("pulls") if project.sync_mirror?
|
|
# 默认code类型自动创建
|
|
types << "code"
|
|
before_units = project.project_units.pluck(:unit_type).sort
|
|
project.project_units.where.not(unit_type: types).each(&:destroy!)
|
|
types.each do |type|
|
|
project.project_units.find_or_create_by!(unit_type: type)
|
|
end
|
|
after_units = project.project_units.pluck(:unit_type).sort
|
|
return before_units, after_units
|
|
end
|
|
|
|
end
|