#!/usr/bin/env ruby
# Copyright (c) 2021 Huawei Device Co., Ltd.
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

require 'optparse'
require 'ostruct'
require 'yaml'

def data_instructions(data)
  data['groups'].flat_map { |g| g['instructions'] }
end

options = OpenStruct.new
optparser = OptionParser.new do |opts|
  opts.banner = 'Usage: combine.rb [options]'

  opts.on('-d', '--data FILE1,FILE2,...', Array, 'List of source data files in YAML format')
  opts.on('-o', '--output FILE', 'Output file (default is stdout)')

  opts.on('-h', '--help', 'Prints this help') do
    puts opts
    exit
  end
end
optparser.parse!(into: options)

exit unless options.data
exit if options.data.empty?

# TODO: Unify check with combining loop when we have assigned opcodes
options.data.drop(1).each do |path|
  tmp_data = YAML.load_file(File.expand_path(path))
  # check that all instructions are prefixed:
  instructions = data_instructions(tmp_data)
  raise 'Plugged in instructions must be prefixed' unless instructions.reject { |i| i['prefix'] }.empty?
end

# TODO: Remove when we have assigned opcodes:
options.data = options.data.rotate # put plug-in prefixes first

data = YAML.load_file(File.expand_path(options.data.first))
options.data.drop(1).each do |path|
  tmp_data = YAML.load_file(File.expand_path(path))
  %w[prefixes groups properties exceptions verification version min_version chapters].each do |attr|
    if data[attr]
      data[attr] += tmp_data[attr] if tmp_data[attr]
    else
      data[attr] = tmp_data[attr]
    end
  end
end

output = options.output ? File.open(File.expand_path(options.output), 'w') : $stdout
output.write(data.to_yaml)
output.close