edit_shell.c

/*
 *  Copyright (C) 2014-2016  Christian Heckendorf <heckendorfc@gmail.com>
 *
 *  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 3 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, see <http://www.gnu.org/licenses/>.
 */

#include "edit_shell.h"
#include "dbact.h"
#include "insert.h"

int yyparse();
TokenList *tlist;
extern commandline_t *fullcmd;
typedef int(*seltypefun)(commandline_t*);
typedef int(*acttypefun)(command_t*,command_t*);

static char *seltypes[]={
	"song",
	"album",
	"artist",
	"playlist",
	"tag",
	NULL
};
void printEditSelect(int id, int type){
	char query[100];
	fprintf(stderr,"type: %d %s\n",type,seltypes[type]);
	snprintf(query,100,"select SelectID from TempSelect where TempID=%d",id);
	doTitleQuery(query,NULL,20);
}

struct selgroup{
	acttypefun func;
	char *name;
};

static int runGenericEditAction(commandline_t *cmd,struct selgroup *selactlist){
	int i;
	command_t *act;

	for(act=cmd->actions;act;act=act->next){
		for(i=0;selactlist[i].name;i++){
			if(strcmp(selactlist[i].name,act->cmd->word)==0){
				selactlist[i].func(cmd->selector,act);
				break;
			}
		}
	}

	return HARP_RET_ERR;
}

static int listSongs(command_t *sel, command_t *act){
	char query[300];
	int exception[10];
	int i;

	for(i=0;i<10;i++)exception[i]=i<5?1:listconf.exception;
	snprintf(query,300,"SELECT SongID, SongTitle, SongTrack, Location, AlbumTitle AS Album, ArtistName AS Artist FROM SongPubInfo WHERE SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	doTitleQuery(query,exception,listconf.maxwidth);

	return HARP_RET_OK;
}

static int editSongTitle(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Song SET Title='%s' WHERE SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int deleteSong(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"DELETE FROM Song WHERE SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editSongAlbumArtist(command_t *sel, command_t *act){
	char query[300];
	int artistid,albumid;
	struct dbitem dbi;
	dbiInit(&dbi);

	if(act->args->next==NULL){
		fprintf(stderr,"albumartist: Two arguments required for this action.\n");
		return HARP_RET_ERR;
	}

	if(act->args->next->words->flag==WORD_DEFAULT)
		artistid=strtol(act->args->next->words->word,NULL,10);
	else
		artistid=getArtist(act->args->next->words->word);

	if(act->args->words->flag==WORD_DEFAULT)
		albumid=strtol(act->args->words->word,NULL,10);
	else
		albumid=getAlbum(act->args->words->word,artistid);

	snprintf(query,300,"UPDATE Song SET AlbumID=%d WHERE SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",albumid,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int songActivation(command_t *sel, command_t *act){
	char query[300];

	/*TODO: test argument to disable toggle */
	snprintf(query,300,"UPDATE Song SET Active=NOT(Active) WHERE SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editSongLocation(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Song SET Location='%s' WHERE SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editSongTrack(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Song SET Track=%s WHERE SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static struct selgroup selsongacts[]={
	{listSongs,"list"},
	{editSongTitle,"title"},
	{editSongTrack,"track"},
	{editSongLocation,"location"},
	{editSongAlbumArtist,"albumArtist"},
	{deleteSong,"delete"},
	{songActivation,"activate"},
	{NULL,NULL}
};
static int runSongEditAction(commandline_t *cmd){
	return runGenericEditAction(cmd,selsongacts);
}

static int listAlbums(command_t *sel, command_t *act){
	char query[300];
	int exception[10];
	int i;

	for(i=0;i<10;i++)exception[i]=i<5?1:listconf.exception;
	snprintf(query,300,"SELECT AlbumID, Title, \"Date\" FROM Album WHERE AlbumID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	doTitleQuery(query,exception,listconf.maxwidth);

	return HARP_RET_OK;
}
static int editAlbumDate(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Album SET Date=%s WHERE AlbumID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editAlbumTitle(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Album SET Title='%s' WHERE AlbumID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editAlbumArtist(command_t *sel, command_t *act){
	char query[300];
	int artistid;

	if(act->args->words->flag==WORD_DEFAULT)
		artistid=strtol(act->args->words->word,NULL,10);
	else
		artistid=getArtist(act->args->words->word);

	snprintf(query,300,"UPDATE AlbumArtist SET ArtistID=%d WHERE AlbumID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",artistid,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static struct selgroup selalbumacts[]={
	{listAlbums,"list"},
	{editAlbumTitle,"title"},
	{editAlbumArtist,"artist"},
	{editAlbumDate,"date"},
	{NULL,NULL}
};
static int runAlbumEditAction(commandline_t *cmd){
	return runGenericEditAction(cmd,selalbumacts);
}

static int listArtists(command_t *sel, command_t *act){
	char query[300];
	int exception[10];
	int i;

	for(i=0;i<10;i++)exception[i]=i<5?1:listconf.exception;
	snprintf(query,300,"SELECT ArtistID, Name FROM Artist WHERE ArtistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	doTitleQuery(query,exception,listconf.maxwidth);

	return HARP_RET_OK;
}

static int editArtistName(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Artist SET Name='%s' WHERE ArtistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static struct selgroup selartistacts[]={
	{listArtists,"list"},
	{editArtistName,"name"},
	{NULL,NULL}
};
static int runArtistEditAction(commandline_t *cmd){
	return runGenericEditAction(cmd,selartistacts);
}

static int listPlaylists(command_t *sel, command_t *act){
	char query[300];
	int exception[10];
	int i;

	for(i=0;i<10;i++)exception[i]=i<5?1:listconf.exception;
	snprintf(query,300,"SELECT PlaylistID, Title FROM Playlist WHERE PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	doTitleQuery(query,exception,listconf.maxwidth);

	return HARP_RET_OK;
}

static int listPlaylistContents(command_t *sel, command_t *act){
	char query[300];
	int exception[10];
	int i;

	for(i=0;i<10;i++)exception[i]=i<5?1:listconf.exception;
	snprintf(query,300,"SELECT PlaylistID, \"Order\" AS \"#\", SongID,SongTitle,AlbumTitle,ArtistName FROM PlaylistSong NATURAL JOIN SongPubInfo WHERE PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d) ORDER BY PlaylistID,\"Order\"",sel->tlid);
	doTitleQuery(query,exception,listconf.maxwidth);

	return HARP_RET_OK;
}

static int editPlaylistName(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Playlist SET Name='%s' WHERE PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int deletePlaylist(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"DELETE FROM PlaylistSong WHERE PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);
	snprintf(query,300,"DELETE FROM Playlist WHERE PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editPlaylistOrder(command_t *sel, command_t *act){
	char query[300];
	int oldO,newO;

	if(act->args->next==NULL){
		fprintf(stderr,"order: Two arguments required for this action.\n");
		return HARP_RET_ERR;
	}

	oldO=strtol(act->args->words->word,NULL,10);
	newO=strtol(act->args->next->words->word,NULL,10);

	if(oldO<newO)
		harp_sqlite3_exec(conn,"CREATE TEMPORARY TRIGGER PLIMv AFTER UPDATE ON PlaylistSong BEGIN "
			"UPDATE PlaylistSong SET \"Order\"=\"Order\"-1 WHERE PlaylistID=OLD.PlaylistID AND \"Order\" BETWEEN OLD.\"Order\" AND NEW.\"Order\" AND PlaylistSongID!=OLD.PlaylistSongID;"
			"END",NULL,NULL,NULL);
	else
		harp_sqlite3_exec(conn,"CREATE TEMPORARY TRIGGER PLIMv AFTER UPDATE ON PlaylistSong BEGIN "
			"UPDATE PlaylistSong SET \"Order\"=\"Order\"+1 WHERE PlaylistID=OLD.PlaylistID AND \"Order\" BETWEEN NEW.\"Order\" AND OLD.\"Order\" AND PlaylistSongID!=OLD.PlaylistSongID;"
			"END",NULL,NULL,NULL);

	snprintf(query,300,"UPDATE PlaylistSong SET \"Order\"=%d WHERE \"Order\"=%d AND PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",newO,oldO,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	harp_sqlite3_exec(conn,"DROP TRIGGER PLIMv",NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editPlaylistAdd(command_t *sel, command_t *act){
	char query[300];
	char *songs_start,*songs_end;

	if(act->args->next==NULL){
		fprintf(stderr,"add: Two arguments required for this action.\n");
		return HARP_RET_ERR;
	}

	if(act->args->words->flag==WORD_DEFAULT){
		songs_start="SongID=";
		songs_end="";
	}
	else{
		songs_start="Song.Title LIKE '%%";
		songs_end="%%'";
	}

	snprintf(query,300,"UPDATE PlaylistSong SET \"Order\"=\"Order\"+1 WHERE \"Order\">=%s AND PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d AND SelectID IN (SELECT PlaylistID FROM PlaylistSong WHERE \"Order\"=%s))",act->args->next->words->word,sel->tlid,act->args->next->words->word);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	snprintf(query,300,"INSERT INTO PlaylistSong (SongID,PlaylistID,\"Order\") SELECT SongID,PlaylistID,%s FROM Song,Playlist WHERE %s%s%s AND PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->next->words->word,songs_start,act->args->words->word,songs_end,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editPlaylistRemove(command_t *sel, command_t *act){
	char query[300];
	char *songs_start,*songs_end;

	if(strcmp(act->cmd->word,"removeOrder")==0){
		songs_start="\"Order\"=";
		songs_end="";
	}
	else{
		if(act->args->words->flag==WORD_DEFAULT){
			songs_start="SongID=";
			songs_end="";
		}
		else{
			songs_start="Song.Title LIKE '%%";
			songs_end="%%'";
		}
	}

	harp_sqlite3_exec(conn,"CREATE TEMPORARY TRIGGER PLIDel AFTER DELETE ON PlaylistSong BEGIN "
			"UPDATE PlaylistSong SET \"Order\"=\"Order\"-1 WHERE PlaylistID=OLD.PlaylistID AND \"Order\">OLD.\"Order\";"
			"END",NULL,NULL,NULL);

	//snprintf(query,300,"DELETE FROM PlaylistSong (SongID,PlaylistID,\"Order\") SELECT SongID,PlaylistID,%s FROM Song,Playlist WHERE %s%s%s AND PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->next->words->word,songs_start,act->args->words->word,songs_end,sel->tlid);
	snprintf(query,300,"DELETE FROM PlaylistSong WHERE PlaylistSongID IN (SELECT PlaylistSongID FROM PlaylistSong NATURAL JOIN Song WHERE %s%s%s AND PlaylistID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d))",songs_start,act->args->words->word,songs_end,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	harp_sqlite3_exec(conn,"DROP TRIGGER PLIDel",NULL,NULL,NULL);

	return HARP_RET_OK;
}

static struct selgroup selplaylistacts[]={
	{listPlaylists,"list"},
	{listPlaylistContents,"contents"},
	{editPlaylistName,"name"},
	{deletePlaylist,"delete"},
	{editPlaylistOrder,"order"},
	{editPlaylistAdd,"add"},
	{editPlaylistRemove,"removeOrder"},
	{editPlaylistRemove,"removeSong"},
	{NULL,NULL}
};
static int runPlaylistEditAction(commandline_t *cmd){
	char query[200];
	snprintf(query,200,"DELETE FROM TempSelect WHERE SelectID=1 AND TempID=%d",cmd->selector->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL); // Failsafe, disable editing of library directly
	return runGenericEditAction(cmd,selplaylistacts);
}

static int listTags(command_t *sel, command_t *act){
	char query[300];
	int exception[10];
	int i;

	for(i=0;i<10;i++)exception[i]=i<5?1:listconf.exception;
	snprintf(query,300,"SELECT TagID, Name FROM Tag WHERE TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	doTitleQuery(query,exception,listconf.maxwidth);

	return HARP_RET_OK;
}

static int listTagContents(command_t *sel, command_t *act){
	char query[300];
	int exception[10];
	int i;

	for(i=0;i<10;i++)exception[i]=i<5?1:listconf.exception;
	snprintf(query,300,"SELECT TagID,SongID,SongTitle,AlbumTitle,ArtistName FROM SongTag NATURAL JOIN SongPubInfo WHERE TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d) ORDER BY TagID",sel->tlid);
	doTitleQuery(query,exception,listconf.maxwidth);

	return HARP_RET_OK;
}

static int editTagName(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"UPDATE Tag SET Name='%s' WHERE TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->args->words->word,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int deleteTag(command_t *sel, command_t *act){
	char query[300];

	snprintf(query,300,"DELETE FROM SongTag WHERE TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);
	snprintf(query,300,"DELETE FROM Tag WHERE TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editTagAdd(command_t *sel, command_t *act){
	char query[300];
	char *songs_start,*songs_end;

	if(act->tlid>=0){
		snprintf(query,300,"INSERT INTO SongTag (SongID,TagID) SELECT SongID,TagID FROM Song,Tag WHERE Song.SongID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d) AND TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",act->tlid,sel->tlid);
		harp_sqlite3_exec(conn,query,NULL,NULL,NULL);
		return HARP_RET_OK;
	}
	if(act->args->words->flag==WORD_DEFAULT){
		songs_start="SongID=";
		songs_end="";
	}
	else{
		songs_start="Song.Title LIKE '%%";
		songs_end="%%'";
	}

	snprintf(query,300,"INSERT INTO SongTag (SongID,TagID) SELECT SongID,TagID FROM Song,Tag WHERE %s%s%s AND TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",songs_start,act->args->words->word,songs_end,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static int editTagRemove(command_t *sel, command_t *act){
	char query[300];
	char *songs_start,*songs_end;

	if(act->args->words->flag==WORD_DEFAULT){
		songs_start="SongID=";
		songs_end="";
	}
	else{
		songs_start="Song.Title LIKE '%%";
		songs_end="%%'";
	}

	snprintf(query,300,"DELETE FROM SongTag WHERE %s%s%s AND TagID IN (SELECT SelectID FROM TempSelect WHERE TempID=%d)",songs_start,act->args->words->word,songs_end,sel->tlid);
	harp_sqlite3_exec(conn,query,NULL,NULL,NULL);

	return HARP_RET_OK;
}

static struct selgroup seltagacts[]={
	{listTags,"list"},
	{listTagContents,"contents"},
	{editTagName,"name"},
	{deleteTag,"delete"},
	{editTagAdd,"add"},
	{editTagRemove,"remove"},
	{NULL,NULL}
};
static int runTagEditAction(commandline_t *cmd){
	return runGenericEditAction(cmd,seltagacts);
}

static seltypefun selacts[]={
	runSongEditAction,
	runAlbumEditAction,
	runArtistEditAction,
	runPlaylistEditAction,
	runTagEditAction,
	NULL
};
static int runEditAction(commandline_t *cmd){
	if(cmd->selector->tltype>=0 && cmd->selector->tltype<SEL_NULL){
		return selacts[cmd->selector->tltype](cmd);
	}
	return HARP_RET_ERR;
}

static void cleanOrphans(){
	// Clean orphaned albums
	harp_sqlite3_exec(conn,"DELETE FROM Album WHERE AlbumID NOT IN (SELECT AlbumID FROM Song)",NULL,NULL,NULL);
	// Clean orphaned albumartists
	harp_sqlite3_exec(conn,"DELETE FROM AlbumArtist WHERE AlbumID NOT IN (SELECT AlbumID FROM Song)",NULL,NULL,NULL);
	// Clean orphaned artists
	harp_sqlite3_exec(conn,"DELETE FROM Artist WHERE ArtistID NOT IN (SELECT ArtistID FROM AlbumArtist)",NULL,NULL,NULL);
}

int editShell(){
	char src[512];
	int ret;

	if(!arglist[AEDIT].subarg){
		printf("Edit> ");
		while(fgets(src,512,stdin)){
			tlist = lex(src);
			fullcmd=NULL;
			ret=yyparse();

			if(!ret)
				ret=runEditAction(fullcmd);
			else
				ret=HARP_RET_ERR;

			printf("Edit> ");
		}
	}
	else{
		tlist = lex(arglist[AEDIT].subarg);
		fullcmd=NULL;
		ret=yyparse();

		if(!ret)
			ret=runEditAction(fullcmd);
		else
			ret=HARP_RET_ERR;
	}

	cleanOrphans();

	return ret;
}