Modify generator to insert anchors for pointers in constructors where necessary, fixing #102.

Author: arik-so

src/generation/base_type_generator.mts

@@ -150,6 +150,7 @@ export abstract class BaseTypeGenerator<Type extends RustType> {
 		let nativeCallPrefix = '';
 		let nativeCallWrapperPrefix = '', nativeCallWrapperSuffix = '';
 		let nativeCallSuffix = '';
+		const anchors = [];
 
 		const semantics = this.methodSemantics(method, containerType);
 		if (semantics.isConstructor && forceStaticConstructor) {
@@ -193,6 +194,9 @@ export abstract class BaseTypeGenerator<Type extends RustType> {
 			nativeCallWrapperSuffix += preparedArgument.methodCallWrapperSuffix;
 			nativeCallValueAccessors.push(preparedArgument.accessor);
 			nativeCallSuffix += preparedArgument.deferredCleanup;
+			if(preparedArgument.requiresAnchoring){
+				anchors.push(swiftArgumentName);
+			}
 		}
 
 		let cloneabilityDeprecationWarning = '';
@@ -226,11 +230,13 @@ export abstract class BaseTypeGenerator<Type extends RustType> {
 			visibility = 'internal';
 		}
 
+		let anchoringCommand = anchors.map(a => `try! returnValue.addAnchor(anchor: ${a})\n`).join('\t\t\t\t\t\t');
 		let methodDeclarationKeywords = `${visibility} ${staticInfix}func`;
-		let returnCommand = 'return returnValue';
+		let returnCommand = `${anchoringCommand}return returnValue`;
 		let returnValueHandlingPrefix = '';
 		let returnValueHandlingSuffix = '';
 		if (swiftMethodName === 'init') {
+			anchoringCommand = anchoringCommand.replaceAll('returnValue.addAnchor(', 'self.addAnchor(')
 			// it's a constructor
 			methodDeclarationKeywords = visibility;
 			returnCommand = `
@@ -239,6 +245,7 @@ export abstract class BaseTypeGenerator<Type extends RustType> {
 				Self.instanceCounter += 1
 				self.instanceNumber = Self.instanceCounter
 				super.init(conflictAvoidingVariableName: 0)
+				${anchoringCommand}
 			`;
 
 			returnValueHandlingPrefix = '/*';
@@ -625,7 +632,8 @@ export abstract class BaseTypeGenerator<Type extends RustType> {
 			methodCallWrapperPrefix: '',
 			methodCallWrapperSuffix: '',
 
-			deferredCleanup: ''
+			deferredCleanup: '',
+			requiresAnchoring: false
 		};
 
 		// this argument is the content of an elided container, like the iteratee of a Vec
@@ -807,6 +815,10 @@ export abstract class BaseTypeGenerator<Type extends RustType> {
 						}
 				`;
 
+				if(memoryContext && memoryContext.isConstructor && (argument.type instanceof RustStruct || argument.type instanceof RustTaggedValueEnum || argument.type instanceof RustResult)){
+					preparedArgument.requiresAnchoring = true;
+				}
+
 				// the wrapper accesses the variable normally, and introduces a new variable name by which to refer to the value
 				preparedArgument.accessor = preparedArgument.name;
 			}
@@ -1284,6 +1296,11 @@ export interface PreparedArgument {
 	 * memory deallocation or, ironically, retention beyond the call site
 	 */
 	deferredCleanup: string
+
+	/**
+	 * If true (only applicable in constructors), add a `self.addAnchor` call after super.init()
+	 */
+	requiresAnchoring: boolean
 }
 
 export interface PreparedReturnValue {

src/generation/trait_generator.mts

@@ -57,6 +57,7 @@ export default class TraitGenerator extends BaseTypeGenerator<RustTrait> {
 				}
 			} else {
 				// currentField is definitely instanceof ContextualRustType
+				console.log(currentField.contextualName);
 				const swiftArgumentName = Generator.snakeCaseToCamelCase(currentField.contextualName);
 				let initializationValue = swiftArgumentName;
 
@@ -352,7 +353,8 @@ export default class TraitGenerator extends BaseTypeGenerator<RustTrait> {
 			methodCallWrapperPrefix: '',
 			methodCallWrapperSuffix: '',
 
-			deferredCleanup: ''
+			deferredCleanup: '',
+			requiresAnchoring: false
 		};
 
 		if (argumentType.contextualName === 'init') {