/*
Copyright (c) 2012 "LliureX"
Framebuffer boot splash

This file is part of Bootcookie.

Bootcookie 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 <fcntl.h>
#include <sys/stat.h>
#include <unistd.h>
#include <iostream>
#include <string>

using namespace std;


/**
 * displays usage on screen
 */ 
void print_usage()
{
	cout<<"Usage:"<<endl;
	cout<<"bootcookie_ctrl [set [boot | shutdown | fdisk] | quit]"<<endl;
}

/**
 * Sends a command
 */ 
void send_command(char cmd)
{
	int fd;
    char * myfifo = "/tmp/bootcookie";
    
    
     /* open, read, and display the message from the FIFO */
    fd = open(myfifo, O_WRONLY);
    
    if(fd==-1)
    {
		cerr<<"Error: Failed to open FIFO "<<myfifo<<endl;
	}
    else
    {
		if(write(fd, &cmd, 1)==-1)
			cerr<<"Error: Failed to write on FIFO "<<myfifo<<endl;
    	close(fd);
	}
}

/**
 * Main
 */ 
int main(int argc,char * argv[])
{
   
	string str;
	
	if(argc>1)
	{
		cout<<"command:"<<argv[0]<<":"<<argv[1]<<endl;
		str=string(argv[1]);
		if(str=="set")
		{
			if(argc>2)
			{
				str=string(argv[2]);
				if(str=="boot")
				{
					send_command('B');
				}
				if(str=="shutdown")
				{
					send_command('S');
				}
				if(str=="fdisk")
				{
					send_command('D');
				}
			}
			else
			{
				print_usage();
			}
		}
		
		if(str=="quit")
		{
			send_command('Q');
		}
	}
	else
	{
		print_usage();
	}
	
	

   

    return 0;
}