|
| 1 | +# frozen_string_literal: true |
| 2 | + |
| 3 | +require 'digest' |
| 4 | + |
| 5 | +require_relative '../account_manager' |
| 6 | +require_relative '../auth' |
| 7 | +require_relative '../auto_source' |
| 8 | +require_relative '../cache_ttl' |
| 9 | +require_relative '../exceptions' |
| 10 | +require_relative '../local_config' |
| 11 | +require_relative '../url_validator' |
| 12 | +require_relative '../api/v1/contract' |
| 13 | +require_relative 'resolved_source' |
| 14 | + |
| 15 | +module Html2rss |
| 16 | + module Web |
| 17 | + module Feeds |
| 18 | + ## |
| 19 | + # Resolves static and token-backed requests into shared generator inputs. |
| 20 | + module Resolver |
| 21 | + class << self |
| 22 | + # @param feed_request [Html2rss::Web::Feeds::Request] |
| 23 | + # @return [Html2rss::Web::Feeds::ResolvedSource] |
| 24 | + def call(feed_request) |
| 25 | + case feed_request.target_kind |
| 26 | + when :static |
| 27 | + resolve_static(feed_request) |
| 28 | + when :token |
| 29 | + resolve_token(feed_request) |
| 30 | + else |
| 31 | + raise BadRequestError, "Unsupported feed target: #{feed_request.target_kind}" |
| 32 | + end |
| 33 | + end |
| 34 | + |
| 35 | + private |
| 36 | + |
| 37 | + # @param feed_request [Html2rss::Web::Feeds::Request] |
| 38 | + # @return [Html2rss::Web::Feeds::ResolvedSource] |
| 39 | + def resolve_static(feed_request) |
| 40 | + config = LocalConfig.find(feed_request.feed_name) |
| 41 | + config[:params] = (config[:params] || {}).merge(feed_request.params) if feed_request.params.any? |
| 42 | + config[:strategy] ||= Html2rss::RequestService.default_strategy_name |
| 43 | + |
| 44 | + ResolvedSource.new( |
| 45 | + source_kind: :static, |
| 46 | + cache_identity: static_cache_identity(feed_request.feed_name, feed_request.params), |
| 47 | + generator_input: config, |
| 48 | + ttl_seconds: CacheTtl.seconds_from_minutes(config.dig(:channel, :ttl)) |
| 49 | + ) |
| 50 | + end |
| 51 | + |
| 52 | + # @param feed_request [Html2rss::Web::Feeds::Request] |
| 53 | + # @return [Html2rss::Web::Feeds::ResolvedSource] |
| 54 | + def resolve_token(feed_request) |
| 55 | + feed_token = validated_feed_token(feed_request.token) |
| 56 | + strategy = resolved_strategy(feed_token) |
| 57 | + generator_input = token_generator_input(feed_token.url, strategy) |
| 58 | + |
| 59 | + ResolvedSource.new( |
| 60 | + source_kind: :token, |
| 61 | + cache_identity: token_cache_identity(feed_request.token), |
| 62 | + generator_input: generator_input, |
| 63 | + ttl_seconds: CacheTtl.seconds_from_minutes(generator_input.dig(:channel, :ttl), default: 300) |
| 64 | + ) |
| 65 | + end |
| 66 | + |
| 67 | + # @param feed_name [String] |
| 68 | + # @param params [Hash{Object=>Object}] |
| 69 | + # @return [String] |
| 70 | + def static_cache_identity(feed_name, params) |
| 71 | + normalized_params = params.to_h.sort_by { |key, _| key.to_s } |
| 72 | + digest = Digest::SHA256.hexdigest(Marshal.dump(normalized_params)) |
| 73 | + "static:#{feed_name}:#{digest}" |
| 74 | + end |
| 75 | + |
| 76 | + # @param token [String] |
| 77 | + # @return [String] |
| 78 | + def token_cache_identity(token) |
| 79 | + "token:#{Digest::SHA256.hexdigest(token.to_s)}" |
| 80 | + end |
| 81 | + |
| 82 | + # @param token [String] |
| 83 | + # @return [Html2rss::Web::FeedToken] |
| 84 | + def validated_feed_token(token) |
| 85 | + feed_token = Auth.validate_and_decode_feed_token(token) |
| 86 | + raise UnauthorizedError, 'Invalid token' unless feed_token |
| 87 | + |
| 88 | + account = AccountManager.get_account_by_username(feed_token.username) |
| 89 | + raise UnauthorizedError, 'Account not found' unless account |
| 90 | + |
| 91 | + ensure_token_access!(account, feed_token.url) |
| 92 | + ensure_auto_source_enabled! |
| 93 | + feed_token |
| 94 | + end |
| 95 | + |
| 96 | + # @param account [Hash{Symbol=>Object}] |
| 97 | + # @param url [String] |
| 98 | + # @return [void] |
| 99 | + def ensure_token_access!(account, url) |
| 100 | + raise ForbiddenError, 'Access Denied' unless UrlValidator.url_allowed?(account, url) |
| 101 | + end |
| 102 | + |
| 103 | + # @return [void] |
| 104 | + def ensure_auto_source_enabled! |
| 105 | + raise ForbiddenError, Api::V1::Contract::MESSAGES[:auto_source_disabled] unless AutoSource.enabled? |
| 106 | + end |
| 107 | + |
| 108 | + # @param feed_token [Html2rss::Web::FeedToken] |
| 109 | + # @return [String] |
| 110 | + def resolved_strategy(feed_token) |
| 111 | + strategy = feed_token.strategy.to_s.strip |
| 112 | + strategy = Html2rss::RequestService.default_strategy_name.to_s if strategy.empty? |
| 113 | + supported = Html2rss::RequestService.strategy_names.map(&:to_s) |
| 114 | + raise BadRequestError, 'Unsupported strategy' unless supported.include?(strategy) |
| 115 | + |
| 116 | + strategy |
| 117 | + end |
| 118 | + |
| 119 | + # @param url [String] |
| 120 | + # @param strategy [String] |
| 121 | + # @return [Hash{Symbol=>Object}] |
| 122 | + def token_generator_input(url, strategy) |
| 123 | + LocalConfig.global |
| 124 | + .slice(:stylesheets, :headers) |
| 125 | + .merge( |
| 126 | + strategy: strategy.to_sym, |
| 127 | + channel: { url: url }, |
| 128 | + auto_source: {} |
| 129 | + ) |
| 130 | + end |
| 131 | + end |
| 132 | + end |
| 133 | + end |
| 134 | + end |
| 135 | +end |
0 commit comments