handler.go

package macouidatabase

import (
	"fmt"
	"os"
	"path/filepath"
	"runtime"
)

type database string

func SetDatabase(dtype int) database {
    _, filename, _, _ := runtime.Caller(0)
    dir := filepath.Dir(filename)

    var relPath string
    switch dtype {
    case SqliteDatabase:
        relPath = "./LISTS/master_oui.db"
    case CsvDatabase:
        relPath = "./LISTS/master_oui.csv"
    case JsonDatabase:
        relPath = "./LISTS/master_oui.json"
    default:
        return ""
    }
    
    fullPath := filepath.Join(dir, relPath)
    
    if _, err := os.Stat(fullPath); os.IsNotExist(err) {
        pwd, _ := os.Getwd()
        fullPath = filepath.Join(pwd, relPath)
    }

    return database(fullPath)
}

func LoadDatabase(instance database) (db DatabaseModel, err error) {
	switch filepath.Ext(string(instance)) {
	case ".db":
		db.path = string(instance)
		db.Type = "SQLITE"
		db.Handler = &MacSqlite{
			path: string(instance),
		}
	case ".json":
		db.path = string(instance)
		db.Type = "JSON"
		db.Handler = &MacJson{
			path: string(instance),
		}
	case ".csv":
		db.path = string(instance)
		db.Type = "CSV"
		db.Handler = &MacCsv{
			path: string(instance),
		}
	default:
		return db, fmt.Errorf("unsupported database type")
	}
	return
}