verification.go

package freeproxy

import (
	"bytes"
	"encoding/json"
	"io"
	"net/http"
	"net/url"
	"strings"

	"github.com/sirupsen/logrus"
	"github.com/soluchok/freeproxy/providers"
)

type checkIP struct {
	IP string
}

func verifyProxy(proxy string) bool {
	req, err := http.NewRequest("GET", "https://api.ipify.org/?format=json", nil)
	if err != nil {
		logrus.Errorf("cannot create new request for verify err:%s", err)
		return false
	}

	proxyURL, err := url.Parse("http://" + proxy)
	if err != nil {
		logrus.Errorf("cannot parse proxy %q err:%s", proxy, err)
		return false
	}

	client := providers.NewClient()
	client.Transport.(*http.Transport).Proxy = http.ProxyURL(proxyURL)

	resp, err := client.Do(req)
	if resp != nil {
		defer resp.Body.Close()
	}

	if err != nil {
		logrus.Debugf("cannot verify proxy %q err:%s", proxy, err)
		return false
	}

	var body bytes.Buffer
	if _, err := io.Copy(&body, resp.Body); err != nil {
		logrus.Errorf("cannot copy resp.Body err:%s", err)
		return false
	}

	if resp.StatusCode != http.StatusOK {
		return false
	}

	var check checkIP
	if err := json.Unmarshal(body.Bytes(), &check); err != nil {
		logrus.Errorf("%d cannot unmarshal %q to checkIP struct err:%s", resp.StatusCode, body.String(), err)
		return false
	}

	return strings.HasPrefix(proxy, check.IP)
}