/* Makefile
debug: xdwaddpage2.cpp
	#cls
	gcc -g -O0 -I. xdwaddpage2.cpp xdwapi.lib

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

clean:
	rm -rf tempXXXX
*/

#include <unistd.h>
#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>
#include <xdwapian.h>


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() {
	char prog[12] = "xdwaddpage2";
	fprintf( stderr, "Usage:\n" );
	fprintf( stderr, " %s infile\n", prog );
	fprintf( stderr, " %s -l infile\n", prog );
	fprintf( stderr, " %s -ltxt infile\n", prog );
	fprintf( stderr, " %s -s S infile\n", prog );
	fprintf( stderr, " %s -c infile1 infile2 ...\n", prog );
	fprintf( stderr, " %s -s S -c infile1 infile2 ...\n", prog );
	fprintf( stderr, " %s -s S -ltxt -c infile1 infile2 ...\n", prog );
	fprintf( stderr, " %s -v infile\n", prog );
}

void print_now( char const *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 add_annotation( XDW_DOCUMENT_HANDLE h, int page, int x, int y, char* string, int* sz, int tr )
{
	// テキストアノテーションを作成する
	XDW_ANNOTATION_HANDLE annoation;
	int api_result = XDW_AddAnnotation( h, XDW_AID_TEXT, page, x, y, NULL, &annoation, NULL );
	if ( api_result < 0 ) return api_result;

	// 作成したアノテーションに文字列を設定する
	api_result = XDW_SetAnnotationAttribute( h, annoation, XDW_ATN_Text, XDW_ATYPE_STRING, string, 0, NULL ); // 文字テキスト
	api_result = XDW_SetAnnotationAttribute( h, annoation, XDW_ATN_FontSize, XDW_ATYPE_INT, reinterpret_cast<char*>(sz), 0, NULL ); // フォントサイズ
    if ( tr ) {
		int color = XDW_COLOR_NONE;
		api_result = XDW_SetAnnotationAttribute( h, annoation, XDW_ATN_BackColor, XDW_ATYPE_INT, reinterpret_cast<char*>(&color), 0, NULL );
    }

	return api_result;
}

int add_page( const char* file, int sp, bool pleft, const char* ltext )
{
	XDW_DOCUMENT_HANDLE h = NULL; // 文書ハンドルを開く
	XDW_OPEN_MODE_EX mode = { sizeof( XDW_OPEN_MODE_EX ), XDW_OPEN_UPDATE, XDW_AUTH_NODIALOGUE };
	int api_result = XDW_OpenDocumentHandle( file, &h, (XDW_OPEN_MODE*)&mode );
	if ( api_result < 0 ) {
		print_error( api_result );
		return 0;
	}

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

    int sz = 80;    // 文字サイズ
    int tr = 1;     // 透明フラグ
    char pagenum[18], lpagenum[30];

	// アノテーションを貼り付ける
    for ( int p = 0; p < last_page; p++ ) {
        sprintf( pagenum, "%05d-%04d-%04d", p + sp, p + 1, last_page );
        api_result = add_annotation( h, p + 1, 17500, 28500, pagenum, &sz, tr );        // 右下
		if ( pleft ) {
			if ( ltext != NULL ) 
				sprintf( lpagenum, "%s%s", pagenum, ltext );
			else
				sprintf( lpagenum, "%s", pagenum );
			api_result = add_annotation( h, p + 1,  1700,  5600, lpagenum, &sz, tr );    // 左上
		}
        if ( api_result < 0 ) {
            print_error( api_result );
            break;
        }
    }

	if ( api_result >= 0 ) // 変更をファイルに反映する
		api_result = XDW_SaveDocument( h, NULL );

	XDW_CloseDocumentHandle( h, NULL ); // 文書ハンドルを閉じる
	return sp + last_page;
}

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

	/* オプションの解析 */
	bool pnow  = false;
	bool pleft = false;
	bool chain = false;
	char *ltext = NULL;
	int sp = 1;

	int opt;
	while ( ( opt = getopt( argc, argv, "vs:l::c" ) ) != -1 ) {
		switch ( opt ) {
			case 'v':
				pnow = true;
				print_now( "start." );
				break;
			case 'l':
				pleft = true;
				ltext = optarg;
				break;
			case 's':
				sp = atoi( optarg );
				if ( sp == 0 ) {
					print_help();
					exit( 1 );
				}
				break;
			case 'c':	// chain mode
				chain = true;
				break;
			default:
				print_help();
				exit( 1 );
		}
	}

	char in_path[ _MAX_PATH ];

	/* 本処理 */
	int api_result = 0;
	if ( chain ) {
		int lp = sp;
		for ( int i = optind; i < argc; i++ ) {
			_fullpath( in_path, argv[ i ], _MAX_PATH );
			lp = add_page( in_path, lp, pleft, ltext );
			if ( pnow ) print_now( in_path );
		}
	}
	else {
		_fullpath( in_path, argv[ argc - 1 ], _MAX_PATH );
		add_page( in_path, sp, pleft, ltext );
	}

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