module Docs
  class Pytorch < UrlScraper
    self.name = 'PyTorch'
    self.slug = 'pytorch'
    self.type = 'sphinx'
    self.force_gzip = true
    self.links = {
      home: 'https://pytorch.org/',
      code: 'https://github.com/pytorch/pytorch'
    }

    html_filters.push 'pytorch/entries', 'pytorch/clean_html', 'sphinx/clean_html'

    options[:skip] = ['cpp_index.html', 'packages.html', 'py-modindex.html', 'genindex.html']
    options[:skip_patterns] = [/\Acommunity/, /\A_modules/, /\Anotes/, /\Aorg\/pytorch\//]
    options[:max_image_size] = 256_000

    options[:attribution] = <<-HTML
    &copy; 2019-2024 Torch Contributors<br>
    Licensed under the 3-clause BSD License.
    HTML

    version '2.1' do
      self.release = '2.1'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '2.0' do
      self.release = '2.0'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.13' do
      self.release = '1.13'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.12' do
      self.release = '1.12'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.11' do
      self.release = '1.11'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.10' do
      self.release = '1.10'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.9' do
      self.release = '1.9.1'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.8' do
      self.release = '1.8.1'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.7' do
      self.release = '1.7.1'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.6' do
      self.release = '1.6.0'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.5' do
      self.release = '1.5.1'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.4' do
      self.release = '1.4.0'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.3' do
      self.release = '1.3.1'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.2' do
      self.release = '1.2.0'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.1' do
      self.release = '1.1.0'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    version '1.0' do
      self.release = '1.0.1'
      self.base_url = "https://pytorch.org/docs/#{release}/"
    end

    def get_latest_version(opts)
      get_latest_github_release('pytorch', 'pytorch', opts)
    end
  end
end