1 files changed, 324 insertions, 0 deletions
diff --git a/src/database.c b/src/database.c
new file mode 100644
index 000000000..75038839c
--- /dev/null
+++ b/src/database.c
@@ -0,0 +1,324 @@
+/* the Music Player Daemon (MPD)
+ * Copyright (C) 2003-2007 by Warren Dukes (warren.dukes@gmail.com)
+ * Copyright (C) 2008 Max Kellermann <max@duempel.org>
+ * This project's homepage is: http://www.musicpd.org
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include "database.h"
+#include "directory.h"
+#include "song.h"
+#include "conf.h"
+#include "log.h"
+#include "ls.h"
+#include "path.h"
+#include "stats.h"
+#include "utils.h"
+#include "dbUtils.h"
+#include "update.h"
+
+#include <assert.h>
+#include <string.h>
+
+static struct directory *music_root;
+
+static time_t directory_dbModTime;
+
+void directory_init(void)
+{
+	music_root = newDirectory(NULL, NULL);
+	updateDirectory(music_root);
+	stats.numberOfSongs = countSongsIn(NULL);
+	stats.dbPlayTime = sumSongTimesIn(NULL);
+}
+
+void directory_finish(void)
+{
+	freeDirectory(music_root);
+}
+
+struct directory *
+directory_get_root(void)
+{
+	assert(music_root != NULL);
+
+	return music_root;
+}
+
+struct directory *
+getDirectory(const char *name)
+{
+	if (name == NULL)
+		return music_root;
+
+	return getSubDirectory(music_root, name);
+}
+
+struct song *
+getSongFromDB(const char *file)
+{
+	struct song *song = NULL;
+	struct directory *directory;
+	char *dir = NULL;
+	char *duplicated = xstrdup(file);
+	char *shortname = strrchr(duplicated, '/');
+
+	DEBUG("get song: %s\n", file);
+
+	if (!shortname) {
+		shortname = duplicated;
+	} else {
+		*shortname = '\0';
+		++shortname;
+		dir = duplicated;
+	}
+
+	if (!(directory = getDirectory(dir)))
+		goto out;
+	if (!(song = songvec_find(&directory->songs, shortname)))
+		goto out;
+	assert(song->parent == directory);
+
+out:
+	free(duplicated);
+	return song;
+}
+
+int
+traverseAllIn(const char *name,
+	      int (*forEachSong) (struct song *, void *),
+	      int (*forEachDir) (struct directory *, void *), void *data)
+{
+	struct directory *directory;
+
+	if ((directory = getDirectory(name)) == NULL) {
+		struct song *song;
+		if ((song = getSongFromDB(name)) && forEachSong) {
+			return forEachSong(song, data);
+		}
+		return -1;
+	}
+
+	return traverseAllInSubDirectory(directory, forEachSong, forEachDir,
+					 data);
+}
+
+int printDirectoryInfo(struct client *client, const char *name)
+{
+	struct directory *directory;
+
+	if ((directory = getDirectory(name)) == NULL)
+		return -1;
+
+	return directory_print(client, directory);
+}
+
+static char *getDbFile(void)
+{
+	ConfigParam *param = parseConfigFilePath(CONF_DB_FILE, 1);
+
+	assert(param);
+	assert(param->value);
+
+	return param->value;
+}
+
+int checkDirectoryDB(void)
+{
+	struct stat st;
+	char *dbFile = getDbFile();
+
+	/* Check if the file exists */
+	if (access(dbFile, F_OK)) {
+		/* If the file doesn't exist, we can't check if we can write
+		 * it, so we are going to try to get the directory path, and
+		 * see if we can write a file in that */
+		char dirPath[MPD_PATH_MAX];
+		parent_path(dirPath, dbFile);
+		if (*dirPath == '\0')
+			strcpy(dirPath, "/");
+
+		/* Check that the parent part of the path is a directory */
+		if (stat(dirPath, &st) < 0) {
+			ERROR("Couldn't stat parent directory of db file "
+			      "\"%s\": %s\n", dbFile, strerror(errno));
+			return -1;
+		}
+
+		if (!S_ISDIR(st.st_mode)) {
+			ERROR("Couldn't create db file \"%s\" because the "
+			      "parent path is not a directory\n", dbFile);
+			return -1;
+		}
+
+		/* Check if we can write to the directory */
+		if (access(dirPath, R_OK | W_OK)) {
+			ERROR("Can't create db file in \"%s\": %s\n", dirPath,
+			      strerror(errno));
+			return -1;
+		}
+
+		return 0;
+	}
+
+	/* Path exists, now check if it's a regular file */
+	if (stat(dbFile, &st) < 0) {
+		ERROR("Couldn't stat db file \"%s\": %s\n", dbFile,
+		      strerror(errno));
+		return -1;
+	}
+
+	if (!S_ISREG(st.st_mode)) {
+		ERROR("db file \"%s\" is not a regular file\n", dbFile);
+		return -1;
+	}
+
+	/* And check that we can write to it */
+	if (access(dbFile, R_OK | W_OK)) {
+		ERROR("Can't open db file \"%s\" for reading/writing: %s\n",
+		      dbFile, strerror(errno));
+		return -1;
+	}
+
+	return 0;
+}
+
+int writeDirectoryDB(void)
+{
+	FILE *fp;
+	char *dbFile = getDbFile();
+	struct stat st;
+
+	DEBUG("removing empty directories from DB\n");
+	deleteEmptyDirectoriesInDirectory(music_root);
+
+	DEBUG("sorting DB\n");
+
+	sortDirectory(music_root);
+
+	DEBUG("writing DB\n");
+
+	fp = fopen(dbFile, "w");
+	if (!fp) {
+		ERROR("unable to write to db file \"%s\": %s\n",
+		      dbFile, strerror(errno));
+		return -1;
+	}
+
+	/* block signals when writing the db so we don't get a corrupted db */
+	fprintf(fp, "%s\n", DIRECTORY_INFO_BEGIN);
+	fprintf(fp, "%s%s\n", DIRECTORY_MPD_VERSION, VERSION);
+	fprintf(fp, "%s%s\n", DIRECTORY_FS_CHARSET, getFsCharset());
+	fprintf(fp, "%s\n", DIRECTORY_INFO_END);
+
+	if (writeDirectoryInfo(fp, music_root) < 0) {
+		ERROR("Failed to write to database file: %s\n",
+		      strerror(errno));
+		while (fclose(fp) && errno == EINTR);
+		return -1;
+	}
+
+	while (fclose(fp) && errno == EINTR);
+
+	if (stat(dbFile, &st) == 0)
+		directory_dbModTime = st.st_mtime;
+
+	return 0;
+}
+
+int readDirectoryDB(void)
+{
+	FILE *fp = NULL;
+	char *dbFile = getDbFile();
+	struct stat st;
+
+	if (!music_root)
+		music_root = newDirectory(NULL, NULL);
+	while (!(fp = fopen(dbFile, "r")) && errno == EINTR) ;
+	if (fp == NULL) {
+		ERROR("unable to open db file \"%s\": %s\n",
+		      dbFile, strerror(errno));
+		return -1;
+	}
+
+	/* get initial info */
+	{
+		char buffer[100];
+		int bufferSize = 100;
+		int foundFsCharset = 0;
+		int foundVersion = 0;
+
+		if (!myFgets(buffer, bufferSize, fp))
+			FATAL("Error reading db, fgets\n");
+		if (0 == strcmp(DIRECTORY_INFO_BEGIN, buffer)) {
+			while (myFgets(buffer, bufferSize, fp) &&
+			       0 != strcmp(DIRECTORY_INFO_END, buffer)) {
+				if (!prefixcmp(buffer, DIRECTORY_MPD_VERSION))
+				{
+					if (foundVersion)
+						FATAL("already found version in db\n");
+					foundVersion = 1;
+				} else if (!prefixcmp(buffer,
+				                      DIRECTORY_FS_CHARSET)) {
+					char *fsCharset;
+					char *tempCharset;
+
+					if (foundFsCharset)
+						FATAL("already found fs charset in db\n");
+
+					foundFsCharset = 1;
+
+					fsCharset = &(buffer[strlen(DIRECTORY_FS_CHARSET)]);
+					if ((tempCharset = getConfigParamValue(CONF_FS_CHARSET))
+					    && strcmp(fsCharset, tempCharset)) {
+						WARNING("Using \"%s\" for the "
+							"filesystem charset "
+							"instead of \"%s\"\n",
+							fsCharset, tempCharset);
+						WARNING("maybe you need to "
+							"recreate the db?\n");
+						setFsCharset(fsCharset);
+					}
+				} else {
+					FATAL("directory: unknown line in db info: %s\n",
+					     buffer);
+				}
+			}
+		} else {
+			ERROR("db info not found in db file\n");
+			ERROR("you should recreate the db using --create-db\n");
+			while (fclose(fp) && errno == EINTR) ;
+			return -1;
+		}
+	}
+
+	DEBUG("reading DB\n");
+
+	readDirectoryInfo(fp, music_root);
+	while (fclose(fp) && errno == EINTR) ;
+
+	stats.numberOfSongs = countSongsIn(NULL);
+	stats.dbPlayTime = sumSongTimesIn(NULL);
+
+	if (stat(dbFile, &st) == 0)
+		directory_dbModTime = st.st_mtime;
+
+	return 0;
+}
+
+time_t getDbModTime(void)
+{
+	return directory_dbModTime;
+}