lib/tsumanne.rb in tsumanne-0.0.1 vs lib/tsumanne.rb in tsumanne-0.0.2

- old
+ new

@@ -1,17 +1,12 @@ # typed: strict # frozen_string_literal: true -require "net/http" -require "uri" -require "json" -require "zlib" - -require "mhtml" require "sorbet-runtime" +require_relative "tsumanne/api" require_relative "tsumanne/version" # API module for tsumanne.net includes knowledge as const. module Tsumanne extend T::Sig @@ -23,106 +18,6 @@ BOARD_IDS = T.let({ img: "si", may: "my", jun: "tj", dat: "sa", special: "sp" }.freeze, T::Hash[Symbol, String]) # Corresponing orders to list indexes. (`index` is same as `category`) INDEXES_ORDERS = T.let({ hira: "↓あ", newer: "↓新" }.freeze, T::Hash[Symbol, String]) class Error < StandardError; end - - # API to get information fron tsumanne.net - class API - extend T::Sig - - sig{ returns(String) } - attr_reader :board_id - - sig { params(board_id: Symbol).void } - def initialize(board_id:) - @board_id = T.let(T.must(BOARD_IDS[board_id]), String) - end - - sig { params(index: String, page: Integer).returns(T::Hash[String, T.untyped]) } - def get_threads(index: "all", page: 1) - # https://tsumanne.net/si/all/1 - # https://tsumanne.net/si/hoge/1 - fetch_json(paths: [index, page.to_s]) - end - - sig { params(thread_id: String).returns(Mhtml::RootDocument) } - def get_thread_mht(thread_id) - # https://tsumanne.net/si/mht.php?id=129691 - res = fetch_json(paths: ["mht.php"], query: { id: thread_id }, method: :get_response) - mht_gz = fetch(paths: [T.let(res["path"], String)]) - # https://ksef-3go.hatenadiary.org/entry/20070924/1190563143 - # https://docs.ruby-lang.org/ja/latest/method/Zlib=3a=3aInflate/s/new.html - zstream = Zlib::Inflate.new(Zlib::MAX_WBITS + 32) - buf = zstream.inflate(mht_gz) - zstream.finish - zstream.close - Mhtml::RootDocument.new(buf) - end - - sig { params(thread_path: String).returns(T.nilable(Mhtml::RootDocument)) } - def get_thread_from_path(thread_path) - # https://tsumanne.net/si/data/2023/08/30/8883354/ - match_data = %r{^\d{4}/\d{2}/\d{2}/(?<thread_id>\d+)$}.match(thread_path) - return if match_data.nil? - get_thread_mht(T.must(match_data[:thread_id])) - end - - sig { params(uri: URI).returns(T::Hash[String, T.untyped]) } - def search_thread_from_uri(uri) - # https://tsumanne.net/si/indexes.php?format=json&w=&sbmt=URL - fetch_json(paths: ["indexes.php"], query: { w: uri, sbmt: :URL }) - end - - sig { params(keyword: T.nilable(String), order: Symbol, page: Integer).returns(T::Hash[String, T.untyped]) } - def search_indexes(keyword: nil, order: :newer, page: 1) - # https://tsumanne.net/si/indexes.php?format=json&w=&sbmt=URL - # https://tsumanne.net/si/indexes.php?format=json&w=&sbmt=%E2%86%93%E6%96%B0 - fetch_json(paths: ["indexes.php"], query: { w: keyword, sbmt: INDEXES_ORDERS[order], p: page }) - end - - sig { params(uri: URI, indexes: T.nilable(T::Array[String])).returns(T::Hash[String, T.untyped]) } - def register_thread(uri, indexes: nil) - # post, https://tsumanne.net/si/input.php?format=json&url=...&category=... - fetch_json(paths: ["input.php?format=json"], query: { url: uri, category: (indexes || []).join(",") }, method: :post) - end - - private - - sig do - params(paths: T.nilable(T::Array[String]), - query: T.nilable(T::Hash[Symbol, T.any(Integer, String, Symbol)]), - method: Symbol).returns(T.nilable(String)) - end - def fetch(paths: nil, query: nil, method: :get) - uri = join_paths(BASE_URL, [@board_id] + (paths || [])) - query = URI.encode_www_form(query || {}) - - case method - when :get - uri.query = query - Net::HTTP.get(uri) - when :get_response - uri.query = query - Net::HTTP.get_response(uri).body - when :post - Net::HTTP.post(uri, query).body - end - end - - sig do - params(paths: T.nilable(T::Array[String]), - query: T.nilable(T::Hash[Symbol, T.any(Integer, String, Symbol)]), - method: Symbol).returns(T::Hash[String, T.untyped]) - end - def fetch_json(paths: nil, query: nil, method: :get) - query ||= {} - json = fetch(paths:, query: query.merge({ format: :json }), method:) - JSON.parse(T.must(json)) - end - - sig { params(base: String, paths: T::Array[String]).returns(URI::Generic) } - def join_paths(base, *paths) - URI.parse(T.must(([base] + paths).reduce { File.join(_1, _2) })) - end - end end