request.go

package goinside

import (
	"encoding/base64"
	"io"
	"net/http"
	"net/url"
	"time"
)

// urls
const (
	gallerysURL    = "http://m.dcinside.com/category_gall_total.html"
	commentMoreURL = "http://m.dcinside.com/comment_more_new.php"
)

// apis
const (
	loginAPI              dcinsideAPI = "https://dcid.dcinside.com/join/mobile_app_login.php"
	appKeyVerificationAPI dcinsideAPI = "https://dcid.dcinside.com/join/mobile_app_key_verification_3rd.php"
	appCheckAPI           dcinsideAPI = "http://json2.dcinside.com/json0/app_check_A_rina.php"
	writeArticleAPI       dcinsideAPI = "http://upload.dcinside.com/_app_write_api.php"
	deleteArticleAPI      dcinsideAPI = "http://m.dcinside.com/api/gall_del.php"
	writeCommentAPI       dcinsideAPI = "http://m.dcinside.com/api/comment_ok.php"
	deleteCommentAPI      dcinsideAPI = "http://m.dcinside.com/api/comment_del.php"
	recommendUpAPI        dcinsideAPI = "http://m.dcinside.com/api/_recommend_up.php"
	recommendDownAPI      dcinsideAPI = "http://m.dcinside.com/api/_recommend_down.php"
	reportAPI             dcinsideAPI = "http://m.dcinside.com/api/report_upload.php"
	redirectAPI           dcinsideAPI = "http://m.dcinside.com/api/redirect.php"
	readListAPI           dcinsideAPI = "http://m.dcinside.com/api/gall_list_new.php"
	readArticleAPI        dcinsideAPI = "http://m.dcinside.com/api/view2.php"
	readArticleDetailAPI  dcinsideAPI = "http://m.dcinside.com/api/gall_view.php"
	readArticleImageAPI   dcinsideAPI = "http://m.dcinside.com/api/view_img.php"
	readCommentAPI        dcinsideAPI = "http://m.dcinside.com/api/comment_new.php"
	majorGalleryListAPI   dcinsideAPI = "http://json.dcinside.com/App/gall_name.php"
	minorGalleryListAPI   dcinsideAPI = "http://json.dcinside.com/App/gall_name_sub.php"
)

// content types
const (
	defaultContentType    = "application/x-www-form-urlencoded; charset=UTF-8"
	nonCharsetContentType = "application/x-www-form-urlencoded"
)

var (
	apiRequestHeader = map[string]string{
		"User-Agent": "dcinside.app",
		"Referer":    "http://www.dcinside.com",
		"Host":       "upload.dcinside.com",
		"Connection": "Keep-Alive",
	}
	mobileRequestHeader = map[string]string{
		"User-Agent": "Linux Android",
		"Referer":    "http://m.dcinside.com",
	}
	imageRequestHeader = map[string]string{
		"Referer": "http://www.dcinside.com",
	}
)

type session interface {
	articleWriteForm(string, string, string, ...string) (form io.Reader, contentType string)
	articleDeleteForm(string, string) (form io.Reader, contentType string)
	commentWriteForm(string, string, string, ...string) (form io.Reader, contentType string)
	commentDeleteForm(string, string, string) (form io.Reader, contentType string)
	actionForm(string, string) (form io.Reader, contentType string)
	getAppID() string
	connector
}

type dcinsideAPI string

func (api dcinsideAPI) post(c connector, form io.Reader, contentType string) (*http.Response, error) {
	return do(c, "POST", string(api), nil, form, contentType, apiRequestHeader)
}

func (api dcinsideAPI) get(m map[string]string) (*http.Response, error) {
	URL, err := url.Parse(string(api))
	if err != nil {
		return nil, err
	}
	data := url.Values{}
	for k, v := range m {
		data.Add(k, v)
	}
	URL.RawQuery = data.Encode()
	encodedParams := base64.StdEncoding.EncodeToString([]byte(URL.String()))

	URL, err = url.Parse(string(redirectAPI))
	if err != nil {
		return nil, err
	}
	data = url.Values{}
	data.Add("hash", encodedParams)
	URL.RawQuery = data.Encode()

	return do(&GuestSession{}, "GET", URL.String(), nil, nil, defaultContentType, apiRequestHeader)
}

func (api dcinsideAPI) getWithoutHash() (*http.Response, error) {
	URL, err := url.Parse(string(api))
	if err != nil {
		return nil, err
	}
	return do(&GuestSession{}, "GET", URL.String(), nil, nil, defaultContentType, apiRequestHeader)
}

type connector interface {
	Connection() *Connection
}

func post(c connector, URL string, cookies []*http.Cookie, form io.Reader, contentType string) (*http.Response, error) {
	return do(c, "POST", URL, cookies, form, contentType, mobileRequestHeader)
}

func get(c connector, URL string) (*http.Response, error) {
	return do(c, "GET", URL, nil, nil, defaultContentType, mobileRequestHeader)
}

func do(c connector, method, URL string, cookies []*http.Cookie, form io.Reader, contentType string, requestHeader map[string]string) (*http.Response, error) {
	req, err := http.NewRequest(method, URL, form)
	if err != nil {
		return nil, err
	}
	for _, cookie := range cookies {
		req.AddCookie(cookie)
	}
	if contentType != "" {
		req.Header.Set("Content-Type", contentType)
	}
	for k, v := range requestHeader {
		req.Header.Set(k, v)
	}
	client := func() *http.Client {
		proxy := c.Connection().proxy
		if proxy != nil {
			return &http.Client{Transport: &http.Transport{Proxy: proxy}}
		}
		return &http.Client{}
	}()
	client.CheckRedirect = func(req *http.Request, via []*http.Request) error {
		for k, v := range requestHeader {
			req.Header.Set(k, v)
		}
		return nil
	}
	if c.Connection().timeout != 0 {
		client.Timeout = c.Connection().timeout
	}
	return client.Do(req)
}

func doImage(URL ImageURLType) (*http.Response, error) {
	req, err := http.NewRequest("GET", string(URL), nil)
	if err != nil {
		return nil, err
	}
	client := &http.Client{
		Timeout: time.Second * 3,
	}
	for k, v := range imageRequestHeader {
		req.Header.Set(k, v)
	}
	return client.Do(req)
}