create function to setup new export set

use fileDialog2 to perform export so that we get the export path

Author: vkovec

Assets/com.unity.formats.fbx/Editor/Integrations/Autodesk/maya/scripts/unityCommands.mel

@@ -446,6 +446,146 @@ proc string[] getUnityExportSets(){
     return $unityExportSets;
 }
 
+proc setupNewExportSet(string $exportPath, int $exportAnimOnly, string $selectedObjects[]){
+    global string $UnityFbxFilePathAttr;
+    global string $UnityFbxFileNameAttr;
+    global string $UnityFbxAnimFilePathAttr;
+    global string $UnityFbxAnimFileNameAttr;
+    global string $UnityFbxNamespaceAttr;
+    global string $UnityExportSetNameFormat;
+    // extract name from path
+    // create export set
+    // create namespace
+    // move objects into namespace
+    // add objects to export set
+    // add attributes to set
+    
+    $exportDir = dirname($exportPath);
+    $exportFileName = basename($exportPath, "");
+    $exportAnimDir = $exportDir;
+    $exportAnimFileName = "";
+    $fileNameWithoutExt = basename($exportPath, ".fbx");
+    $exportAnimFileName = ($fileNameWithoutExt + "@Take1.fbx");
+    $isAnimFile = $exportAnimOnly;
+    
+    if($exportAnimOnly){
+        // import as animation
+        $exportAnimDir = $exportDir;
+        $exportAnimFileName = $exportFileName;
+        
+        if(match("@", basename($exportDir, ".fbx")) != ""){
+            $fileNameWithoutExt = match("[^@]+", $fileNameWithoutExt);
+        }
+    }
+    $fileNameWithoutExt = formValidObjectName($fileNameWithoutExt);
+    
+    $unityExportSet = `format -stringArg $fileNameWithoutExt $UnityExportSetNameFormat`;
+    
+    string $origNamespace = `namespaceInfo -cur -an`;
+    string $targetNamespace = ":" + $fileNameWithoutExt;
+    // make sure there are no duplicate colons in namespace name
+    if($origNamespace != ":"){
+        $targetNamespace = `format -s $origNamespace -s $fileNameWithoutExt "^1s:^2s"`;
+    }
+    
+    $setNamespaceExists = false;
+    if (setExists($unityExportSet)){
+        string $unitySetContents[] = `listConnections $unityExportSet`;
+        
+        // get namespace from set
+        $setNamespace = getAttribute($unityExportSet, $UnityFbxNamespaceAttr);
+        if(size($unitySetContents) > 0){
+            $setNamespace = checkNamespaceNeedsUpdate($unityExportSet, $setNamespace, getObjectNamespace($unitySetContents[0]));
+        }
+        
+        if($setNamespace != "" && `namespace -exists $setNamespace`){
+            $targetNamespace = $setNamespace;
+            $setNamespaceExists = true;
+        }
+    }
+    
+    // warn if namespace already exists
+    if(!$setNamespaceExists){
+        if(`namespace -exists $targetNamespace`){
+            if(!showConfirmDialog("Warning: " + $exportFileName,
+                $targetNamespace + " namespace already exists, the selected objects will be added to the existing namespace and export set.",
+                "Continue", "Cancel"
+               )){
+                // cancelled, don't import this fbx
+                return;
+            }
+        }
+        else{
+            namespace -add $targetNamespace;
+        }
+    }
+    
+    // Get or create the Unity Fbx Export Set
+    $setCreated = false;
+    if (!setExists($unityExportSet)){
+        if(!`namespaceInfo -isRootNamespace $origNamespace`){
+            namespace -set ":";
+        }
+        
+        // if a set is selected when creating a new set, then
+        // the selected set will be added into the new set.
+        // avoid this by temporarily deselecting everything.
+        $origSelection = `ls -sl`;
+        select -clear;
+        
+        // couldn't find export set so create it
+        sets -name $unityExportSet;
+        $setCreated = true;
+        
+        if(size($origSelection) > 0){
+            select -r $origSelection;
+        }
+    }
+    
+    // unlock set so we can add attributes to it
+    lockNode -lock false $unityExportSet;
+    
+    if(!$isAnimFile){
+        // reset attribute values, in case import fails
+        storeAttribute($unityExportSet, $UnityFbxFilePathAttr, "");
+        storeAttribute($unityExportSet, $UnityFbxFileNameAttr, "");
+        storeAttribute($unityExportSet, $UnityFbxNamespaceAttr, "");
+    }
+    
+    // add to the targetNamespace
+    // iterate over all selected objects and rename
+    // rename FOO:sphere2 :BAR:sphere2;
+    
+    if ((!$isAnimFile || ($isAnimFile && $setCreated)) && $exportDir != ""){
+        storeAttribute($unityExportSet, $UnityFbxFilePathAttr, $exportDir);
+    }
+    
+    if ((!$isAnimFile || ($isAnimFile && $setCreated)) && $exportFileName != ""){
+        storeAttribute($unityExportSet,$UnityFbxFileNameAttr,$exportFileName);
+    }
+    
+    if($exportAnimDir != ""){
+        storeAttribute($unityExportSet,$UnityFbxAnimFilePathAttr,$exportAnimDir);
+    }
+    
+    if($exportAnimFileName != ""){
+        storeAttribute($unityExportSet,$UnityFbxAnimFileNameAttr,$exportAnimFileName);
+    }
+    
+    if($fileNameWithoutExt != ""){
+        storeAttribute($unityExportSet, $UnityFbxNamespaceAttr, $targetNamespace);
+    }
+    
+    if (setExists($unityExportSet) == true){
+        // add newly imported items to set
+        if (size($selectedObjects) > 0){
+            sets -include $unityExportSet $selectedObjects;
+        }
+        // lock set so it doesn't get deleted when empty
+        lockNode -lock true $unityExportSet;
+    }
+}
+
 proc unityExport(int $exportType){
 
     if(!loadUnityDependencies()){
@@ -501,7 +641,16 @@ proc unityExport(int $exportType){
 
     // if selection doesn't belong to a set, export to a new file
     if(size($setsToExport) <= 0){
-        eval "SendToUnitySelection";
+        //eval "GamePipelineSendToUnity 1";
+        $unityProject = `optionVar -q "UnityProject"`;
+        $exportPath = `fileDialog2 -ds 2 -cap "FBX Export Selection" -dir $unityProject -ff "*.fbx" -fm 0`;
+        if(size($exportPath)<=0){
+            return;
+        }
+        $exportCmd = "file -force -options \"\" -typ \"FBX export\" -pr -es \"" + $exportPath[0] + "\"";
+        eval $exportCmd;
+        
+        setupNewExportSet($exportPath[0], $exportAnimOnly, $origSelection);
         return;
     }