Module: BiDiGenerate Private

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 920

def self.build_event(schema, event)
  params = event['params']
  payload_ref = params && params['ref'] && schema.structured_ref(params['ref'])
  Event.new(wire_name: event['method'], event_name: camel_to_snake(event['name']), payload_ref: payload_ref)
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 883

def self.build_ir(schema)
  schema.domains.map do |domain|
    Module.new(
      name: domain,
      ruby_class: snake_to_class_name(camel_to_snake(domain)),
      filename: camel_to_snake(domain),
      commands: schema.commands_for(domain).map { |cmd| build_command(schema, cmd) },
      events: schema.events_for(domain).map { |ev| build_event(schema, ev) },
      enums: schema.enums_for(domain),
      types: nest_synthetic(schema.types_for(domain)),
      spec_href: schema.domain_href(domain)
    )
  end
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 946

def self.call(schema_path, output_dir)
  raw = load_json(schema_path)
  schema = Schema.new(raw)
  modules = build_ir(schema)

  emit(modules, output_dir, 'module.rb.erb', 'rb')
  emit(modules, sig_dir(output_dir), 'module.rbs.erb', 'rbs')
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 50

def self.camel_to_snake(str)
  str
    .gsub(/([A-Z]+)([A-Z][a-z])/, '\1_\2')
    .gsub(/([a-z\d])([A-Z])/, '\1_\2')
    .downcase
end

# File 'rb/lib/selenium/webdriver/bidi/support/check_generated.rb', line 27

def self.check!(schema_rootpath)
  modules = build_ir(Schema.new(JSON.parse(File.read(schema_path(schema_rootpath)))))
  protocol_dir = File.expand_path('../protocol', __dir__)
  template = File.join(__dir__, 'templates', 'module.rb.erb')

  stale = modules.reject do |mod|
    path = File.join(protocol_dir, "#{mod.filename}.rb")
    File.exist?(path) && File.read(path) == render(mod, template)
  end
  return if stale.empty?

  warn "Generated BiDi protocol code is stale or hand-edited: #{stale.map { |m| "#{m.filename}.rb" }.sort.join(', ')}"
  warn 'Regenerate with: bazel run //rb/lib/selenium/webdriver:bidi-generate'
  exit 1
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 957

def self.emit(modules, output_dir, template, extension)
  target = File.join(workspace_root, output_dir)
  FileUtils.mkdir_p(target)

  tmpl = File.join(File.dirname(__FILE__), 'templates', template)
  modules.each do |mod|
    path = File.join(target, "#{mod.filename}.#{extension}")
    File.write(path, render(mod, tmpl))
    warn "bidi-generate: wrote #{path}"
  end
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 128

def self.enum_const_path(type_name)
  domain, local = type_name.split('.', 2)
  "#{snake_to_class_name(camel_to_snake(domain))}::#{screaming_snake(local)}"
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 138

def self.enum_key(value)
  camel_to_snake(value.to_s)
    .sub(/\A-(?=\d)/, 'neg')
    .sub(/\A-/, 'neg_')
    .gsub(/[^a-z0-9]+/, '_')
    .gsub(/\A_|_\z/, '')
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 975

private_class_method def self.load_json(path)
  resolved = File.exist?(path) ? path : File.join(Dir.pwd, path)
  JSON.parse(File.read(resolved))
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 930

def self.nest_synthetic(types)
  index = types.to_h { |t| [t.schema_name, t] }
  children = types.select { |t| !t.union? && t.synthetic }
  children.each do |child|
    owner = index[child.owner] ||
            raise("synthetic type #{child.schema_name} has no emitted owner #{child.owner}")
    owner.nested = (owner.nested || []) << child
    child.ruby_name = child.label
  end
  types - children
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 119

def self.rbs_nilable(type)
  return type if type == 'untyped' || type == 'nil' || type.end_with?('?')

  "#{type}?"
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 942

def self.render(mod, template_path)
  ERB.new(File.read(template_path), trim_mode: '-').result(binding)
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 75

def self.ruby_literal(value)
  return 'nil' if value.nil?

  value.is_a?(String) ? "'#{value}'" : value.to_s
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 106

def self.safe_field_name(name)
  RESERVED_FIELD_NAMES.include?(name) ? "#{name}_" : name
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 93

def self.safe_method_name(name)
  RUBY_RESERVED.include?(name) ? "#{name}_" : name
end

# File 'rb/lib/selenium/webdriver/bidi/support/check_generated.rb', line 48

def self.schema_path(rootpath)
  runfiles_root = File.expand_path('../../../../../..', __dir__)
  [File.join(runfiles_root, rootpath), rootpath].find { |p| File.exist?(p) } ||
    raise("BiDi schema not found (looked for #{rootpath})")
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 112

def self.screaming_snake(camel)
  camel_to_snake(camel).upcase
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 971

def self.sig_dir(output_dir)
  output_dir.sub(%r{(\A|/)lib/}, '\1sig/lib/')
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 57

def self.snake_to_class_name(snake)
  snake.split('_').map(&:capitalize).join
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 64

def self.type_class_name(type_name)
  type_name.split('.', 2).last.sub(/\A[a-z]/, &:upcase)
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 69

def self.type_ruby_path(type_name)
  domain = type_name.split('.', 2).first
  "#{snake_to_class_name(camel_to_snake(domain))}::#{type_class_name(type_name)}"
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 980

private_class_method def self.workspace_root
  ENV['BUILD_WORKSPACE_DIRECTORY'] || Dir.pwd
end

# File 'rb/lib/selenium/webdriver/bidi/support/bidi_generate.rb', line 84

def self.wrap_call(prefix, args, indent, open: '(', close: ')')
  one_line = "#{prefix}#{open}#{args.join(', ')}#{close}"
  return one_line if args.empty? || indent + one_line.length <= LINE_LIMIT

  pad = ' ' * (indent + 2)
  "#{prefix}#{open}\n#{pad}#{args.join(",\n#{pad}")}\n#{' ' * indent}#{close}"
end