///
///  This source code is freeware and is provided on an "as is" basis without warranties of any kind, 
///  whether express or implied, including without limitation warranties that the code is free of defect, 
///  fit for a particular purpose or non-infringing.  The entire risk as to the quality and performance of 
///  the code is with the user.
///

using System;
using System.Collections.Generic;
using System.Text;
using Microsoft.SharePoint;
using Microsoft.SharePoint.Publishing;

namespace sample_name.SupportTools
{
    class FixPageLayout
    {
        const string USAGE = "Usage: FixPageLayout.exe <url-to-sitecollection>";
        const string MASTER_PAGE_LIB = "_catalogs/masterpage";

        static void Main(string[] args)
        {
            if (args.Length < 1)
            {
                Console.WriteLine(USAGE);
                return;
            }

            string SiteCollectionUrl = args[0];

            if (!Uri.IsWellFormedUriString(SiteCollectionUrl, UriKind.Absolute))
            {
                Console.WriteLine(USAGE);
                return;
            }

            try
            {

                // obtain the site collection
                using (SPSite site = new SPSite(SiteCollectionUrl))
                {

                    // get the servername from RootWeb as it should be
                    using (SPWeb RootWeb = site.RootWeb)
                    {
                        SiteCollectionUrl = RootWeb.Url.ToLower();
                        if (!SiteCollectionUrl.EndsWith(""))
                            SiteCollectionUrl += "http://support.microsoft.com";
                    }

                    // check each site in the site collection
                    foreach (SPWeb web in site.AllWebs)
                    {
                        try
                        {
                            // check if the site is a publishing site
                            if (PublishingWeb.IsPublishingWeb(web))
                            {
                                PublishingWeb pubWeb = PublishingWeb.GetPublishingWeb(web);

                                // check each Publishing page in the current publishing site
                                foreach (PublishingPage page in pubWeb.GetPublishingPages())
                                {
                                    // retrieve the page layout of the publishing page
                                    string pageLayout = page.ListItem.Properties["PublishingPageLayout"] as string;

                                    // check for correct syntax of the page layout URL
                                    if (pageLayout != null)
                                        pageLayout = pageLayout.ToLower();

                                    if (String.IsNullOrEmpty(pageLayout))
                                    {
                                        Console.WriteLine("Error: Page \"{0}\" does not have a Page Layout assigned.\n", page.Uri);
                                    }
                                    else if (!pageLayout.Contains(MASTER_PAGE_LIB))
                                    {
                                        Console.WriteLine("Error: The Page Layout {0} for Page \"{1}\" does not point to the masterpage document library.\n",
                                            pageLayout, page.Uri);
                                    }
                                    else if (!pageLayout.StartsWith(SiteCollectionUrl) && pageLayout.StartsWith("http"))
                                    {
                                        // here we have a page which has a page layout that has a different URL which we have to fix
                                        Console.WriteLine("Page {0} has incorrect PageLayout Url", page.Uri);
                                        Console.WriteLine("Old URL: {0}", pageLayout);
                                        string pageLayoutWithoutPrefix = pageLayout.Substring(pageLayout.IndexOf(MASTER_PAGE_LIB));
                                        string newPageLayout = SiteCollectionUrl + pageLayoutWithoutPrefix;

                                        // perform the update in a try/catch block to cover problems
                                        try
                                        {
                                            int version = page.ListItem.File.MinorVersion;
                                            page.CheckOut();
                                            page.ListItem.Properties["PublishingPageLayout"] = newPageLayout;
                                            page.ListItem.File.Properties["PublishingPageLayout"] = newPageLayout;
                                            page.ListItem.Update();
                                            page.CheckIn("PublishingPageLayout corrected");

                                            //major version means that the item was published. Let's publish it again.
                                            if (version == 0)
                                                page.ListItem.File.Publish("PublishingPageLayout corrected");

                                            Console.WriteLine("Fixed URL: {0}\n", newPageLayout);
                                        }
                                        catch (Exception e)
                                        {
                                            Console.WriteLine("An error occurred while trying to fix the URL to the page layout:\n{0}", e);
                                        }
                                    }
                                }
                            }//if Publishing web
                        }
                        finally
                        {
                            // don't forget to dispose the SPWeb object
                            web.Dispose();
                        }
                    }
                }//using SPSite
            }
            catch (System.IO.FileNotFoundException ex)
            {
                Console.WriteLine("{0}", ex.Message);
            }
            catch (Exception ex)
            {
                Console.WriteLine("An error has occured: {0}", ex.Message);
            }
            
        }
    }
}