// Copyright (c) 2014, Outercurve Foundation.
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
// - Redistributions of source code must  retain  the  above copyright notice, this
//   list of conditions and the following disclaimer.
//
// - Redistributions in binary form  must  reproduce the  above  copyright  notice,
//   this list of conditions  and  the  following  disclaimer in  the documentation
//   and/or other materials provided with the distribution.
//
// - Neither  the  name  of  the  Outercurve Foundation  nor   the   names  of  its
//   contributors may be used to endorse or  promote  products  derived  from  this
//   software without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
// ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,  BUT  NOT  LIMITED TO, THE IMPLIED
// WARRANTIES  OF  MERCHANTABILITY   AND  FITNESS  FOR  A  PARTICULAR  PURPOSE  ARE
// DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
// ANY DIRECT, INDIRECT, INCIDENTAL,  SPECIAL,  EXEMPLARY, OR CONSEQUENTIAL DAMAGES
// (INCLUDING, BUT NOT LIMITED TO,  PROCUREMENT  OF  SUBSTITUTE  GOODS OR SERVICES;
// LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)  HOWEVER  CAUSED AND ON
// ANY  THEORY  OF  LIABILITY,  WHETHER  IN  CONTRACT,  STRICT  LIABILITY,  OR TORT
// (INCLUDING NEGLIGENCE OR OTHERWISE)  ARISING  IN  ANY WAY OUT OF THE USE OF THIS
// SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

using System;
using System.IO;
using System.Data;
using System.Configuration;
using System.Collections;
using System.Web;
using System.Web.Security;
using System.Web.UI;
using System.Web.UI.WebControls;
using System.Web.UI.WebControls.WebParts;
using System.Web.UI.HtmlControls;

using WebsitePanel.Providers.Database;

namespace WebsitePanel.Portal
{
    public partial class SqlRestoreDatabase : WebsitePanelModuleBase
    {
        protected void Page_Load(object sender, EventArgs e)
        {
            if (!IsPostBack)
            {
                BindDatabase();
            }
        }

        private void BindDatabase()
        {
            try
            {
                SqlDatabase database = ES.Services.DatabaseServers.GetSqlDatabase(PanelRequest.ItemID);
                litDatabaseName.Text = database.Name;
                fileLookup.SelectedFile = "";
                fileLookup.PackageId = database.PackageId;

                ToggleControls();
            }
            catch (Exception ex)
            {
                ShowErrorMessage("SQL_GET_DATABASE", ex);
                return;
            }
        }

        private void ToggleControls()
        {
            cellFile.Visible = radioFile.Checked;
            cellUploadFile.Visible = radioUpload.Checked;
        }

        private void RestoreDatabase()
        {
            try
            {
                string[] uploadedFiles = null;
                string[] packageFiles = null;

                if (radioUpload.Checked)
                {
                    if (uploadFile.PostedFile.FileName != "")
                    {
                        Stream stream = uploadFile.PostedFile.InputStream;

                        // save uploaded file
                        int FILE_BUFFER_LENGTH = 5000000;
                        string path = null;
                        int readBytes = 0;
                        string fileName = Path.GetFileName(uploadFile.PostedFile.FileName);

                        int offset = 0;
                        do
                        {
                            // read input stream
                            byte[] buffer = new byte[FILE_BUFFER_LENGTH];
                            readBytes = stream.Read(buffer, 0, FILE_BUFFER_LENGTH);

                            if (readBytes < FILE_BUFFER_LENGTH)
                                Array.Resize<byte>(ref buffer, readBytes);

                            // write remote backup file
                            string tempPath = ES.Services.DatabaseServers.AppendSqlBackupBinaryChunk(PanelRequest.ItemID, fileName, path, buffer);
                            if (path == null)
                                path = tempPath;

                            offset += FILE_BUFFER_LENGTH;
                        }
                        while (readBytes == FILE_BUFFER_LENGTH);

                        uploadedFiles = new string[] { path };
                    }
                }
                else
                {
                    // package files
                    packageFiles = new string[] { fileLookup.SelectedFile };
                }

                int result = ES.Services.DatabaseServers.RestoreSqlDatabase(PanelRequest.ItemID, uploadedFiles, packageFiles);
                if (result < 0)
                {
                    ShowResultMessage(result);
                    return;
                }
            }
            catch (Exception ex)
            {
                ShowErrorMessage("SQL_RESTORE_DATABASE", ex);
                return;
            }

            RedirectBack();
        }

        protected void btnRestore_Click(object sender, EventArgs e)
        {
            RestoreDatabase();
        }
        protected void btnCancel_Click(object sender, EventArgs e)
        {
            RedirectBack();
        }

        private void RedirectBack()
        {
            Response.Redirect(EditUrl("ItemID", PanelRequest.ItemID.ToString(), "edit_item",
                PortalUtils.SPACE_ID_PARAM + "=" + PanelSecurity.PackageId));
        }
        protected void radioUpload_CheckedChanged(object sender, EventArgs e)
        {
            ToggleControls();
        }
    }
}