process_remove_unprotected.go

package main

import (
	"fmt"
	"time"

	"github.com/sirupsen/logrus"
)

func ProcessRemoveUnprotected() error {
	logrus.Info("pruning unprotected chunks from the database")

	err := LoadProtectedNodes()
	if err != nil {
		return fmt.Errorf("can't load 'mapcleaner_protect.txt' because of '%v' (i'm refusing to work without that file!)", err)
	}

	err = LoadState()
	if err != nil {
		return err
	}

	for {
		if state.ChunkX > state.ToX {
			// move to next z stride
			state.ChunkX = state.FromX
			state.ChunkZ++

			logrus.WithFields(logrus.Fields{
				"chunk_y": state.ChunkY,
				"chunk_z": state.ChunkZ,
			}).Info("Processing next z-stride")

			err := SaveState()
			if err != nil {
				return err
			}
		}
		if state.ChunkZ > state.ToZ {
			// move to next y stride
			state.ChunkX = state.FromX
			state.ChunkY++
			state.ChunkZ = state.FromZ

			logrus.WithFields(logrus.Fields{
				"chunk_y": state.ChunkY,
			}).Info("Processing next y-layer")
		}
		if state.ChunkY > state.ToY {
			// done
			return SaveState()
		}

		logrus.WithFields(logrus.Fields{
			"chunk_x": state.ChunkX,
			"chunk_y": state.ChunkY,
			"chunk_z": state.ChunkZ,
		}).Debug("Processing")

		emerged, err := IsEmerged(state.ChunkX, state.ChunkY, state.ChunkZ)
		if err != nil {
			return err
		}
		if emerged {
			protected, err := IsProtectedWithNeighbors(state.ChunkX, state.ChunkY, state.ChunkZ)
			if err != nil {
				return err
			}

			if !protected {
				logrus.WithFields(logrus.Fields{
					"chunk_x": state.ChunkX,
					"chunk_y": state.ChunkY,
					"chunk_z": state.ChunkZ,
				}).Info("Removing chunk")

				err = RemoveChunk(state.ChunkX, state.ChunkY, state.ChunkZ)
				if err != nil {
					return err
				}

				state.RemovedChunks++
			} else {
				state.RetainedChunks++
			}
		}

		state.ProcessedChunks++

		// shift to next chunk
		state.ChunkX++
		time.Sleep(time.Millisecond * time.Duration(state.Delay))
	}
}