DNSPodHelper.rb

#coding: utf-8
require 'singleton'
require 'net/http'
require 'net/https'
require 'open-uri'
require 'json'
require 'yaml'
require './Logger'

class DNSPodHelper

  #$format = "json"
 # $lang = "en"

  #$login_email = ""
  #$login_password = ""
 # $postFormat = ""

  include Singleton

  CONFIG=YAML.load_file(File.expand_path('./Config.yml'))

  def initialize
  	@format=CONFIG["format"]
	@lang=CONFIG["lang"]
	@login_email=CONFIG["login_email"]
	@login_password=CONFIG["login_password"]
	@postFormat="login_email=#{@login_email}&login_password=#{@login_password}&format=#{@format}&lang=#{@lang}"
	@userAgent="DynDNS/0.1 (#{@login_email})"
	@getIpUrl=CONFIG["ipUrl"]
  end
 
  #def SetUserInfo(email, pass)
  #    $login_email =  email
  #    $login_password = pass
  #    $postFormat = "login_email=" + $login_email + "&login_password=" + $login_password + "&format=" + $format + "&lang=" + $lang
 # end

  #$userAgent = "DynDNS/0.1 (#{$login_email})"
 # $getIpUrl = "http://members.3322.org/dyndns/getip"

# Defination of functions

  def PostRequest(functionAddr, postContent)
    http = Net::HTTP.new("dnsapi.cn", 443)
    http.use_ssl = true
    headers = {
        'Content-Type' => 'application/x-www-form-urlencoded',
        'User-Agent' => @userAgent
    }

    response = http.post2(functionAddr, postContent, headers)

    return response
  end

  def GetPublicIPAddr
    return open(@getIpUrl).read
  end

  def GetAPIVersion
    response = PostRequest("/Info.Version", @postFormat)
    content = JSON.parse(response.body)

    if(content['status']['code'] == "1")
      return content['status']['message']
    else
      puts "Failed to get API version!"
    end
  end

  def GetDomainInfo
    response = PostRequest("/Domain.List", @postFormat + "&type=all&offset=0&length=20")
    content = JSON.parse(response.body)
    domainInfo = Hash.new

    if(content['status']['code'] == "1")
      content['domains'].each { |obj|
        domainInfo[obj['name']] = obj['id']
      }
    else
      #puts "Failed to get domain id..."
      Utility::Log.info("Failed to get domain id...")
    end

    return domainInfo
  end

  def GetSubDomain(domainId, subDomain)
    response = PostRequest("/Record.List", @postFormat + "&domain_id=" + domainId.to_s() + "&offset=0&length=30")
    content = JSON.parse(response.body)
    subDomains = Hash.new

    if(content['status']['code'] == "1")
      content['records'].each { |obj|
        subDomains[obj['name']] = obj
      }
    else
      #puts "Failed to get sub-domain records..."
      Utility::Log.info("Failed to get sub-domain records...")
    end

    return subDomains[subDomain]
  end

  def UpdateSubDomainIP(domainId, recordId, subDomain, newIP)
    response = PostRequest("/Record.Modify", @postFormat + "&domain_id=" + domainId.to_s() + "&record_id=" + recordId.to_s() + "&sub_domain=" + subDomain + "&record_type=A" + "&record_line=默认" + "&value=" + newIP)
    content = JSON.parse(response.body)

    if(content['status']['code'] == "1")
      Utility.Log.info("DDNS IP updated successful!")
    end
  end
end