SharePoint Automation Gary Lapointe – Founding Partner, Aptillon, Inc.

   1: using System;

   2: using System.Text;

   3: using Lapointe.SharePoint.STSADM.Commands.OperationHelpers;

   4: using Lapointe.SharePoint.STSADM.Commands.SPValidators;

   5: using System.Collections.Specialized;

   6: using Microsoft.SharePoint;

   7: using Microsoft.SharePoint.Administration;

   8:  

   9: namespace Lapointe.SharePoint.STSADM.Commands.SiteCollectionSettings

  10: {

  11:     /// <summary>

  12:     /// 

  13:     /// </summary>

  14:     public class RecalculateUsage : SPOperation

  15:     {

  16:         /// <summary>

  17:         /// Initializes a new instance of the <see cref="RecalculateUsage"/> class.

  18:         /// </summary>

  19:         public RecalculateUsage()

  20:         {

  21:             SPParamCollection parameters = new SPParamCollection();

  22:             parameters.Add(new SPParam("url", "url", false, null, new SPUrlValidator()));

  23:             parameters.Add(new SPParam("scope", "s", false, "site", new SPRegexValidator("(?i:^Farm$|^WebApplication$|^Site$)")));

  24:  

  25:  

  26:             StringBuilder sb = new StringBuilder();

  27:             sb.Append("\r\n\r\nRecalculates usage statistics for the given site(s).\r\n\r\nParameters:");

  28:             sb.Append("\r\n\t[-scope <Farm | WebApplication | Site>]");

  29:             sb.Append("\r\n\t[-url <url>]");

  30:  

  31:             Init(parameters, sb.ToString());

  32:         }

  33:  

  34:         /// <summary>

  35:         /// Gets the help message.

  36:         /// </summary>

  37:         /// <param name="command">The command.</param>

  38:         /// <returns></returns>

  39:         public override string GetHelpMessage(string command)

  40:         {

  41:             return HelpMessage;

  42:         }

  43:  

  44:         /// <summary>

  45:         /// Executes the specified command.

  46:         /// </summary>

  47:         /// <param name="command">The command.</param>

  48:         /// <param name="keyValues">The key values.</param>

  49:         /// <param name="output">The output.</param>

  50:         /// <returns></returns>

  51:         public override int Execute(string command, StringDictionary keyValues, out string output)

  52:         {

  53:             output = string.Empty;

  54:             Verbose = true;

  55:  

  56:             string scope = Params["scope"].Value.ToLowerInvariant();

  57:  

  58:             SPEnumerator enumerator;

  59:             if (scope == "farm")

  60:             {

  61:                 enumerator = new SPEnumerator(SPFarm.Local);

  62:             }

  63:             else if (scope == "webapplication")

  64:             {

  65:                 enumerator = new SPEnumerator(SPWebApplication.Lookup(new Uri(Params["url"].Value.TrimEnd('/'))));

  66:             }

  67:             else

  68:             {

  69:                 // scope == "site"

  70:                 using (SPSite site = new SPSite(Params["url"].Value.TrimEnd('/')))

  71:                 {

  72:                     Recalculate(site);

  73:                 }

  74:                 return OUTPUT_SUCCESS;

  75:             }

  76:  

  77:             enumerator.SPSiteEnumerated += new SPEnumerator.SPSiteEnumeratedEventHandler(enumerator_SPSiteEnumerated);

  78:             enumerator.Enumerate();

  79:  

  80:             return OUTPUT_SUCCESS;

  81:         }

  82:  

  83:         /// <summary>

  84:         /// Validates the specified key values.

  85:         /// </summary>

  86:         /// <param name="keyValues">The key values.</param>

  87:         public override void Validate(StringDictionary keyValues)

  88:         {

  89:             if (Params["scope"].Validate())

  90:             {

  91:                 Params["url"].IsRequired = true;

  92:                 Params["url"].Enabled = true;

  93:                 if (Params["scope"].Value.ToLowerInvariant() == "farm")

  94:                 {

  95:                     Params["url"].IsRequired = false;

  96:                     Params["url"].Enabled = false;

  97:                 }

  98:  

  99:             }

 100:             base.Validate(keyValues);

 101:         }

 102:  

 103:         /// <summary>

 104:         /// Handles the SPSiteEnumerated event of the enumerator control.

 105:         /// </summary>

 106:         /// <param name="sender">The source of the event.</param>

 107:         /// <param name="e">The <see cref="Lapointe.SharePoint.STSADM.Commands.OperationHelpers.SPEnumerator.SPSiteEventArgs"/> instance containing the event data.</param>

 108:         private void enumerator_SPSiteEnumerated(object sender, SPEnumerator.SPSiteEventArgs e)

 109:         {

 110:             Recalculate(e.Site);

 111:         }

 112:  

 113:         /// <summary>

 114:         /// Recalculates the specified site.

 115:         /// </summary>

 116:         /// <param name="site">The site.</param>

 117:         private void Recalculate(SPSite site)

 118:         {

 119:             Log("Recalculating {0}", site.Url);

 120:             site.RecalculateStorageUsed();

 121:             using (SPSite site2 = new SPSite(site.ID))

 122:                 Log("Storage updated from {0} to {1} (in bytes)\r\n", site.Usage.Storage.ToString(), site2.Usage.Storage.ToString());

 123:         }

 124:     }

 125: }