package models

import (
	"database/sql"

	// import sqlite driver
	_ "github.com/mattn/go-sqlite3"
)

// InitDB returns sqlite3 connection
func InitDB(dataSource string) (*sql.DB, error) {
	db, err := sql.Open("sqlite3", dataSource)
	if err != nil {
		return nil, err
	}

	return db, nil
}