Method getFolderDB

internal/db/sqlite/db_folderdb.go:28–93

(folder string, create bool)

var errNoSuchFolder = errors.New("no such folder")

func (s *DB) getFolderDB(folder string, create bool) (*folderDB, error) {
	// Check for an already open database
	s.folderDBsMut.RLock()
	fdb, ok := s.folderDBs[folder]
	s.folderDBsMut.RUnlock()
	if ok {
		return fdb, nil
	}

	// Check for an existing database. If we're not supposed to create the
	// folder, we don't move on if it doesn't already have a database name.
	var dbns sql.NullString
	if err := s.stmt(`
		SELECT database_name FROM folders
		WHERE folder_id = ?
	`).Get(&dbns, folder); err != nil && !errors.Is(err, sql.ErrNoRows) {
		return nil, wrap(err)
	}

	var dbName string
	if dbns.Valid {
		dbName = dbns.String
	}
	if dbName == "" && !create {
		return nil, errNoSuchFolder
	}

	// Create a folder ID and database if it does not already exist
	s.folderDBsMut.Lock()
	defer s.folderDBsMut.Unlock()
	if fdb, ok := s.folderDBs[folder]; ok {
		return fdb, nil
	}

	if dbName == "" {
		// First time we want to access this folder, need to create a new
		// folder ID

		s.updateLock.Lock()
		defer s.updateLock.Unlock()

		idx, err := s.folderIdxLocked(folder)
		if err != nil {
			return nil, wrap(err)
		}

		// The database name is the folder index ID and a random slug.
		slug := strings.ToLower(rand.String(8))
		dbName = fmt.Sprintf("folder.%04x-%s.db", idx, slug)
		if _, err := s.stmt(`UPDATE folders SET database_name = ? WHERE idx = ?`).Exec(dbName, idx); err != nil {
			return nil, wrap(err, "set name")
		}
	}

	slog.Debug("Folder database opened", "folder", folder, "db", dbName)
	path := dbName
	if !filepath.IsAbs(path) {
		path = filepath.Join(s.pathBase, dbName)