phone_generator_fr.go

package phone_generator_fr

import (
	"fmt"
	"math/rand"
	"os"
	"strconv"
	"strings"
	"sync"

	"github.com/Oni-kuki/operative-framework/session"
	"github.com/segmentio/ksuid"
	"gopkg.in/cheggaaa/pb.v1"
	"syreclabs.com/go/faker"
)

type PhoneGeneratorFr struct {
	session.SessionModule
	Sess    *session.Session `json:"-"`
	Current int              `json:"-"`
	Bar     *pb.ProgressBar  `json:"-"`
}

func PushPhoneGeneratorFrModule(s *session.Session) *PhoneGeneratorFr {
	mod := PhoneGeneratorFr{
		Sess:    s,
		Current: 1,
	}

	mod.CreateNewParam("NUMBER_PREFIX", "Country prefix ex: (33)", "33", false, session.STRING)
	mod.CreateNewParam("NAME_PREFIX", "Prefix of contact random name ex: (BHILLS_)", "", false, session.STRING)
	mod.CreateNewParam("FILE_PATH", "Location for generated VCards", "", false, session.STRING)
	mod.CreateNewParam("VCARD", "Generate vcard to file", "true", false, session.BOOL)
	mod.CreateNewParam("LIMIT", "Limit of phone numbers", "100", false, session.INT)
	return &mod
}

func (module *PhoneGeneratorFr) Name() string {
	return "phone.generator.fr"
}

func (module *PhoneGeneratorFr) Description() string {
	return "Generate VCard (.vcf) with random french numbers"
}

func (module *PhoneGeneratorFr) Author() string {
	return "Tristan Granier"
}

func (module *PhoneGeneratorFr) GetType() []string {
	return []string{
		session.T_TARGET_COUNTRY,
	}
}

func (module *PhoneGeneratorFr) GetInformation() session.ModuleInformation {
	information := session.ModuleInformation{
		Name:        module.Name(),
		Description: module.Description(),
		Author:      module.Author(),
		Type:        module.GetType(),
		Parameters:  module.Parameters,
	}
	return information
}

func (module *PhoneGeneratorFr) Start() {

	argumentPrefix, err := module.GetParameter("NUMBER_PREFIX")
	if err != nil {
		argumentPrefix = session.Param{
			Value: "",
		}
	}
	argumentFilePath, err2 := module.GetParameter("FILE_PATH")
	if err2 != nil {
		argumentFilePath = session.Param{
			Value: "",
		}
	}
	argumentNamePrefix, err3 := module.GetParameter("NAME_PREFIX")
	if err3 != nil {
		argumentNamePrefix = session.Param{
			Value: "",
		}
	}

	argumentVCard, err4 := module.GetParameter("VCARD")
	if err4 != nil {
		argumentVCard = session.Param{
			Value: "",
		}
	}

	argumentLimit, err5 := module.GetParameter("LIMIT")
	if err5 != nil {
		fmt.Println(err5.Error())
		return
	}
	if argumentLimit.Value == "" {
		argumentLimit.Value = "100"
	}

	module.Bar = pb.New(module.Sess.StringToInteger(argumentLimit.Value))

	pool, err := pb.StartPool(module.Bar)
	if err != nil {
		panic(err)
	}
	wg := new(sync.WaitGroup)
	for {
		if module.Current < module.Sess.StringToInteger(argumentLimit.Value) {
			wg.Add(1)
			go func(module *PhoneGeneratorFr, bar *pb.ProgressBar) {
				phone := faker.PhoneNumber().CellPhone()
				if strings.Contains(phone, "(") && strings.Contains(phone, ")") {
					newPhone := strings.Split(phone, ")")[1]
					if argumentPrefix.Value != "" {
						randomNumber := rand.Intn(9)
						newPhone = "+" + strings.TrimSpace(argumentPrefix.Value) + " 6" + strings.TrimSpace(strings.Replace(newPhone, "-", "", -1)) + strconv.Itoa(randomNumber)
					} else {
						newPhone = "+1 (213)" + newPhone
					}
					module.Results = append(module.Results, newPhone)
					module.Current = module.Current + 1
					bar.Increment()
				}
			}(module, module.Bar)
			wg.Done()
		} else {
			break
		}
	}
	wg.Wait()
	_ = pool.Stop()

	var file *os.File
	var errPath error

	if argumentFilePath.Value != "" {
		file, errPath = os.OpenFile(strings.TrimSpace(argumentFilePath.Value), os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
	} else {
		file, errPath = os.OpenFile("/beverlyHills-5000_1.vcf", os.O_WRONLY|os.O_APPEND|os.O_CREATE, 0644)
	}
	if errPath != nil {
		fmt.Println(errPath.Error())
		return
	}
	defer file.Close()
	for _, number := range module.Results {
		var uuid string
		if argumentNamePrefix.Value == "" {
			uuid = "BHills_GO_" + ksuid.New().String()
		} else {
			uuid = strings.TrimSpace(argumentNamePrefix.Value) + "_" + ksuid.New().String()
		}
		if argumentVCard.Value == "true" {
			_, _ = file.WriteString("BEGIN:VCARD\nVERSION:3.0\nN:" + uuid + ";;;\nFN:" + uuid + "\nTEL;type=HOME:" + number + "\nEND:VCARD\n")
		} else {
			_, _ = file.WriteString("\"" + number + "\",\n")
		}
	}
	module.Sess.Stream.Success("VCards successfully generated to '" + argumentFilePath.Value + "'")
}