repopulate transactions partly done

main.go

@@ -394,6 +394,7 @@ func main() {
 							s.Expenses.Repopulate(s)
 							s.Invoices.Repopulate(s)
 							s.MiscRecords.Repopulate(s)
+							s.Statement.Repopulate(s)
 							s.Save()
 							return nil
 						},

pkg/schema/statement.go

@@ -113,6 +113,12 @@ func (t Statement) TransactionSearchItems() util.SearchItems {
 	return result
 }
 
+func (t Statement) Repopulate(s Schema) {
+	for i := range t.Transactions {
+		t.Transactions[i].Repopulate(s)
+	}
+}
+
 func (t *Statement) AssistedCompletion(s Schema, doAll, autoSave, autoMode, askSkip, documentsOnly bool) {
 	first := true
 	for i := range t.Transactions {

pkg/schema/transaction.go

@@ -321,6 +321,35 @@ func (t Transaction) Validate() util.ValidateResults {
 	return []util.ValidateResult{util.Check(t)}
 }
 
+func (t *Transaction) Repopulate(s Schema) {
+	docDone := false
+	for i := range s.Expenses {
+		if s.Expenses[i].SettlementTransaction.Match(t) {
+			t.AssociatedDocument = NewRef(s.Expenses[i].SettlementTransaction.Id)
+			docDone = true
+			break
+		}
+	}
+	if !docDone {
+		for i := range s.Invoices {
+			if s.Invoices[i].SettlementTransaction.Match(t) {
+				t.AssociatedDocument = NewRef(s.Invoices[i].SettlementTransaction.Id)
+				docDone = true
+				break
+			}
+		}
+	}
+	if !docDone {
+		for i := range s.MiscRecords {
+			if s.MiscRecords[i].Transaction.Match(t) {
+				t.AssociatedDocument = NewRef(s.MiscRecords[i].Transaction.Id)
+				docDone = true
+				break
+			}
+		}
+	}
+}
+
 func (t Transaction) SearchItem() util.SearchItem {
 	return util.SearchItem{
 		Name:        t.Description,