// frmContactEditor-Picture.cpp - frmContactEditor Picture tab subroutines.
//
// (c) 2012-2016 Xestia Software Development.
//
// This file is part of Xestia Address Book.
//
// Xestia Address Book 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, version 3 of the license.
//
// Xestia Address Book 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 Xestia Address Book. If not, see <http://www.gnu.org/licenses/>

#include <wx/file.h>
#include "frmContactEditor.h"
#include "frmContactEditorPicture.h"
#include "../common/base64.h"

void frmContactEditor::AddPicture( wxCommandEvent& event )
{
    int intResult = 0;
    
    frmContactEditorPicture *frameCEPicture = new frmContactEditorPicture ( this );
    frameCEPicture->SetEditorMode(FALSE);
    intResult = GetLastInt(&ContactEditorData.PicturesList);
    frameCEPicture->SetupPointers(&ContactEditorData.PicturesList,
                                  &ContactEditorData.PicturesListAltID,
                                  &ContactEditorData.PicturesListPID,
                                  &ContactEditorData.PicturesListType,
                                  &ContactEditorData.PicturesListPicEncType,
                                  &ContactEditorData.PicturesListPictureType,
                                  &ContactEditorData.PicturesListTokens,
                                  &ContactEditorData.PicturesListMediatype,
                                  &ContactEditorData.PicturesListPref,
                                  lboPictures,
                                  (intValueSeek),
				  TRUE);
    frameCEPicture->ShowModal();
    delete frameCEPicture;
    frameCEPicture = NULL;
}

void frmContactEditor::ModifyPicture( wxCommandEvent& event )
{
    long longSelected = -1;
    int intSelectedData = 0;
    
    if (!GetSelectedItem(lboPictures,
                         &longSelected,
                         &intSelectedData)){
        return;
    }
    
    frmContactEditorPicture *frameCEPicture = new frmContactEditorPicture ( this );
    frameCEPicture->SetupPointers(&ContactEditorData.PicturesList,
                                  &ContactEditorData.PicturesListAltID,
                                  &ContactEditorData.PicturesListPID,
                                  &ContactEditorData.PicturesListType,
                                  &ContactEditorData.PicturesListPicEncType,
                                  &ContactEditorData.PicturesListPictureType,
                                  &ContactEditorData.PicturesListTokens,
                                  &ContactEditorData.PicturesListMediatype,
                                  &ContactEditorData.PicturesListPref,
                                  lboPictures,
                                  intSelectedData,
				  TRUE);
    frameCEPicture->SetEditorMode(TRUE);
    frameCEPicture->ShowModal();
    delete frameCEPicture;
    frameCEPicture = NULL;
    
}

void frmContactEditor::DeletePicture( wxCommandEvent& event )
{
    long longSelected = -1;
    int intSelectedData = 0;
    
    if (!GetSelectedItem(lboPictures,
                         &longSelected,
                         &intSelectedData)){
        return;
    }
    
    lboPictures->DeleteItem(longSelected);
    
    DeleteMapData(intSelectedData,
                  &ContactEditorData.PicturesList, &ContactEditorData.PicturesListAltID, &ContactEditorData.PicturesListPID, &ContactEditorData.PicturesListType,
                  &ContactEditorData.PicturesListPicEncType, &ContactEditorData.PicturesListPictureType, &ContactEditorData.PicturesListTokens,
                  &ContactEditorData.PicturesListMediatype, &ContactEditorData.PicturesListPref);
    
    wxMemoryInputStream istream(misc_emptyimage_png, sizeof(misc_emptyimage_png));
    wxImage misc_emptyimage_png(istream, wxBITMAP_TYPE_PNG);
    PictureImage = (wxBitmap)misc_emptyimage_png;
    
    imgPicture->SetBitmap(PictureImage);
}

void frmContactEditor::LoadPicture( wxListEvent& event )
{
    
    long longSelected = -1;
    int intSelectedData = 0;
    
    if (!GetSelectedItem(lboPictures,
                         &longSelected,
                         &intSelectedData)){
        return;
    }
    
    std::string base64dec;
    
    std::map<int,std::string>::iterator initier;
    std::map<int,wxString>::iterator pictypeiter;
    
    initier = ContactEditorData.PicturesList.find(intSelectedData);
    pictypeiter = ContactEditorData.PicturesListPictureType.find(intSelectedData);
    
    base64dec = base64_decode(initier->second);
    
    wxMemoryInputStream istream(base64dec.c_str(), (size_t)base64dec.size());
    wxImage photo;
    //wxMemoryInputStream istream(base64dec, (size_t)base64dec.size());
    
    if (pictypeiter->second == wxT("image/jpeg")){
        
        if (!photo.LoadFile(istream, wxBITMAP_TYPE_JPEG)){
        
            return;
        
        }
        
    } else if (pictypeiter->second == wxT("image/png")){
        
        if (!photo.LoadFile(istream, wxBITMAP_TYPE_PNG)){
            
            return;
            
        }
        
    } else if (pictypeiter->second == wxT("image/gif")){
        
        if (!photo.LoadFile(istream, wxBITMAP_TYPE_GIF)){
            
            return;
            
        }
        
    } else {
    
        return;
    
    }
    
    PictureImage = (wxBitmap)photo;
    imgPicture->SetBitmap(PictureImage);
    
}

void frmContactEditor::SavePicture( wxCommandEvent &event )
{

	long longSelected = -1;
	int intSelectedData = 0;
    
	if (!GetSelectedItem(lboPictures,
                         &longSelected,
                         &intSelectedData)){
		
		return;
		
	}
	
	wxString FileTypes;
	
	wxString FinalFilename;
	
	std::map<int,std::string>::iterator initier;
	std::map<int,wxString>::iterator pictypeiter;
    
	initier = ContactEditorData.PicturesList.find(intSelectedData);
	pictypeiter = ContactEditorData.PicturesListPictureType.find(intSelectedData);
	
	if (pictypeiter->second == wxT("image/jpeg")){
    
		FileTypes.Append("JPEG Image (*.jpeg)|*.jpeg");
	
	} else if (pictypeiter->second == wxT("image/png")){

		FileTypes.Append("PNG Image (*.png)|*.png");        
        
	} else if (pictypeiter->second == wxT("image/gif")){
        
		FileTypes.Append("GIF Image (*.gif)|*.gif"); 
	
	} else {

		wxMessageBox(_("Picture format for saving is not supported."), _("Picture format unsupported"), wxOK, this);
		return;

	}
	
	// Open up the dialog to save the picture.
	
	wxFileDialog ExportDlg(this, wxT("Save Picture"), wxT(""), wxT(""),
		FileTypes, wxFD_SAVE|wxFD_OVERWRITE_PROMPT);
	
	if (ExportDlg.ShowModal() == wxID_CANCEL){
	
		return;
	
	}
	
	// Write out the picture to the file.
	
	wxFile PictureFile;
	
	if (PictureFile.Open(ExportDlg.GetPath(), wxFile::write)){
	
		std::string base64dec;
		
		initier = ContactEditorData.PicturesList.find(intSelectedData);
		base64dec = base64_decode(initier->second);
		
		PictureFile.Write(base64dec.c_str(), (size_t)base64dec.size());

		PictureFile.Close();
	
	} else {
	
		wxMessageBox(_("An error occured whilst saving the picture. Check that you have free space and permissions to write the picture at this location."), _("Picture save error"), wxOK, this);
		return;
	
	}
			
}