scinote-web/app/utilities/protocols_importer.rb

148 lines
3.8 KiB
Ruby
Raw Normal View History

2016-07-21 19:11:15 +08:00
module ProtocolsImporter
include RenamingUtil
2016-07-21 19:11:15 +08:00
2017-01-25 17:24:50 +08:00
def import_new_protocol(protocol_json, team, type, user)
2016-07-21 19:11:15 +08:00
remove_empty_inputs(protocol_json)
protocol = Protocol.new(
name: protocol_json["name"],
description: protocol_json["description"],
authors: protocol_json["authors"],
protocol_type: (type == :public ? :in_repository_public : :in_repository_private),
published_on: (type == :public ? Time.now : nil),
added_by: user,
2017-01-25 22:00:14 +08:00
team: team
2016-07-21 19:11:15 +08:00
)
# Try to rename record
if protocol.invalid? then
rename_record(protocol, :name)
end
# Okay, now save the protocol
protocol.save!
# Protocol is saved, populate it
populate_protocol(protocol, protocol_json, user, team)
2016-07-21 19:11:15 +08:00
return protocol
end
def import_into_existing(protocol, protocol_json, user, team)
2016-07-21 19:11:15 +08:00
# Firstly, destroy existing protocol's contents
protocol.destroy_contents(user)
protocol.reload
# Alright, now populate the protocol
populate_protocol(protocol, protocol_json, user, team)
2016-07-21 19:11:15 +08:00
protocol.reload
2016-08-12 22:31:40 +08:00
# Unlink the protocol
protocol.unlink
protocol
2016-07-21 19:11:15 +08:00
end
private
def populate_protocol(protocol, protocol_json, user, team)
2016-07-21 19:11:15 +08:00
protocol.reload
asset_ids = []
step_pos = 0
# Check if protocol has steps
if protocol_json['steps']
protocol_json['steps'].values.each do |step_json|
2016-07-21 19:11:15 +08:00
step = Step.create!(
name: step_json['name'],
description: step_json['description'],
2016-07-21 19:11:15 +08:00
position: step_pos,
completed: false,
user: user,
last_modified_by: user,
protocol: protocol
)
step_pos += 1
if step_json["checklists"]
step_json["checklists"].values.each do |checklist_json|
checklist = Checklist.create!(
name: checklist_json["name"],
step: step,
created_by: user,
last_modified_by: user
)
if checklist_json["items"]
item_pos = 0
checklist_json["items"].values.each do |item_json|
item = ChecklistItem.create!(
text: item_json["text"],
checked: false,
position: item_pos,
created_by: user,
last_modified_by: user,
checklist: checklist
)
item_pos += 1
end
end
end
end
2017-01-18 21:40:24 +08:00
if step_json['tables']
step_json['tables'].values.each do |table_json|
2016-07-21 19:11:15 +08:00
table = Table.create!(
2017-01-18 21:40:24 +08:00
name: table_json['name'],
contents: Base64.decode64(table_json['contents']),
2016-07-21 19:11:15 +08:00
created_by: user,
last_modified_by: user,
team: team
2016-07-21 19:11:15 +08:00
)
StepTable.create!(
step: step,
table: table
)
end
end
if step_json["assets"]
step_json["assets"].values.each do |asset_json|
asset = Asset.new(
created_by: user,
last_modified_by: user,
team: team
2016-07-21 19:11:15 +08:00
)
# Decode the file bytes
asset.file = StringIO.new(Base64.decode64(asset_json["bytes"]))
asset.file_file_name = asset_json["fileName"]
asset.file_content_type = asset_json["fileType"]
asset.save!
asset_ids << asset.id
StepAsset.create!(
step: step,
asset: asset
)
end
end
end
end
2016-07-21 19:11:15 +08:00
# Post process assets
asset_ids.each do |asset_id|
2017-01-25 17:24:50 +08:00
Asset.find(asset_id).post_process_file(protocol.team)
2016-07-21 19:11:15 +08:00
end
end
def remove_empty_inputs(obj)
obj.keys.each do |key|
if obj[key] == ""
obj[key] = nil
elsif obj[key].kind_of? Hash
# Recursive call
remove_empty_inputs(obj[key])
end
end
end
2016-08-12 22:31:40 +08:00
end