@@ -134,46 +134,46 @@ func (s CCIPChainState) GenerateView(e *cldf.Environment, selector uint64, chain
134134 return tonView , errGroup .Wait ()
135135}
136136
137- func SaveOnchainState (chainSelector uint64 , state CCIPChainState , e cldf.Environment ) error {
137+ func GetAddressBook (chainSelector uint64 , state CCIPChainState ) ( cldf.AddressBook , error ) {
138138 // TODO: use DataStore
139- ab := e . ExistingAddresses
139+ ab := cldf . NewMemoryAddressBook ()
140140 if ! state .LinkTokenAddress .IsAddrNone () {
141141 err := ab .Save (chainSelector , state .LinkTokenAddress .String (), cldf .NewTypeAndVersion (LinkToken , Version1_6_0 ))
142142 if err != nil {
143- return err
143+ return nil , err
144144 }
145145 }
146146 if ! state .ReceiverAddress .IsAddrNone () {
147147 err := ab .Save (chainSelector , state .ReceiverAddress .String (), cldf .NewTypeAndVersion (TonReceiver , Version1_6_0 ))
148148 if err != nil {
149- return err
149+ return nil , err
150150 }
151151 }
152152 if ! state .OffRamp .IsAddrNone () {
153153 err := ab .Save (chainSelector , state .OffRamp .String (), cldf .NewTypeAndVersion (OffRamp , Version1_6_0 ))
154154 if err != nil {
155- return err
155+ return nil , err
156156 }
157157 }
158158 if ! state .Router .IsAddrNone () {
159159 err := ab .Save (chainSelector , state .Router .String (), cldf .NewTypeAndVersion (Router , Version1_6_0 ))
160160 if err != nil {
161- return err
161+ return nil , err
162162 }
163163 }
164164 if ! state .OnRamp .IsAddrNone () {
165165 err := ab .Save (chainSelector , state .OnRamp .String (), cldf .NewTypeAndVersion (OnRamp , Version1_6_0 ))
166166 if err != nil {
167- return err
167+ return nil , err
168168 }
169169 }
170170 if ! state .FeeQuoter .IsAddrNone () {
171171 err := ab .Save (chainSelector , state .FeeQuoter .String (), cldf .NewTypeAndVersion (FeeQuoter , Version1_6_0 ))
172172 if err != nil {
173- return err
173+ return nil , err
174174 }
175175 }
176- return nil
176+ return ab , nil
177177}
178178
179179func LoadOnchainState (e cldf.Environment ) (map [uint64 ]CCIPChainState , error ) {
0 commit comments