asaaki/ryodo

# frozen_string_literal: true

module Ryodo
  class Domain
    # DomainString is a String with extended methods
    class DomainString < String
      def reverse
        to_a(:r).join('.')
      end
      alias r reverse

      def to_a(option = nil)
        case option
        when :reverse, :r
          dsplit.reverse
        else
          dsplit
        end
      end

      private

      def dsplit
        split('.', -1)
      end
    end

    # remove own class comparison (we will use String#== via method_missing)
    undef_method :==

    def initialize(domainStr)
      raise TypeError, 'Not a valid domain string!' unless domainStr.is_a?(String)

      @domain_string = DomainString.new domainStr.downcase
      parse_domain_string
      retrieve_domain_parts
    end

    def suffix
      DomainString.new(@suffix) if @suffix
    end
    alias tld suffix

    def domain
      DomainString.new(@domain) if @domain
    end
    alias registered_domain domain
    alias regdomain domain

    def second_level
      DomainString.new(@secondary) if @secondary
    end
    alias sld second_level
    alias registered_name second_level

    def subdomain
      DomainString.new(@subdomain) if @subdomain
    end

    def fqdn
      DomainString.new("#{self}.")
    end

    def valid?
      @suffix && @secondary && true
    end
    alias is_valid? valid?

    def to_s
      @domain_string
    end

    def inspect
      @domain_string.inspect
    end

    # pass all missings to @domain_string (String class)
    def method_missing(name, *args, &block)
      @domain_string.send(name, *args, &block)
    end

    # explicit definition of class' send
    def send(symbol, *args)
      __send__(symbol, *args)
    end

    private

    def parse_domain_string
      no_leading_dot     = @domain_string[0] != '.'
      @_parts            = Ryodo::Parser.run(@domain_string)
      @_no_dot_but_parts = no_leading_dot && @_parts
    end

    def retrieve_domain_parts
      retrieve_suffix
      retrieve_domain
      retrieve_secondary
      retrieve_subdomain
    end

    def retrieve_suffix
      @suffix = @_parts[0].reverse.join('.') if @_no_dot_but_parts
    end

    def retrieve_domain
      @domain = (@_parts[0] + @_parts[1]).reverse.join('.') if @_no_dot_but_parts && !@_parts[1].empty?
    end

    def retrieve_secondary
      @secondary = @_parts[1].first if @_no_dot_but_parts && !@_parts[1].empty?
    end

    def retrieve_subdomain
      @subdomain = (@_parts[2]).reverse.join('.') if @_no_dot_but_parts && !@_parts[2].empty?
    end
  end
end