kurier.go

// What's the delivery status?
// dRbiG, 2014
// See LICENSE.txt

package main

import (
	"errors"
	"fmt"
	"github.com/kennygrant/sanitize"
	"github.com/moovweb/gokogiri"
	"io/ioutil"
	"net/http"
	"net/url"
	"os"
	"regexp"
	"strings"
	"sync"
)

type DlFunc func(base, id string) (body []byte, err error)

type Service struct {
	Name       string
	Downloader DlFunc
	Matcher    string
	URL        string
	XPath      string
	Extractor  *regexp.Regexp
}

func (s *Service) IsMatch(id string) bool {
	matched, err := regexp.MatchString(s.Matcher, id)
	if err != nil {
		panic(err)
	}
	return matched
}

func (s *Service) Check(id string) (status string, err error) {
	body, err := s.Downloader(s.URL, id)
	if err != nil {
		return
	}

	if s.Extractor != nil {
		parts := s.Extractor.FindSubmatch(body)
		if parts == nil {
			return "", nil
		}

		status = string(parts[1])
	} else {
		doc, err := gokogiri.ParseHtml(body)
		if err != nil {
			return "", err
		}
		defer doc.Free()

		res, err := doc.Search(s.XPath)
		if err != nil {
			return "", err
		}
		if len(res) < 1 {
			return "", nil
		}

		status = sanitize.HTML(res[0].String())
		status = replacer.ReplaceAllString(status, " ")
		status = strings.TrimSpace(status)
	}

	return
}

func dlSimpleGet(base, id string) (body []byte, err error) {
	url := fmt.Sprintf(base, id)

	res, err := client.Get(url)
	if err != nil {
		return
	}
	defer res.Body.Close()

	if res.StatusCode != 200 {
		err = errors.New(fmt.Sprintf("DownloadPage: %d %s", res.StatusCode, url))
		return
	}

	body, err = ioutil.ReadAll(res.Body)
	return
}

func dlPocztex(base, id string) (body []byte, err error) {
	form := url.Values{}
	form.Add("n", id)
	form.Add("s", "1")

	req, err := http.NewRequest("POST", base, strings.NewReader(form.Encode()))
	if err != nil {
		return
	}
	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
	req.AddCookie(&http.Cookie{Name: "PHPSESSID", Value: "1"})

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

	if res.StatusCode != 200 {
		err = errors.New(fmt.Sprintf("DownloadPocztex: %d %s", res.StatusCode, base))
		return
	}

	body, err = ioutil.ReadAll(res.Body)
	return
}

func dlDPD(base, id string) (body []byte, err error) {
	form := url.Values{}
	form.Add("q", id)
	form.Add("typ", "1")

	req, err := http.NewRequest("POST", base, strings.NewReader(form.Encode()))
	if err != nil {
		return
	}
	req.Header.Set("Content-Type", "application/x-www-form-urlencoded")

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

	if res.StatusCode != 200 {
		err = errors.New(fmt.Sprintf("DownloadDPD: %d %s", res.StatusCode, base))
		return
	}

	body, err = ioutil.ReadAll(res.Body)
	return
}

func checkService(id string, s Service) {
	res, err := s.Check(id)
	if (err == nil) && (res != "") {
		fmt.Printf("%20s %-8s %s\n", id, s.Name, res)
	}
	wg.Done()
}

var (
	client   *http.Client
	replacer *regexp.Regexp
	wg       sync.WaitGroup
)

func main() {
	services := [...]Service{
		Service{
			"DHL",
			dlSimpleGet,
			"^\\d{11}$",
			"http://www.dhl.com.pl/sledzenieprzesylkikrajowej/szukaj.aspx?m=0&sn=%s",
			"//*[@id='middle']/table/tbody/tr[2]/td[4]/text()[1]",
			nil,
		},
		Service{
			"DPD",
			dlDPD,
			"^\\w{14}$",
			"https://tracktrace.dpd.com.pl/findPackage",
			"//table/tr[2]/td[3]/text()",
			nil,
		},
		Service{
			"SIÓDEMKA",
			dlSimpleGet,
			"^\\d{13}$",
			"https://siodemka.com/tracking/%s/",
			"//*[@id='page']/div[2]/table[2]/tbody/tr[4]/td[4]",
			nil,
		},
		Service{
			"UPS",
			dlSimpleGet,
			"^1Z\\w{16}$",
			"http://wwwapps.ups.com/WebTracking/track?loc=pl_PL&HTMLVersion=5.0&Requester=UPSHome&WBPM_lid=homepage/ct1.html_pnl_trk&trackNums=%s&track.x=Monitoruj",
			"//*[@id='tt_spStatus']/text()",
			nil,
		},
		Service{
			"GLS",
			dlSimpleGet,
			"^\\d{11}$",
			"https://gls-group.eu/app/service/open/rest/PL/pl/rstt001?match=%s&caller=witt002",
			"",
			regexp.MustCompile("\"statusText\":\"(.*?)\""),
		},
		Service{
			"K-EX",
			dlSimpleGet,
			"^\\d{9}$",
			"http://kurier.k-ex.pl/tnt_szczegoly.php?nr=%s",
			"//table[last()]/tr[last()]/td[4]/text()",
			nil,
		},
		Service{
			"Pocztex",
			dlPocztex,
			"^\\d{20}$",
			"http://www.pocztex.pl/sledzenie/wssClient.php",
			"//table[@id='zadarzenia_td']/tr[last()]/td[1]/text()",
			nil,
		},
		Service{
			"InPost",
			dlSimpleGet,
			"^\\d{24}$",
			"https://paczkomaty.pl/pl/znajdz-paczke?parcel=%s",
			"//*[@id='find-parcel']/div[3]/div/table/tbody/tr[3]/td",
			nil,
		},
	}

	if len(os.Args) == 2 && os.Args[1] == "list" {
		for nr, s := range services {
			fmt.Printf(
				"%2d. %s  (%s)\n    %s\n    %s\n",
				nr+1, s.Name, s.Matcher, s.URL, s.XPath,
			)
		}
		os.Exit(0)
	}

	client = &http.Client{}
	replacer = regexp.MustCompile("\\s{2,}|\\t+|\\\\n")

	for _, id := range os.Args[1:] {
		for _, s := range services {
			if s.IsMatch(id) {
				wg.Add(1)
				go checkService(id, s)
			}
		}
	}

	wg.Wait()
}