utils.go

// Copyright 2017 hIMEI

// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//    http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

/** file utils.go contains constants, global vars and some helper functions*/

package main

import (
	"log"
	"math/rand"
	"strconv"
	"strings"
	"time"
)

// Size of buffered channel
const BUFFSIZE int = 1200

// Output colorizing
const (
	RED   string = "\x1B[31m"
	GRN          = "\x1B[32m"
	YEL          = "\x1B[33m"
	BLU          = "\x1B[34m"
	CYN          = "\x1B[36m"
	WHT          = "\x1B[97m"
	RESET        = "\x1B[0m"
	BOLD         = "\x1B[1m"
)

// Connect to search engine
const (
	ICHIDAN string = "ichidanv34wrx7m7.onion:80"
	SEARCH         = "/search?query="
	AUTH           = "/users/sign_in"
)

// Html parsing and logic expressions
const (
	ADDED          string = "Added on "
	LONGFORM              = "2017-09-09 01:30:35 UTC"
	PRE                   = "//pre"
	SPAN                  = "//span"
	LINK                  = "//a"
	HREF                  = "href"
	H2                    = "//h2"
	H3                    = "//h3"
	VERSION               = "//small"
	NONE                  = " "
	CURRENT               = "//em[@class='current']"
	DISABLED              = "//span[@class='next_page disabled']"
	SEARCHRESULT          = "//div[@id='search-results']"
	PAGINATION            = "//div[@class='pagination']"
	DETAILS               = "//a[@class='details']"
	SUMMARY               = "//div[@class='search-result-summary col-xs-4']"
	ROW                   = "//div[@class='row']"
	ONION                 = "//div[@class='onion']"
	TOTAL                 = "//div[@class='bignumber']"
	SERVICE               = "//div[@class='service']"
	SERVICES              = "//div[@class='services']"
	SERVICELONG           = "//li[@class='service service-long']"
	SERVICEDETAILS        = "//div[@class='service-details col-sm-2']"
	HOST                  = "//div[@class='search-result row-fluid']"
	NORESULT              = "//div[@class='msg alert alert-info']"
	NOAUTH                = "//div[@class='alert alert-dismissible alert-danger']"
	RESULT                = "//div[@class='col-sm-9']"
	PORT                  = "//div[@class='port']"
	PROTO                 = "//div[@class='protocol']"
	STATE                 = "//div[@class='state']"
	PRIMARY               = "//div[@class='span8 name']"
	PREVIOUS              = "← Previous"
	NEXT                  = "Next →"
	AND                   = "+"
	OR                    = "="
	NOT                   = "-"
)

// Console messages
const (
	NOTHING    string = "Nothing found there, Neo!"
	ERRAUTH           = "This does not work yet"
	ERRFLAGS          = "These flags cannot be used together"
	ONLYONE           = "Only one page"
	UNKNOWN           = "unknown version"
	WAIT              = "Waiting for connect..."
	PARSING           = "All data downloaded. Waiting for parsing"
	PROCESSING        = "in processing"
	RECEIVED          = "Respose received"
	NOTEXIST          = "Given path does not exist"
	EXIST             = "File already exist, we'll not rewrite it "
	FOUND             = "Total hosts found: "
	BYREQ             = "by request "
	PRESS             = "Press Enter to see details"
	FULL              = "Full info: "
	SHORT             = "Short info"
	SAVED             = "Saved to"
	WILL              = " will be printed"
	CONN              = "Connected"
)

// LOGIC is a operators for making logic requests
var LOGIC = []string{
	AND,
	OR,
	NOT,
}

var SLEEPER = sleeper()

// Toint converts string to int and handle errors
func toInt(str string) int {
	intCount, err := strconv.Atoi(str)
	ErrFatal(err)

	return intCount
}

// iToa converts int to string and handle errors
func iToa(i int) string {
	str := strconv.Itoa(i)

	return str
}

// ErrFatal is the basic errors handler
func ErrFatal(err error) {
	if err != nil {
		log.Fatal(err)
	}
}

// TrimString trims trailing and leading spaces from string
func trimString(str string) string {
	return strings.TrimSpace(str)
}

// Sleeper is a closure which calls time.Sleep with random time
// range between 300 and 359 milliseconds. It used to avoid server overloading
func sleeper() func() {
	return func() {
		s := rand.NewSource(time.Now().UnixNano())
		r := rand.New(s)
		p := time.Duration(300 + r.Intn(59))
		time.Sleep(p * time.Millisecond)
	}
}