feat: 0.1.2

1. login page && auth required
  2. file dir cleanup

internal/pkg/db/client.go

@@ -0,0 +1,51 @@
+package db
+
+import (
+	"au99999/internal/opt"
+	"au99999/pkg/tool"
+	"context"
+
+	"gorm.io/gorm"
+)
+
+var Default *Client
+
+type DBType string
+
+const (
+	DBTypeSqlite   = "sqlite"
+	DBTypeMysql    = "mysql"
+	DBTypePostgres = "postgres"
+)
+
+type Client struct {
+	ctx    context.Context
+	cli    *gorm.DB
+	dbType DBType
+}
+
+func (c *Client) Type() DBType {
+	return c.dbType
+}
+
+func (c *Client) Session(ctxs ...context.Context) *gorm.DB {
+	var ctx context.Context
+	if len(ctxs) > 0 && ctxs[0] != nil {
+		ctx = ctxs[0]
+	} else {
+		ctx = tool.Timeout(30)
+	}
+
+	session := c.cli.Session(&gorm.Session{Context: ctx})
+
+	if opt.Cfg.Debug {
+		session = session.Debug()
+	}
+
+	return session
+}
+
+func (c *Client) Close() {
+	d, _ := c.cli.DB()
+	d.Close()
+}

internal/pkg/db/new.go

@@ -0,0 +1,65 @@
+package db
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"github.com/glebarez/sqlite"
+	"gorm.io/driver/mysql"
+	"gorm.io/driver/postgres"
+	"gorm.io/gorm"
+)
+
+const tip = `example:
+for sqlite   -> sqlite::<filepath>
+				sqlite::data.sqlite
+				sqlite::/data/data.db
+for mysql    -> mysql::<gorm_dsn>
+				mysql::user:pass@tcp(127.0.0.1:3306)/dbname?charset=utf8mb4&parseTime=True&loc=Local
+for postgres -> postgres::<gorm_dsn>
+				postgres::host=localhost user=gorm password=gorm dbname=gorm port=9920 sslmode=disable TimeZone=Asia/Shanghai
+`
+
+func New(ctx context.Context, uri string) (*Client, error) {
+	parts := strings.SplitN(uri, "::", 2)
+
+	if len(parts) != 2 {
+		return nil, fmt.Errorf("db.Init: db uri invalid\n%s", tip)
+	}
+
+	c := &Client{}
+
+	var (
+		err error
+		dsn = parts[1]
+	)
+
+	switch parts[0] {
+	case "sqlite":
+		c.dbType = DBTypeSqlite
+		c.cli, err = gorm.Open(sqlite.Open(dsn))
+	case "mysql":
+		c.dbType = DBTypeMysql
+		c.cli, err = gorm.Open(mysql.Open(dsn))
+	case "postgres":
+		c.dbType = DBTypePostgres
+		c.cli, err = gorm.Open(postgres.Open(dsn))
+	default:
+		return nil, fmt.Errorf("db type only support: [sqlite, mysql, postgres], unsupported db type: %s", parts[0])
+	}
+
+	if err != nil {
+		return nil, fmt.Errorf("db.Init: open %s with dsn:%s, err: %w", parts[0], dsn, err)
+	}
+
+	return c, nil
+}
+
+func Init(ctx context.Context, uri string) (err error) {
+	if Default, err = New(ctx, uri); err != nil {
+		return err
+	}
+
+	return nil
+}