Enchant-Cleaner-Patcher.pas

unit userscript;

uses 'lib\mxpf';

function Initialize: integer;

var
    badlist_A, badlist_W, goodlist_A, goodlist_W: IInterface;
    badrec, goodrec, badforms_A, badforms_W, newbadform, goodforms_A, goodforms_W, newgoodform: IInterface;
    KWDA, VMAD: IInterface;
    index, P0, P1, P2, object_index, N: integer;
    item_edid, item_name: string;
    ArmorScriptExceptions, WeaponScriptExceptions: TStringList;
    badobject, goodobject: TObject;
    badobject_list, goodobject_list: TList;
    ArmorIsPlayable, ArmorHasEnch, ArmorHasKeyword, ArmorIsInList: boolean;
    WeaponIsPlayable, WeaponHasEnch, WeaponHasKeyword, WeaponIsInList: boolean;
    NewObject: boolean;

begin
    InitializeMXPF;
    DefaultOptionsMXPF;
    PatchFileByName('EnchantCleaner.esp');

//----------------------------------------------------------------// SCRIPT EXCEPTIONS. DON'T EDIT ANYTHING ABOVE.
    ArmorScriptExceptions := TStringList.Create;
    //  ArmorScriptExceptions.Add('???');
    //  ArmorScriptExceptions.Add('???');
    //  ArmorScriptExceptions.Add('???');
    //  ArmorScriptExceptions.Add('???');

    WeaponScriptExceptions := TStringList.Create;
        WeaponScriptExceptions.Add('SilverSwordScript');
        WeaponScriptExceptions.Add('DLC1EnhancedCrossBowAddPerkScript');
    //  WeaponScriptExceptions.Add('???');
    //  WeaponScriptExceptions.Add('???');
//----------------------------------------------------------------// SCRIPT EXCEPTIONS. DON'T EDIT ANYTHING BELOW.

    badlist_A := MainRecordByEditorID(GroupBySignature(mxPatchFile, 'FLST'), 'EC_ArmorList');
    badforms_A := Add(badlist_A, 'FormIDs', true);
    Remove(ElementByIP(badlist_A, 'FormIDs\[0]'));

    goodlist_A := MainRecordByEditorID(GroupBySignature(mxPatchFile, 'FLST'), 'EC_ArmorList_NoEnch');
    goodforms_A := Add(goodlist_A, 'FormIDs', true);
    Remove(ElementByIP(goodlist_A, 'FormIDs\[0]'));

    badlist_W := MainRecordByEditorID(GroupBySignature(mxPatchFile, 'FLST'), 'EC_WeaponList');
    badforms_W := Add(badlist_W, 'FormIDs', true);
    Remove(ElementByIP(badlist_W, 'FormIDs\[0]'));

    goodlist_W := MainRecordByEditorID(GroupBySignature(mxPatchFile, 'FLST'), 'EC_WeaponList_NoEnch');
    goodforms_W := Add(goodlist_W, 'FormIDs', true);
    Remove(ElementByIP(goodlist_W, 'FormIDs\[0]'));

    badobject_list := TList.Create;
    goodobject_list := TList.Create;

    LoadRecords('ARMO');
    for index := 0 to MaxRecordIndex do begin
        badrec := GetRecord(index);

        ArmorIsPlayable := false;
        ArmorHasEnch := false;
        ArmorHasKeyword := false;

        P0 := StrToInt('0' + GetElementEditValues(badrec, 'Record Header\Record Flags\Non-Playable'));
        P1 := StrToInt('0' + GetElementEditValues(badrec, 'BODT\General Flags\(ARMO)Non-Playable'));
        P2 := StrToInt('0' + GetElementEditValues(badrec, 'BOD2\General Flags\(ARMO)Non-Playable'));
        if (P0 + P1 + P2 = 0) then ArmorIsPlayable := true;
        if (GetElementEditValues(badrec, 'EITM') <> '') then ArmorHasEnch := true;
        if (HasKeyword(badrec, 'MagicDisallowEnchanting') = true) then ArmorHasKeyword := true;

        ArmorIsInList := false;
        if (ArmorIsPlayable = true) and ((ArmorHasEnch = true) or (ArmorHasKeyword = true)) then ArmorIsInList := true;

        if (ArmorIsInList = true) then begin
            item_edid := GetElementEditValues(badrec, 'EDID');
            item_name := Name(badrec);

            AddRequiredMasters(badrec, mxPatchFile);
            newbadform := ElementAssign(badforms_A, HighInteger, nil, false);
            SetEditValue(newbadform, item_name);

            while (GetElementEditValues(badrec, 'TNAM') <> '') do begin
                badrec := LinksTo(ElementByPath(badrec, 'TNAM'));
                badrec := HighestOverrideOrSelf(badrec, 4096);
                AddRequiredMasters(badrec, mxPatchFile);
            end;

            if (HasKeyword(badrec, 'MagicDisallowEnchanting') = false) and (GetElementEditValues(badrec, 'EITM') = '')
            then goodrec := badrec
            else begin
                badobject := TObject(badrec);
                object_index := badobject_list.IndexOf(badobject);
                if (object_index = -1)
                then begin
                    goodrec := wbCopyElementToFile(badrec, mxPatchFile, true, true);
                    goodobject := TObject(goodrec);
                    badobject_list.Add(badobject);
                    goodobject_list.Add(goodobject);
                    NewObject := true;
                end
                else begin
                    goodobject := goodobject_list[object_index];
                    goodrec := ObjectToElement(goodobject);
                    NewObject := false;
                end;

                if (NewObject = true) then begin
                    SetElementEditValues(goodrec, 'DESC', '');
                    Remove(ElementByIP(goodrec, 'EITM'));

                    KWDA := ElementByPath(goodrec, 'KWDA');
                    for N := Pred(ElementCount(KWDA)) downto 0 do if (GetElementEditValues(LinksTo(ElementByIndex(KWDA, N)), 'EDID') = 'MagicDisallowEnchanting') then Remove(ElementByIndex(KWDA, N));
                    if (ElementCount(KWDA) = 0) then Remove(ElementByPath(goodrec, 'KWDA'));

                    VMAD := ElementByPath(goodrec, 'VMAD\Scripts');
                    for N := Pred(ElementCount(VMAD)) downto 0 do if (ArmorScriptExceptions.IndexOf(GetElementEditValues(goodrec, 'VMAD\Scripts\[' + IntToStr(N) + ']\ScriptName')) = -1) then Remove(ElementByPath(goodrec, 'VMAD\Scripts\[' + IntToStr(N) + ']'));
                    if (ElementCount(VMAD) = 0) then Remove(ElementByPath(goodrec, 'VMAD'));

                    SetElementEditValues(goodrec, 'EDID', GetElementEditValues(goodrec, 'EDID') + '_ECDup');
                end;
            end;

            newgoodform := ElementAssign(goodforms_A, HighInteger, nil, false);
            SetEditValue(newgoodform, Name(goodrec));

            AddMessage(item_edid + ' -->-- ' + GetElementEditValues(goodrec, 'EDID'));
        end;
    end;
    for index := MaxRecordIndex downto 0 do RemoveRecord(index);
    badobject_list.Clear;
    goodobject_list.Clear;

    LoadRecords('WEAP');
    for index := 0 to MaxRecordIndex do begin
        badrec := GetRecord(index);

        WeaponIsPlayable := false;
        WeaponHasEnch := false;
        WeaponHasKeyword := false;

        P0 := StrToInt('0' + GetElementEditValues(badrec, 'Record Header\Record Flags\Non-Playable'));
        P1 := StrToInt('0' + GetElementEditValues(badrec, 'DNAM\Flags\Non-playable'));
        if (P0 + P1 = 0) then WeaponIsPlayable := true;
        if (GetElementEditValues(badrec, 'EITM') <> '') then WeaponHasEnch := true;
        if (HasKeyword(badrec, 'MagicDisallowEnchanting') = true) then WeaponHasKeyword := true;

        WeaponIsInList := false;
        if (WeaponIsPlayable = true) and ((WeaponHasEnch = true) or (WeaponHasKeyword = true)) then WeaponIsInList := true;

        if (WeaponIsInList = true) then begin
            item_edid := GetElementEditValues(badrec, 'EDID');
            item_name := Name(badrec);

            AddRequiredMasters(badrec, mxPatchFile);
            newbadform := ElementAssign(badforms_W, HighInteger, nil, false);
            SetEditValue(newbadform, item_name);

            while (GetElementEditValues(badrec, 'CNAM') <> '') do begin
                badrec := LinksTo(ElementByPath(badrec, 'CNAM'));
                badrec := HighestOverrideOrSelf(badrec, 4096);
                AddRequiredMasters(badrec, mxPatchFile);
            end;

            if (HasKeyword(badrec, 'MagicDisallowEnchanting') = false) and (GetElementEditValues(badrec, 'EITM') = '')
            then goodrec := badrec
            else begin
                badobject := TObject(badrec);
                object_index := badobject_list.IndexOf(badobject);
                if (object_index = -1)
                then begin
                    goodrec := wbCopyElementToFile(badrec, mxPatchFile, true, true);
                    goodobject := TObject(goodrec);
                    badobject_list.Add(badobject);
                    goodobject_list.Add(goodobject);
                    NewObject := true;
                end
                else begin
                    goodobject := goodobject_list[object_index];
                    goodrec := ObjectToElement(goodobject);
                    NewObject := false;
                end;

                if (NewObject = true) then begin
                    SetElementEditValues(goodrec, 'DESC', '');
                    Remove(ElementByIP(goodrec, 'EITM'));

                    KWDA := ElementByPath(goodrec, 'KWDA');
                    for N := Pred(ElementCount(KWDA)) downto 0 do if (GetElementEditValues(LinksTo(ElementByIndex(KWDA, N)), 'EDID') = 'MagicDisallowEnchanting') then Remove(ElementByIndex(KWDA, N));
                    if (ElementCount(KWDA) = 0) then Remove(ElementByPath(goodrec, 'KWDA'));

                    VMAD := ElementByPath(goodrec, 'VMAD\Scripts');
                    for N := Pred(ElementCount(VMAD)) downto 0 do if (WeaponScriptExceptions.IndexOf(GetElementEditValues(goodrec, 'VMAD\Scripts\[' + IntToStr(N) + ']\ScriptName')) = -1) then Remove(ElementByPath(goodrec, 'VMAD\Scripts\[' + IntToStr(N) + ']'));
                    if (ElementCount(VMAD) = 0) then Remove(ElementByPath(goodrec, 'VMAD'));

                    SetElementEditValues(goodrec, 'EDID', GetElementEditValues(goodrec, 'EDID') + '_ECDup');
                end;
            end;

            newgoodform := ElementAssign(goodforms_W, HighInteger, nil, false);
            SetEditValue(newgoodform, Name(goodrec));

            AddMessage(item_edid + ' -->-- ' + GetElementEditValues(goodrec, 'EDID'));
        end;
    end;
    for index := MaxRecordIndex downto 0 do RemoveRecord(index);
    badobject_list.Clear;
    goodobject_list.Clear;

    FinalizeMXPF;
end;

procedure AddRequiredMasters(rec: IInterface; target: IwbFile);
var
    masters: TStringList;
begin
    masters := TStringList.Create;
    AddMastersToList(GetFile(rec), masters);
    AddMastersToFile(target, masters, true);
    SortMasters(target);
    masters.Clear;
end;

end.