class GraphQL::Upgrader::Schema

def initialize(schema)
  @schema = schema
end

def upgrade
  transformable = schema.dup
  transformable.sub!(
    /([a-zA-Z_0-9]*) = GraphQL::Schema\.define do/, 'class \1 < GraphQL::Schema'
  )
  transformable.sub!(
    /object_from_id ->\s?\((.*)\) do/, 'def self.object_from_id(\1)'
  )
  transformable.sub!(
    /resolve_type ->\s?\((.*)\) do/, 'def self.resolve_type(\1)'
  )
  transformable.sub!(
    /id_from_object ->\s?\((.*)\) do/, 'def self.id_from_object(\1)'
  )
  transformable
end