package org.litesoft.aokeyhole.swing.mains.mementobeans.support;

import java.io.*;

import org.litesoft.commonfoundation.typeutils.*;

import java.util.*;

import org.litesoft.aokeyhole.persist.*;
import org.litesoft.core.simpletypes.*;
import org.litesoft.textfiledirectory.*;

public class BeanSubSystemState implements BeanMagicStrings
{
    private final String mName;
    private final String mPackage;
    private final TextFileDirectory mTextFileDirectory;
    private List<BeanReader> mBeanReaders = null; // null == unread
    private int mNextReader = 0;

    public BeanSubSystemState( String pName, String pPackage, TextFileDirectory pTextFileDirectory )
    {
        mName = pName;
        mPackage = pPackage;
        mTextFileDirectory = pTextFileDirectory;
    }

    public String getName()
    {
        return mName;
    }

    public synchronized BeanReader getNextBeanReader()
    {
        List<BeanReader> zBeanReaders = getBeanReaders();
        return (mNextReader < zBeanReaders.size()) ? zBeanReaders.get( mNextReader++ ) : null;
    }

    private List<BeanReader> getBeanReaders()
    {
        if ( mBeanReaders == null )
        {
            mBeanReaders = populateOriginalBeans();
        }
        return mBeanReaders;
    }

    public SubSystemWriter getSubSystemWriter( Map<String, BeanObjectBuilder> pObjectBuildersByObjectName )
    {
        return new OurSubSystemWriter( pObjectBuildersByObjectName );
    }

    private List<BeanReader> populateOriginalBeans()
    {
        List<BeanReader> zBeanReaders = Lists.newArrayList();

        for ( String zFile : mTextFileDirectory.getFiles( SOURCE_FILE_EXTENSION ) )
        {
            TextFile zTextFile = mTextFileDirectory.load( zFile );
            BeanReader zReader = BeanReader.createFor( zTextFile.getLines() );
            if ( zReader != null )
            {
                zBeanReaders.add( zReader );
            }
        }
        return zBeanReaders;
    }

    private class OurSubSystemWriter implements SubSystemWriter
    {
        private final Map<String, BeanObjectBuilder> mObjectBuildersByObjectName;

        private OurSubSystemWriter( Map<String, BeanObjectBuilder> pObjectBuildersByObjectName )
        {
            mObjectBuildersByObjectName = pObjectBuildersByObjectName;
        }

        @Override
        public void write()
                throws IOException
        {
            List<BeanReader> zNewBeanReaders = Lists.newArrayList();
            BeanWriter zBeanWriter = new BeanWriter( mTextFileDirectory, mPackage );
            for ( BeanReader zBeanReader : getBeanReaders() )
            {
                addTo( zNewBeanReaders, zBeanWriter.persist( zBeanReader, mObjectBuildersByObjectName.remove( zBeanReader.getName() ) ) );
            }
            for ( BeanObjectBuilder zBeanObjectBuilder : mObjectBuildersByObjectName.values() )
            {
                addTo( zNewBeanReaders, zBeanWriter.persist( null, zBeanObjectBuilder ) );
            }
            mBeanReaders = zNewBeanReaders;
        }

        private void addTo( List<BeanReader> pNewBeanReaders, BeanReader pBeanReader )
        {
            if ( pBeanReader != null )
            {
                pNewBeanReaders.add( pBeanReader );
            }
        }
    }
}