implementing DB driver

cmd/bruter/main.go

@@ -15,6 +15,7 @@ import (
 	"github.com/CyberRoute/bruter/pkg/handlers"
 	"github.com/CyberRoute/bruter/pkg/network"
 	"github.com/CyberRoute/bruter/pkg/render"
+	"github.com/CyberRoute/bruter/pkg/driver"
 	"github.com/alexedwards/scs/v2"
 	"github.com/evilsocket/islazy/async"
 	"github.com/rs/zerolog"
@@ -55,13 +56,49 @@ func main() {
 	IP, _ := network.ResolveByName(*Domain)
 	logger.Info().Msg(fmt.Sprintf("Scanning IP %s %s", IP, "OK"))
 
+	db, err := run(&logger)
+	if err != nil {
+		logger.Info().Msg(err.Error())
+	}
+	defer db.SQL.Close()
+
+
+
+
+	file, err := os.Open(*Dictionary)
+	if err != nil {
+		logger.Fatal().Err(err).Msg("")
+	}
+	defer file.Close()
+
+	list := readDictionary(file)
+	total := len(list)
+	shift := 1
+
+	queue := createQueue(&app.Mu, *Domain, list, shift, total, *Verbose)
+
+	queue.WaitDone()
+
+	fmt.Println("\nAll tasks completed, press Ctrl-C to quit.")
+	select {}
+}
+
+func run(logger *zerolog.Logger) (*driver.DB, error) {
 	app.InProduction = false
 	session = scs.New()
 	session.Lifetime = 24 * time.Hour
 	session.Cookie.Persist = true
 	session.Cookie.SameSite = http.SameSiteLaxMode
 	session.Cookie.Secure = app.InProduction
 	app.Session = session
+	// conct to database
+	logger.Info().Msg("Connecting to Postgres")
+	db, err := driver.ConnectSQL("host=localhost port=5432 dbname= user= password=")
+	if err != nil {
+		return nil, err
+	}
+	logger.Info().Msg("Connected to Postgres")
+
 
 	tc, err := render.CreateTemplateCache()
 	if err != nil {
@@ -87,23 +124,7 @@ func main() {
 			panic(err)
 		}
 	}()
-
-	file, err := os.Open(*Dictionary)
-	if err != nil {
-		logger.Fatal().Err(err).Msg("")
-	}
-	defer file.Close()
-
-	list := readDictionary(file)
-	total := len(list)
-	shift := 1
-
-	queue := createQueue(&app.Mu, *Domain, list, shift, total, *Verbose)
-
-	queue.WaitDone()
-
-	fmt.Println("\nAll tasks completed, press Ctrl-C to quit.")
-	select {}
+	return db, nil
 }
 
 func readDictionary(file *os.File) []string {

go.mod

@@ -6,15 +6,21 @@ require (
 	github.com/alexedwards/scs/v2 v2.5.1
 	github.com/evilsocket/islazy v1.11.0
 	github.com/go-chi/chi/v5 v5.0.8
+	github.com/jackc/pgconn v1.14.1
+	github.com/jackc/pgx/v5 v5.4.3
 	github.com/rs/zerolog v1.29.0
-	github.com/stretchr/testify v1.8.4
 )
 
 require (
-	github.com/davecgh/go-spew v1.1.1 // indirect
+	github.com/jackc/chunkreader/v2 v2.0.1 // indirect
+	github.com/jackc/pgio v1.0.0 // indirect
+	github.com/jackc/pgpassfile v1.0.0 // indirect
+	github.com/jackc/pgproto3/v2 v2.3.2 // indirect
+	github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mattn/go-isatty v0.0.17 // indirect
-	github.com/pmezard/go-difflib v1.0.0 // indirect
-	golang.org/x/sys v0.6.0 // indirect
-	gopkg.in/yaml.v3 v3.0.1 // indirect
+	github.com/stretchr/testify v1.8.4 // indirect
+	golang.org/x/crypto v0.9.0 // indirect
+	golang.org/x/sys v0.8.0 // indirect
+	golang.org/x/text v0.9.0 // indirect
 )

pkg/driver/driver.go

@@ -0,0 +1,61 @@
+package driver
+
+import (
+	"database/sql"
+	"time"
+	_ "github.com/jackc/pgconn"
+	_ "github.com/jackc/pgx/v5"
+	_ "github.com/jackc/pgx/v5/stdlib"
+
+
+)
+
+// DB holds the database connection pool
+type DB struct {
+	SQL *sql.DB
+}
+
+var dbConn = &DB{}
+
+const maxOpenDbConn = 10
+const MaxIdleDbConn = 5
+const MaxDbLifetime = 5 * time.Minute
+
+// creates database pool for postgres
+func ConnectSQL(dsn string) (*DB, error) {
+	d, err := NewDatabase(dsn)
+	if err != nil {
+		panic(err)
+	}
+	d.SetMaxOpenConns(maxOpenDbConn)
+	d.SetMaxIdleConns(MaxIdleDbConn)
+	d.SetConnMaxLifetime(MaxDbLifetime)
+	dbConn.SQL = d
+	err = testDB(d)
+	if err != nil {
+		return nil, err
+	}
+	return dbConn, nil
+
+}
+
+// ping db
+func testDB(d *sql.DB) error {
+	err := d.Ping()
+	if err != nil {
+		return err
+	}
+	return nil
+}
+
+// open new db
+func NewDatabase(dsn string) (*sql.DB, error) {
+	db, err := sql.Open("pgx", dsn)
+	if err != nil {
+		return nil, err
+	}
+	if err = db.Ping(); err != nil {
+		return nil, err
+	}
+	return db, nil
+}

pkg/repository/dbrepo/dbrepo.go

@@ -0,0 +1,21 @@
+package dbrepo
+
+import (
+	"database/sql"
+
+	"github.com/CyberRoute/bruter/pkg/config"
+	"github.com/CyberRoute/bruter/pkg/repository"
+)
+
+
+type postgresDBRepo struct {
+	App *config.AppConfig
+	DB *sql.DB
+}
+
+func NewPostgresRepo(conn *sql.DB, a *config.AppConfig) repository.DatabaseRepo {
+	return &postgresDBRepo{
+		App: a,
+		DB: conn,
+	}
+}

pkg/repository/dbrepo/postgres.go

@@ -0,0 +1,5 @@
+package dbrepo
+
+func (m *postgresDBRepo) AllUrls() bool {
+	return true
+}

pkg/repository/repository.go

@@ -0,0 +1,5 @@
+package repository
+
+type DatabaseRepo interface {
+	AllUrls() bool
+}