CodeIterationStatementExtensions.cs

#region License
//  Copyright 2004-2010 Castle Project - http:www.castleproject.org/
//  
//  Licensed under the Apache License, Version 2.0 (the "License");
//  you may not use this file except in compliance with the License.
//  You may obtain a copy of the License at
//  
//      http:www.apache.org/licenses/LICENSE-2.0
//  
//  Unless required by applicable law or agreed to in writing, software
//  distributed under the License is distributed on an "AS IS" BASIS,
//  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
//  See the License for the specific language governing permissions and
//  limitations under the License.
// 
#endregion

namespace Castle.ActiveWriter.CodeDomExtensions
{
	using System.CodeDom;

	public static class CodeIterationStatementExtensions
    {
        public static CodeIterationStatement Clone(this CodeIterationStatement statement)
        {
            if (statement == null) return null;
            CodeIterationStatement s = new CodeIterationStatement();
            s.EndDirectives.AddRange(statement.EndDirectives);
            s.IncrementStatement = statement.IncrementStatement.Clone();
            s.InitStatement = statement.InitStatement.Clone();
            s.LinePragma = statement.LinePragma;
            s.StartDirectives.AddRange(statement.StartDirectives);
            s.Statements.AddRange(statement.Statements.Clone());
            s.TestExpression = statement.TestExpression.Clone();
            s.UserData.AddRange(statement.UserData);
            return s;
        }

        public static void ReplaceType(this CodeIterationStatement statement, string oldType, string newType)
        {
            if (statement == null) return;
            statement.IncrementStatement.ReplaceType(oldType, newType);
            statement.InitStatement.ReplaceType(oldType, newType);
            statement.Statements.ReplaceType(oldType, newType);
            statement.TestExpression.ReplaceType(oldType, newType);
        }
    }
}