/* Tools - Tools for Museek (muscan)
 *
 * Copyright (C) 2003-2004 Hyriand <hyriand@thegraveyard.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 <config.h>

#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdlib.h>
#include <unistd.h>
#include <dirent.h>

extern "C" {
# include "mp3.h"
}

#if 0
#ifndef HAVE_SCANDIR
#  include "scandir.hh"
#endif
#endif

#ifdef HAVE_VORBIS
# include <vorbis/vorbisfile.h>
#endif

#include <moohelp/DirEntry.hh>
#include <moohelp/string_ext.hh>

#define MULOG_DOMAIN "Muscan.SC"
#include <moohelp/Mulog.hh>

#include <iostream>

using std::map;
using std::vector;
using std::string;
using std::cerr;
using std::cout;
using std::endl;

#ifdef HAVE_VORBIS
# ifdef RELAYED_LIBVORBISFILE
extern int libvorbisfile_is_present;
# else
static int libvorbisfile_is_present = 1;
# endif
#endif

int Scanner_Verbosity = 0;

class DirScanner : public DirEntry {
public:
	DirScanner(bool _f = true) : DirEntry(_f) { };
	DirScanner(const string& _p) : DirEntry(_p) { };
	
	void add(const string& path);
	
	DirEntry* new_folder(bool fake);
	DirEntry* new_folder(const string& path);
	void scan(const struct stat* = NULL);
	
private:
	FileEntry scan_file(const string&);
	void real_scan();
};


void DirScanner::add(const string& path) {
	if(folders.find(path) == folders.end())
		folders[path] = new_folder(path);
}

DirEntry* DirScanner::new_folder(bool fake) {
	CT("new_folder %i", fake);
	
	return new DirScanner(fake);
}

DirEntry* DirScanner::new_folder(const string& path) {
	CT("new_folder %s", path.c_str());
	
	return new DirScanner(path);
}


void DirScanner::scan(const struct stat* s) {
	CT("scan <...>");
	
	bool uptodate = false;
	if (s != NULL) {
		uptodate = s->st_mtime == mtime;
		mtime = s->st_mtime;
	}
	if (! fake && ! uptodate) {
		real_scan();
		return;
	}
	
	if(Scanner_Verbosity > 1 && ! fake)
		cout << "Skipping " << path << endl;
	
	map<string, DirEntry*>::iterator dit = folders.begin();
	for(; dit != folders.end(); ++dit) {
		struct stat s2;
		if (stat((*dit).first.c_str(), &s2)) {
			(*dit).second->files.clear();
			for(map<string,DirEntry*>::iterator it = (*dit).second->folders.begin(); it != (*dit).second->folders.end(); ++it)
				delete (*it).second;
			(*dit).second->folders.clear();
			continue;
		}
		static_cast<DirScanner*>((*dit).second)->scan(&s2);
	}
}


FileEntry DirScanner::scan_file(const string& path) {
	CT("scan file %s", path.c_str());
	
	if(Scanner_Verbosity > 2)
		cout << "Identifying " << path << endl;
	
	FileEntry fe;
	mp3info info;
	int l = path.length();
	if (l > 4)
		if(tolower(path.substr(l-4, l)) == ".mp3") {
			if (mp3_scan(path.c_str(), &info)) {
				fe.attrs.push_back(info.bitrate);
				fe.attrs.push_back(info.length);
				fe.attrs.push_back(info.vbr);
				fe.ext = "mp3";
			} else {
				cerr << "Invalid mp3 " << path << endl;
			}
#ifdef HAVE_VORBIS
		} else if(libvorbisfile_is_present && tolower(path.substr(l-4, l)) == ".ogg") {
			FILE *f = fopen(path.c_str(), "r");
			if (f == NULL)
				return fe;
			OggVorbis_File ovf;
			if(ov_open(f, &ovf, NULL, 0) != 0) {
				fclose(f);
				return fe;
			}
			vorbis_info *vinfo = ov_info(&ovf, -1);
			if (vinfo != NULL) {
				long bitrate = ov_bitrate(&ovf, -1);
				fe.attrs.push_back(bitrate / 1000);
				fe.attrs.push_back((int)ov_time_total(&ovf, -1));
				if (bitrate == vinfo->bitrate_nominal == vinfo->bitrate_upper == vinfo->bitrate_lower)
					fe.attrs.push_back(0);
				else
					fe.attrs.push_back(1);
				fe.ext = "mp3";
			}
			ov_clear(&ovf);
#endif // HAVE_VORBIS
		}
	return fe;
}

void DirScanner::real_scan() {
	CT("real_scan");
	
	if(Scanner_Verbosity > 0)
		cout << "Scanning " << path << endl;
	
	files.clear();

	struct dirent **temp;
	struct stat s;
	int n;
	char *x = strdup(path.c_str());
	if((n = scandir(x, &temp, NULL, NULL)) < 0) {
		free(x);
		folders.clear();
		return;
	}
	free(x);
	
	map<string, DirEntry*>newfolders;

	while (n--) {
		string fn = temp[n]->d_name,
			full = path + "/" + fn;
		free(temp[n]);

		if ((fn == ".") || (fn == "..") || (stat(full.c_str(), &s) != 0))
			continue;

		if(S_ISREG(s.st_mode)) {
			FileEntry fe = scan_file(full);
			fe.size = s.st_size;
			files[fn] = fe;
		} else if (S_ISDIR(s.st_mode)) {
			map<string, DirEntry*>::iterator dit = folders.find(full);
			if (dit != folders.end()) {
				newfolders[full] = (*dit).second;
				static_cast<DirScanner*>(newfolders[full])->scan(&s);
			} else {
				DirEntry* de = new_folder(full);
				static_cast<DirScanner*>(de)->scan(&s);
				newfolders[full] = de;
			}
			
		}
	}
	folders = newfolders;
	free(temp);
}