github.com/TrueBlocks/trueblocks-core/src/apps/chifra@v0.0.0-20241022031540-b362680128f7/internal/chunks/handle_index_belongs.go

// Copyright 2021 The TrueBlocks Authors. All rights reserved.
// Use of this source code is governed by a license that can
// be found in the LICENSE file.

package chunksPkg

import (
	"encoding/binary"
	"fmt"
	"io"

	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/base"
	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/file"
	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/index"
	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/output"
	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/tslib"
	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/types"
	"github.com/TrueBlocks/trueblocks-core/src/apps/chifra/pkg/walk"
)

// HandleIndexBelongs displays the resolved records in a chunk given a single address
func (opts *ChunksOptions) HandleIndexBelongs(rCtx *output.RenderCtx, blockNums []base.Blknum) error {
	chain := opts.Globals.Chain

	fetchData := func(modelChan chan types.Modeler, errorChan chan error) {
		showAddressesBelongs := func(walker *walk.CacheWalker, path string, first bool) (bool, error) {
			return opts.handleResolvedRecords(modelChan, walker, path)
		}

		walker := walk.NewCacheWalker(
			chain,
			opts.Globals.TestMode,
			10000, /* maxTests */
			showAddressesBelongs,
		)

		if err := walker.WalkBloomFilters(blockNums); err != nil {
			errorChan <- err
			rCtx.Cancel()
		}
	}

	return output.StreamMany(rCtx, fetchData, opts.Globals.OutputOpts())
}

// handleResolvedRecords is a helper function for HandleIndexBelongs and verbose versions of
// HandleAddresses and HandleAppearances. It is called once for each chunk in the index and
// depends on the values of opts.Globals.Verbose and opts.Belongs.
func (opts *ChunksOptions) handleResolvedRecords(modelChan chan types.Modeler, walker *walk.CacheWalker, path string) (bool, error) {
	if path != index.ToBloomPath(path) {
		return false, fmt.Errorf("should not happen in showAddressesBelongs")
	}

	path = index.ToIndexPath(path)
	if !file.FileExists(path) {
		// Bloom files exist, but index files don't. It's okay.
		return true, nil
	}

	indexChunk, err := index.OpenIndex(path, true /* check */)
	if err != nil {
		return false, err
	}
	defer indexChunk.Close()

	_, err = indexChunk.File.Seek(int64(index.HeaderWidth), io.SeekStart)
	if err != nil {
		return false, err
	}

	cnt := 0
	for i := 0; i < int(indexChunk.Header.AddressCount); i++ {
		if opts.Globals.TestMode && i > walker.MaxTests() {
			continue
		}

		s := types.AppearanceTable{}
		if err := binary.Read(indexChunk.File, binary.LittleEndian, &s.AddressRecord); err != nil {
			return false, err
		}

		if opts.shouldShow(s.AddressRecord) {
			if uint64(cnt) >= opts.MaxAddrs {
				break
			}

			if s.Appearances, err = indexChunk.ReadAppearancesAndReset(&s.AddressRecord); err != nil {
				return false, err
			}
			s.AddressRecord.Count = uint32(len(s.Appearances))
			if opts.FirstBlock != 0 || opts.LastBlock != base.NOPOSN {
				good := []types.AppRecord{}
				for _, app := range s.Appearances {
					if base.Blknum(app.BlockNumber) >= opts.FirstBlock && base.Blknum(app.BlockNumber) <= opts.LastBlock {
						good = append(good, app)
					}
				}
				s.Appearances = good
				s.AddressRecord.Count = uint32(len(good))
			}
			if len(s.Appearances) == 0 {
				continue
			}
			modelChan <- &s
			cnt++
		}
	}

	return true, nil
}

// handleResolvedRecords1 is a helper function for HandleIndexBelongs and verbose versions of
// HandleAddresses and HandleAppearances. It is called once for each chunk in the index and
// depends on the values of opts.Globals.Verbose and opts.Belongs.
func (opts *ChunksOptions) handleResolvedRecords1(modelChan chan types.Modeler, walker *walk.CacheWalker, path string) (bool, error) {
	chain := opts.Globals.Chain

	if path != index.ToBloomPath(path) {
		return false, fmt.Errorf("should not happen in showAddressesBelongs")
	}

	path = index.ToIndexPath(path)
	if !file.FileExists(path) {
		// Bloom files exist, but index files don't. It's okay.
		return true, nil
	}

	indexChunk, err := index.OpenIndex(path, true /* check */)
	if err != nil {
		return false, err
	}
	defer indexChunk.Close()

	_, err = indexChunk.File.Seek(int64(index.HeaderWidth), io.SeekStart)
	if err != nil {
		return false, err
	}

	cnt := 0
	for i := 0; i < int(indexChunk.Header.AddressCount); i++ {
		if opts.Globals.TestMode && i > walker.MaxTests() {
			continue
		}

		s := types.AppearanceTable{}
		if err := binary.Read(indexChunk.File, binary.LittleEndian, &s.AddressRecord); err != nil {
			return false, err
		}

		if opts.shouldShow(s.AddressRecord) {
			if uint64(cnt) >= opts.MaxAddrs {
				break
			}

			if s.Appearances, err = indexChunk.ReadAppearancesAndReset(&s.AddressRecord); err != nil {
				return false, err
			}
			s.AddressRecord.Count = uint32(len(s.Appearances))
			if opts.FirstBlock != 0 || opts.LastBlock != base.NOPOSN {
				good := []types.AppRecord{}
				for _, app := range s.Appearances {
					if base.Blknum(app.BlockNumber) >= opts.FirstBlock && base.Blknum(app.BlockNumber) <= opts.LastBlock {
						good = append(good, app)
					}
				}
				s.Appearances = good
				s.AddressRecord.Count = uint32(len(good))
			}
			if len(s.Appearances) == 0 {
				continue
			}
			rng := indexChunk.Range
			ss := types.ChunkAddress{
				Address: s.AddressRecord.Address,
				Count:   uint64(s.AddressRecord.Count),
				Offset:  uint64(s.AddressRecord.Offset),
				Range:   rng.String(),
			}
			rd := tslib.RangeToBounds(chain, &rng)
			ss.RangeDates = &rd
			modelChan <- &ss
			cnt++
		}
	}

	return true, nil
}