main.go

package main

import (
	"fmt"
	"os"
	"sync"

	"github.com/kenjoe41/h1scope/pkg/hackerone"
	"github.com/kenjoe41/h1scope/pkg/options"
)

func main() {
	flags := options.ScanFlag()

	if flags.Username == "" || flags.Apikey == "" {
		fmt.Fprintln(os.Stderr, "H1 Username and API key are needed.")
		options.Usage()
		os.Exit(1)
	}

	programsChan := make(chan string)
	outputChan := make(chan string)

	var outputWG sync.WaitGroup
	outputWG.Add(1)
	go func() {
		defer outputWG.Done()

		for scopeAsset := range outputChan {
			fmt.Println(scopeAsset)
		}

	}()

	if flags.Handle != "" {
		err := hackerone.GetProgramScope(outputChan, flags)
		if err != nil {
			fmt.Fprintln(os.Stderr, err)
		}
	} else {
		hackerone.GetProgramsScope(programsChan, outputChan, flags)
	}

	go func() {
		outputWG.Wait()
		close(outputChan)
	}()
}