// This Source Code is in the Public Domain per: http://litesoft.org/License.txt
package org.litesoft.orsup;

import org.litesoft.commonfoundation.exceptions.*;
import org.litesoft.commonfoundation.typeutils.*;
import org.litesoft.core.simpletypes.temporal.*;
import org.litesoft.core.util.*;
import org.litesoft.logger.*;
import org.litesoft.orsup.base.*;
import org.litesoft.orsup.nonpublic.*;
import org.litesoft.orsup.otherattributeaccessors.*;
import org.litesoft.orsup.transact.*;
import org.litesoft.util.*;

import java.io.*;
import java.lang.reflect.*;
import org.litesoft.commonfoundation.typeutils.Objects;
import java.util.*;

public class DB_ImportSupport
{
    public static final String TODAY_RELATIVE = ":T";
    public static final String DELTA_DATE_RELATIVE = ":DD";
    public static final String DELTA_DATE_SEP = "+T-";
    public static final String FILE_ATTRIBUTE = "@";
    public static final String SETTER_ATTRIBUTE = "()";
    public static final String PO_URL_ATTRIBUTE = ":";

    protected static class ImportFromMap
    {
        public void process( Logger pLogger, UnfilteringFinder pFinder, File pReferenceDir, Map<FauxPOref, Attributes> pLineMap )
                throws IOException
        {
            if ( pLineMap.isEmpty() )
            {
                throw new IllegalArgumentException( "No Data Found" );
            }
            new Inner( pLogger, pFinder.createUnaugmentedTransaction(), pReferenceDir, pLineMap ).process().commit();
        }

        private class Inner
        {
            private Logger mLogger;
            private Transaction mTransaction;
            private File mReferenceDir;
            private Map<FauxPOref, Attributes> mLineMap;
            private Map<FauxPOref, PersistentObject<?>> mPOsByRef = new HashMap<FauxPOref, PersistentObject<?>>();

            private Inner( Logger pLogger, Transaction pTransaction, File pReferenceDir, Map<FauxPOref, Attributes> pLineMap )
            {
                mLogger = pLogger;
                mTransaction = pTransaction;
                mReferenceDir = pReferenceDir;
                mLineMap = pLineMap;
            }

            public Transaction process()
            {
                for ( FauxPOref ref : mLineMap.keySet() )
                {
                    putLineInDB( ref );
                }
                return mTransaction;
            }

            private PersistentObject<?> putLineInDB( FauxPOref pPOref )
            {
                PersistentObject<?> zPO = mPOsByRef.get( pPOref );
                if ( zPO == null ) // NOT Already created
                {
                    mPOsByRef.put( pPOref, zPO = mTransaction.create( pPOref.getPersistedObjectRegistrationName() ) );

                    Attributes attributes = mLineMap.get( pPOref );
                    if ( attributes == null )
                    {
                        throw new IllegalArgumentException( "No Attributes (Line) found for URL: " + pPOref );
                    }
                    mLogger.trace.log( "writing line to db for URL: ", pPOref );

                    for ( String attrName : attributes.getNames() )
                    {
                        String value = attributes.getValue( attrName );

                        // If the attribute is calling a setter method ..
                        if ( attrName.endsWith( SETTER_ATTRIBUTE ) )
                        {
                            try
                            {
                                String methodName = attrName.substring( 0, attrName.length() - SETTER_ATTRIBUTE.length() );
                                Method method = zPO.getClass().getMethod( methodName, String.class );
                                if ( method != null )
                                {
                                    method.invoke( zPO, value );
                                }
                            }
                            catch ( InvocationTargetException e )
                            {
                                throw new RuntimeException( e );
                            }
                            catch ( NoSuchMethodException e )
                            {
                                throw new RuntimeException( e );
                            }
                            catch ( IllegalAccessException e )
                            {
                                throw new RuntimeException( e );
                            }
                            // Setter methods aren't actual attributes, so skip the rest of the iteration.
                            continue;
                        }

                        boolean isURLAttribute = false;
                        if ( attrName.endsWith( PO_URL_ATTRIBUTE ) )
                        {
                            attrName = attrName.substring( 0, attrName.length() - PO_URL_ATTRIBUTE.length() );
                            isURLAttribute = true;
                        }
                        if ( attrName.endsWith( FILE_ATTRIBUTE ) )
                        {
                            attrName = attrName.substring( 0, attrName.length() - FILE_ATTRIBUTE.length() );
                            value = readFileAttributeValue( value );
                        }

                        AttributeAccessorSCD scd = zPO.getMetaDataForPO().getAccessorSCDoptional( attrName );
                        if ( scd == null )
                        {
                            throw new IllegalArgumentException( zPO.getObjectName() + " does NOT appear to have an attribute named: '" + attrName + "'" );
                        }
                        if ( !(scd instanceof NonImportableFeature) )
                        {
                            Object setValue = value;
                            if ( AttributeAccessorSCD.Form.ToOne == scd.getForm() )
                            {
                                if ( scd instanceof AttributeAccessorSCDtoOneVariable )
                                {
                                    PO_VarURLstringHelper zVarURL = new PO_VarURLstringHelper( value );
                                    FauxPOref childRef = new FauxPOref( mTransaction, zVarURL.getPersistentObjectURL() );
                                    PersistentObject<?> po = putLineInDB( childRef );
                                    setValue = new PO_VarURLstringHelper( zVarURL.getPersistedObjectAttributeName(), po.getPersistentObjectURL().toString() );
                                }
                                else
                                {
                                    FauxPOref childRef = new FauxPOref( mTransaction, value );
                                    setValue = putLineInDB( childRef );
                                }
                            }
                            else if ( looksLikeAReference( scd, value ) )
                            {
                                // support quasi-foreign keys.
                                FauxPOref childRef = new FauxPOref( mTransaction, value );
                                setValue = putLineInDB( childRef );
                                PersistentObject po = (PersistentObject) setValue;
                                try
                                {
                                    setValue = po.getPersistentObjectUniqueKey();
                                }
                                catch ( Exception e )
                                {
                                    e.printStackTrace();
                                }
                            }
                            else if ( isURLAttribute )
                            {
                                FauxPOref childRef = new FauxPOref( mTransaction, value );
                                setValue = putLineInDB( childRef ).getPersistentObjectURL().toString();
                            }
                            zPO.setAttributeValue( attrName, setValue );
                        }
                    }
                }
                return zPO;
            }

            private boolean looksLikeAReference( AttributeAccessorSCD pScd, String pValue )
            {
                if ( !(pScd.getColumnType().equals( Long.class )) )
                {
                    return false;
                }
                try
                {
                    new FauxPOref( mTransaction, pValue );
                    return true;
                }
                catch ( Exception e )
                {
                    return false;
                }
            }

            private String readFileAttributeValue( String fileReference )
            {
                if ( fileReference.startsWith( "file:" ) )
                {
                    fileReference = fileReference.substring( "file:".length() );
                }

                File f;
                if ( null != mReferenceDir )
                {
                    f = new File( mReferenceDir, fileReference );
                }
                else
                {
                    f = new File( fileReference );
                }

                return FileUtils.loadTextFileAsString( f, "\n" );
            }
        }
    }

    protected static class ImportToMap
    {
        private Map<FauxPOref, Attributes> mLineMap = new HashMap<FauxPOref, Attributes>();

        public ImportToMap process( Logger pLogger, Finder pFinder, Reader[] pReaders )
                throws IOException
        {
            if ( Objects.isNullOrEmpty( pReaders ) )
            {
                throw new IllegalArgumentException( "No Readers provided" );
            }
            try
            {
                new Inner( pLogger, pFinder ).process( pReaders );
            }
            finally
            {
                for ( Reader reader : pReaders )
                {
                    try
                    {
                        reader.close();
                    }
                    catch ( IOException e )
                    {
                        // Whatever!
                    }
                }
            }

            return this;
        }

        public Map<FauxPOref, Attributes> getLineMap()
        {
            return mLineMap;
        }

        private class Inner
        {
            private Logger mLogger;
            private Finder mFinder;
            private StringBuilder mBuilder;

            private Inner( Logger pLogger, Finder pFinder )
            {
                mLogger = pLogger;
                mFinder = pFinder;
            }

            public void process( Reader[] pReaders )
                    throws IOException
            {
                if ( pReaders[0] instanceof ToStringInformative )
                {
                    mLogger.info.log( "Importing:" );
                }
                for ( Reader reader : pReaders )
                {
                    if ( reader instanceof ToStringInformative )
                    {
                        mLogger.info.log( "    ", reader );
                    }
                    BufferedReader bufRead = Utils.getBufferedReader( reader );
                    mBuilder = new StringBuilder();
                    for ( String line; null != (line = bufRead.readLine()); )
                    {
                        putLineInMap( line );
                    }
                    if ( mBuilder.length() != 0 )
                    {
                        throw new UnclosedQuoteException( " did not find end of line " + mBuilder );
                    }
                }
            }

            private void putLineInMap( String pLine )
            {
                if ( mBuilder.length() == 0 )
                {
                    pLine = pLine.trim();
                    if ( (pLine.length() == 0) || (pLine.charAt( 0 ) == ';') || (pLine.charAt( 0 ) == '#') || pLine.startsWith( "//" ) )
                    {
                        return;
                    }
                }
                mBuilder.append( pLine );
                try
                {
                    putBuilderInMap();
                    mBuilder.setLength( 0 );
                }
                catch ( UnclosedQuoteException e )
                {
                    mBuilder.append( "\n" );
                }
            }

            private void putBuilderInMap()
            {
                String[] decodedData = new CsvSupport().decode( mBuilder.toString() );
                FauxPOref ref = new FauxPOref( mFinder, decodedData[0] );
                if ( mFinder.getMetaDataRequired( ref.getPersistedObjectRegistrationName() ).isImportable() )
                {
                    Attributes attributes = parseData( decodedData );
                    if ( !attributes.isEmpty() )
                    {
                        mLineMap.put( ref, attributes );
                    }
                }
            }

            private Attributes parseData( String[] pDecodedData )
            {
                Attributes zAttributes = new Attributes();
                for ( int i = 1; i < pDecodedData.length; ++i )
                {
                    String pair = Strings.noEmpty( pDecodedData[i] );
                    if ( pair != null )
                    {
                        int equidx = pair.indexOf( '=' );
                        if ( equidx == -1 )
                        {
                            throw new IllegalStateException( "No '=' in: " + pair );
                        }
                        String attr = pair.substring( 0, equidx ).trim();
                        String value = pair.substring( equidx + 1 ).trim();
                        zAttributes.setNameValue( attr, value );
                    }
                }
                return zAttributes;
            }
        }
    }

    protected static class FauxPOref
    {
        public static int sWildcardSequence = 0;

        private MetaDataForPO mPOmd;
        private String mIdentifier;

        public FauxPOref( Finder pFinder, String pToString )
        {
            PO_URLstringHelper helper = new PO_URLstringHelper( pToString );
            String zIdentifierOrRegistrationName = helper.getPersistedObjectRegistrationName();
            MetaDataForPO zMD = pFinder.getMetaDataOptionallyByIdentifier( zIdentifierOrRegistrationName );
            mPOmd = (zMD != null) ? zMD : pFinder.getMetaDataRequired( zIdentifierOrRegistrationName );
            mIdentifier = helper.getPersistentObjectUniqueKey();
            if ( "?".equals( mIdentifier ) )
            {
                mIdentifier = "gened-" + ++sWildcardSequence;
            }
        }

        public MetaDataForPO getPOmd()
        {
            return mPOmd;
        }

        public String getPersistedObjectRegistrationName()
        {
            return mPOmd.getPOregistrationName();
        }

        public String getIdentifier()
        {
            return mIdentifier;
        }

        @Override
        public boolean equals( Object o )
        {
            return (this == o) || ((o instanceof FauxPOref) && equals( (FauxPOref) o ));
        }

        public boolean equals( FauxPOref them )
        {
            return (this == them) || //
                   ((them != null) //
                    && (this.mPOmd == them.mPOmd) //
                    && this.mIdentifier.equals( them.mIdentifier ) //
                   );
        }

        @Override
        public int hashCode()
        {
            return 31 * getPersistedObjectRegistrationName().hashCode() + mIdentifier.hashCode();
        }

        @Override
        public String toString()
        {
            return new PO_URLstringHelper( getPersistedObjectRegistrationName(), mIdentifier ).toString();
        }
    }

    protected static class Attributes
    {
        private Map<String, String> mNameValues = new HashMap<String, String>();

        public boolean isEmpty()
        {
            return mNameValues.isEmpty();
        }

        public String[] getNames()
        {
            return mNameValues.keySet().toArray( new String[mNameValues.size()] );
        }

        public String getValue( String pName )
        {
            return mNameValues.get( pName );
        }

        public void setNameValue( String pName, String pValue )
        {
            if ( pName.endsWith( TODAY_RELATIVE ) )
            {
                pName = pName.substring( 0, pName.length() - TODAY_RELATIVE.length() ).trim();
                pValue = makeTodayRelative( pValue );
            }
            else if ( pName.endsWith( DELTA_DATE_RELATIVE ) )
            {
                pName = pName.substring( 0, pName.length() - DELTA_DATE_RELATIVE.length() ).trim();
                pValue = makeDeltaDateRelative( pValue );
            }
            if ( pName.length() != 0 )
            {
                String curValue = (pValue.length() == 0) ? mNameValues.get( pName ) : mNameValues.put( pName, pValue );
                if ( curValue != null )
                {
                    throw new IllegalArgumentException( "Duplicate Attribute for: " + pName );
                }
            }
        }

        private String makeTodayRelative( String pValue )
        {
            boolean zAdd;
            switch ( (pValue = pValue + " ").charAt( 0 ) )
            {
                case '+':
                    zAdd = true;
                    break;
                case '-':
                    zAdd = false;
                    break;
                default:
                    throw new IllegalArgumentException( "Neither a '+' nor a '-' after '" + TODAY_RELATIVE + "='" );
            }
            String zYears, zMonths, zDays;
            String[] zParts = Strings.parseChar( pValue.substring( 1 ).trim(), '/' );
            switch ( zParts.length )
            {
                case 3:
                    zYears = zParts[0];
                    zMonths = zParts[1];
                    zDays = zParts[2];
                    break;
                case 2:
                    zYears = "";
                    zMonths = zParts[0];
                    zDays = zParts[1];
                    break;
                case 1:
                    zYears = zMonths = "";
                    zDays = zParts[0];
                    break;
                default:
                    throw new IllegalArgumentException( "Too many '/'s after '" + TODAY_RELATIVE + "='" );
            }
            int iYears = parseInt( "Year", zYears );
            int iMonths = parseInt( "Month", zMonths );
            int iDays = parseInt( "Day", zDays );

            CalendarYMD zDesiredDate = CalendarYMD.today();

            if ( zAdd )
            {
                zDesiredDate = zDesiredDate.addYears( iYears );
                zDesiredDate = zDesiredDate.addMonths( iMonths );
                zDesiredDate = zDesiredDate.addDays( iDays );
            }
            else
            {
                zDesiredDate = zDesiredDate.minusYears( iYears );
                zDesiredDate = zDesiredDate.minusMonths( iMonths );
                zDesiredDate = zDesiredDate.minusDays( iDays );
            }

            return zDesiredDate.toSQLvalue();
        }

        private String makeDeltaDateRelative( String pValue )
        {
            int zAt = pValue.indexOf( DELTA_DATE_SEP );
            if ( zAt == -1 )
            {
                throw new IllegalArgumentException( "No Delta Date Sep '" + DELTA_DATE_SEP + "'" );
            }
            CalendarYMD zDesiredDate = to_CalendarYMD( "Desired Date", pValue.substring( 0, zAt ) );
            CalendarYMD zDeltaDate = to_CalendarYMD( "Delta Date", pValue.substring( zAt + DELTA_DATE_SEP.length() ) );
            CalendarYMD zToday = CalendarYMD.today();
            zDesiredDate = zDesiredDate.addDays( zDeltaDate.daysTill( zToday ) );
            return zDesiredDate.toString();
        }

        private CalendarYMD to_CalendarYMD( String pWhat, String pString )
        {
            try
            {
                return TypeConverter.to_CalendarYMD( pString.trim() );
            }
            catch ( RuntimeException e )
            {
                throw new IllegalArgumentException( pWhat + ":" + e.getMessage(), e );
            }
        }

        private int parseInt( String pWhat, String pString )
        {
            if ( (pString = pString.trim()).length() == 0 )
            {
                return 0;
            }
            try
            {
                return Integer.parseInt( pString );
            }
            catch ( NumberFormatException e )
            {
                throw new IllegalArgumentException( pWhat + ":" + e.getMessage(), e );
            }
        }
    }
}