/* Makefile
debug: xdwextpage2.cpp
	gcc -g -O0 -I. xdwextpage2.cpp xdwapi.lib
	#date
	#./a.exe target.xdw
	#date

release: xdwextpage2.cpp
	gcc -I. xdwextpage2.cpp xdwapi.lib -static -o xdwextpage2.exe
	strip xdwextpage2.exe

*/

#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <time.h>
#include <io.h>
#include <windows.h>
#include <xdw_api.h>

#define MAXCOL  1024
#define MAXLINE 9999

void print_error( int code ) {
	fprintf( stderr, "Error code : %d\n", code );
	switch ( code ) {
	case XDW_E_NOT_INSTALLED:
		fprintf( stderr, "DocuWorksがインストールされていません。" );
		break;
	case XDW_E_FILE_NOT_FOUND:
		fprintf( stderr, "指定されたファイルが見つかりません。" );
		break;
	case XDW_E_FILE_EXISTS:
		fprintf( stderr, "指定されたファイルはすでに存在します。" );
		break;
	case XDW_E_ACCESSDENIED:
	case XDW_E_INVALID_NAME:
	case XDW_E_BAD_NETPATH:
		fprintf( stderr, "指定されたファイルを開くことができません。" );
		break;
	case XDW_E_BAD_FORMAT:
		fprintf( stderr, "指定されたファイルは正しいフォーマットではありません。" );
		break;
	case XDW_E_INVALID_ACCESS:
		fprintf( stderr, "指定された操作をする権利がありません。" );
		break;
	default:
		fprintf( stderr, "エラーが発生しました。" );
		break;
	}
}

void print_help( const char *prog ) {
	fprintf( stderr, "%s p infile outfile\n",     prog );
	fprintf( stderr, "%s -i infile outfile\n",    prog );
	fprintf( stderr, "%s -i -v infile outfile\n", prog );
	fprintf( stderr, "%s -l p infile outfile\n",  prog );
	fprintf( stderr, "%s p:q infile outfile\n",   prog );
	fprintf( stderr, "%s p:  infile outfile\n",   prog );
	fprintf( stderr, "%s  :q infile outfile\n",   prog );
}

void print_now( const char *msg ) {
   time_t now = time( NULL );
   struct tm *ts = localtime( &now );

   char buf[80];
   strftime( buf, sizeof( buf ), "%H:%M:%S %Z", ts );
   printf( "%s\t%s\n", buf, msg );
}

int main( int argc, char* argv[] ) {

	/* オプションの解析 */
	char prog[128];
	strcpy( prog, argv[0] );

	bool pnow  = false;
	bool plist = false;
	bool inv   = false;
	char c;
	while ( --argc > 0 && ( *++argv )[0] == '-' ) {
		while ( c = *++argv[0] ) {
			switch ( c ) {
				case 'l':
					pnow = true;
					print_now( "start." );
					break;
				case 'v':
					inv = true;
					break;
				case 'i':
					plist = true;
					break;
				case 'h':
					print_help( prog );
					exit( 1 );
				default:
					fprintf( stderr, "error: illegal option '%c'.\n", c );
					exit( 1 );
			}
		}
	}

	if ( argc != 3 && !plist ) {
		print_help( prog );
		exit( 1 );
	}

	/* メイン処理 */

	char in_path[ _MAX_PATH ], out_path[ _MAX_PATH ], tmp_path[ _MAX_PATH ];
	_fullpath( in_path,  argv[1], _MAX_PATH );
	_fullpath( out_path, argv[2], _MAX_PATH );
	if ( plist ) {
		_fullpath( in_path,  argv[0], _MAX_PATH );
		_fullpath( out_path, argv[1], _MAX_PATH );
	}
	_fullpath( tmp_path, "tempXXXX.xdw", _MAX_PATH );

	remove( tmp_path );
	remove( out_path );

	char buf[ MAXCOL ];
	sprintf( buf, "copy %s %s", in_path, tmp_path );
	system( buf );

	int api_result = 0;

	// 文書ハンドルを開く
	XDW_DOCUMENT_HANDLE h = NULL;
	XDW_OPEN_MODE_EX mode = { sizeof( XDW_OPEN_MODE_EX ), XDW_OPEN_UPDATE, XDW_AUTH_NODIALOGUE };
	api_result = XDW_OpenDocumentHandle( tmp_path, &h, ( XDW_OPEN_MODE* )&mode );
	if ( api_result < 0 ) {
		print_error( api_result );
		return 1;
	}

	XDW_DOCUMENT_INFO info = { sizeof( XDW_DOCUMENT_INFO ), 0, 0, 0 }; // 総ページ数を得る
	XDW_GetDocumentInformation( h, &info );
	int last_page = info.nPages;

	if ( plist ) {
		/* リストの取り込み */
		char *al = (char*)malloc( MAXLINE * sizeof( char ) * MAXCOL );
		int alN = 0;
		if ( plist ) {
			if ( pnow ) print_now( "reading list." );
			FILE *fp;

			if ( al == NULL ) {
				fprintf( stderr, "can't allocate memory\n" );
				exit( 1 );
			}

			if ( ( fp = fopen( "extpage.list", "r" ) ) == NULL ) {
				fprintf( stderr, "%s: can't open file [extpage.list]\n", prog );
				exit ( 1 );
			}
			char *p;
			while ( fgets( buf, sizeof buf, fp ) ) {
				if ( !strncmp( buf, "#",  1 ) ) continue;
				if ( !strncmp( buf, "//", 2 ) ) continue;
				if ( !strcmp( buf, "\n" )     ) continue;

				if ( ( p = strchr( buf, '\n' ) ) != NULL ) {
					*p = '\0';
				}
				strncpy( &al[ alN * MAXCOL ], buf, MAXCOL );
				alN++;
			}
			fclose( fp );
		}

		bool ext[MAXLINE];
		for ( int p = 0; p < last_page; p++ ) ext[p] = false;
		for ( int i = 0; i < alN; i++ ) {
			int p = atoi( &al[ i * MAXCOL ] );
			if ( p != 0 ) ext[ p - 1 ] = true;
		}
		if ( inv ) for ( int p = 0; p < last_page; p++ ) ext[p] = !ext[p];

		for ( int p = last_page; p > 0; p-- ) {
			if ( !ext[ p - 1 ] ) {
				api_result = XDW_DeletePage( h, p, NULL );
				if ( api_result < 0 ) {
					print_error( api_result );
					exit( 1 );
				}
			}
		}
	}
	else {
		int bgn = 1;
		int end = 0;
		if ( strchr( argv[0], ':' ) == NULL ) {
			bgn = end = atoi( argv[0] );
		} else {
			if ( strcmp( argv[0], ":" ) == 0 ) {
				fprintf( stderr, "bad page range.\n" );
				exit( 1 );
			}
			if ( argv[0][0] == ':' ) {
				end = atoi( strtok( argv[0], ":" ) );
			} else {
				bgn = atoi( strtok( argv[0], ":" ) );
				end = atoi( strtok( NULL, ":" ) );
				if ( end == 0 ) end = last_page;
			}
		}

		//printf( "bgn = %d, end = %d\n", bgn, end );
		if ( bgn < 1 || end < 1 || bgn > end || bgn > last_page || end > last_page ) {
			fprintf( stderr, "bad page range.\n" );
			exit( 1 );
		}

		for ( int i = last_page; i > 0; i-- ) {
			if ( i < bgn || i > end ) api_result = XDW_DeletePage( h, i, NULL );
		}
	}

	// 文書ハンドルを閉じる
	XDW_SaveDocument( h, NULL );
	XDW_CloseDocumentHandle( h, NULL );

	// 最適化
	api_result = XDW_OptimizeDocument( tmp_path, out_path, NULL );
	if ( api_result < 0 ) {
		print_error( api_result );
		exit( 1 );
	}
	remove( tmp_path );

	if ( pnow ) print_now( "done." );
	return 0;
}