//---------------------------------------------------------------------------------------
// Copyright (c) 2001-2008 by PDFTron Systems Inc. All Rights Reserved.
// Consult legal.txt regarding legal and license information.
//---------------------------------------------------------------------------------------

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;

import pdftron.Common.PDFNetException;
import pdftron.Filters.FilterReader;
import pdftron.Filters.FlateEncode;
import pdftron.Filters.StdFile;
import pdftron.PDF.*;
import pdftron.SDF.*;


//---------------------------------------------------------------------------------------
// This sample shows encryption support in PDFNet. The sample reads an encrypted document and 
// sets a new SecurityHandler. The sample also illustrates how password protection can 
// be removed from an existing PDF document.
//---------------------------------------------------------------------------------------
public class EncTest
{
	public static void main(String[] args)
	{
		PDFNet.initialize();
		PDFNet.setResourcesPath("../../../resources");

		// Relative path to the folder containing test files.
		String input_path =  "../../TestFiles/";
		String output_path = "../../TestFiles/Output/";

		// Example 1: 
		// secure a document with password protection and 
		// adjust permissions 
		
		try
		{
			// Open the test file
			System.out.println("Securing an existing document ...");
			PDFDoc doc=new PDFDoc((input_path + "fish.pdf"));


			
			// Perform some operation on the document. In this case we use low level SDF API
			// to replace the content stream of the first page with contents of file 'my_stream.txt'
			if (true)  // Optional
			{
				System.out.println("Repacing the content stream, use flate compression...");

				// Get the page dictionary using the following path: trailer/Root/Pages/Kids/0
				Obj page_dict = doc.getTrailer().get("Root").value()
					.get("Pages").value()
					.get("Kids").value()
					.getAt(0);

				// Embed a custom stream (file mystream.txt) using Flate compression.
				StdFile embed_file=new StdFile((input_path + "my_stream.txt"), StdFile.e_read_mode );
				FilterReader mystm=new FilterReader(embed_file);
				page_dict.put("Contents", 
					doc.createIndirectStream(mystm,  
					new FlateEncode(null)));
			}

			//encrypt the document


			// Apply a new security handler with given security settings. 
			// In order to open saved PDF you will need a user password 'test'.
			SecurityHandler new_handler=new SecurityHandler();

			// Set a new password required to open a document
			String user_password="test";
			new_handler.changeUserPassword(user_password);

			// Set Permissions
			new_handler.setPermission (SecurityHandler.e_print, true);
			new_handler.setPermission (SecurityHandler.e_extract_content, false);

			// Note: document takes the ownership of new_handler.
			doc.setSecurityHandler(new_handler);

			// Save the changes.
			System.out.println("Saving modified file...");
			doc.save((output_path + "secured.pdf"), 0, null);

		}
		catch(PDFNetException e) {
			e.printStackTrace();
		}

		// Example 2:
		// Opens the encrypted document and removes all of 
		// its security.
		try
		{
			PDFDoc doc=new PDFDoc((output_path + "secured.pdf"));

			//If the document is encrypted prompt for the password
			if (!doc.initSecurityHandler()) 
			{
				boolean success=false;
				System.out.println("The password is: test");
				for(int count=0; count<3;count++)
				{
					BufferedReader r=new BufferedReader(new InputStreamReader(System.in));
					System.out.println("A password required to open the document.");
					System.out.print("Please enter the password:");
					String password=r.readLine();
					if(doc.initStdSecurityHandler(password))
					{
						success=true;
						System.out.println("The password is correct.");
						break;
					}
					else if(count<3)
					{
						System.out.println("The password is incorrect, please try again");
					}
				}
				if(!success)
				{
					System.out.println("Document authentication error....");
					PDFNet.terminate();
				}
			}

			//remove all security on the document
			doc.removeSecurity();
			doc.save((output_path + "not_secured.pdf"), 0, null);

		}
		catch(PDFNetException e) {
			e.printStackTrace();
		}
		catch(IOException e2) {
			e2.printStackTrace();
		}

		System.out.println("Test completed.");
		PDFNet.terminate();
	}
}