批量增加脚本命名空间

using System.IO;
using System.Text;
using UnityEditor;
using UnityEngine;

public class Test : MonoBehaviour
{
    public static string startStr = "\nnamespace Test{";
    public static string endStr = "\n}";


    public static string checkStr = "public class";
    public static string spaceName = "namespace Test";


    public static string rootName = "/Scripts";


    [MenuItem("Assets/AutoAddNameSpace")]
    public static void AutoAddNameSpace()
    {
        CheckFileName(Application.dataPath + rootName);
    }



    public static void CheckFileName(string dataPath)
    {
        foreach (var item in Directory.GetFiles(dataPath))
        {
            if (item.Contains(".cs") && !item.Contains(".meta"))
            {
                if (!ContainNameSpace(item))
                {
                    AddNameSpace(item);
                }
            }
        }


        foreach (var item in Directory.GetDirectories(dataPath))
        {
            CheckFileName(item);
        }
    }




    public static bool ContainNameSpace(string path)
    {
        StreamReader sr_check = new StreamReader(path);
        string all = sr_check.ReadToEnd();


        sr_check.Close();
        return all.Contains(spaceName);
    }


    public static void AddNameSpace(string path)
    {
        StreamReader sr = new StreamReader(path);
        StringBuilder sb = new StringBuilder();


        string line;
        int lineIndex = 0;


        while ((line = sr.ReadLine()) != null)
        {
            if (line.Contains(checkStr))
            {
                sb.Append(startStr);
            }
            if (lineIndex == 0)
                sb.Append(line);
            else
                sb.Append("\n" + line);
            lineIndex++;
        }


        sb.Append(endStr);


        sr.Close();


        StreamWriter sw = new StreamWriter(path);
        sw.WriteLine(sb.ToString());


        sw.Close();
    }

}


使用完流之后 需要将流关闭,否则会出现无法访问文件的情况。因为此时文件正在被流占用。



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值