KeePassDX/app/src/main/java/com/keepassdroid/EntryEditActivityV4.java

/*
 * Copyright 2017 Brian Pellin, Jeremy Jamet / Kunzisoft.
 *     
 * This file is part of KeePassDroid.
 *
 *  KeePassDroid is free software: you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation, either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  KeePassDroid is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with KeePassDroid.  If not, see <http://www.gnu.org/licenses/>.
 *
 */
package com.keepassdroid;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;
import java.util.UUID;

import android.content.Context;
import android.content.Intent;
import android.os.Bundle;
import android.view.LayoutInflater;
import android.view.View;
import android.widget.CheckBox;
import android.widget.ImageButton;
import android.widget.LinearLayout;
import android.widget.RelativeLayout;
import android.widget.ScrollView;
import android.widget.TextView;
import android.widget.Toast;

import com.android.keepass.R;
import com.keepassdroid.app.App;
import com.keepassdroid.database.PwDatabaseV4;
import com.keepassdroid.database.PwEntry;
import com.keepassdroid.database.PwEntryV4;
import com.keepassdroid.database.PwGroupId;
import com.keepassdroid.database.PwGroupIdV4;
import com.keepassdroid.database.PwGroupV4;
import com.keepassdroid.database.security.ProtectedString;
import com.keepassdroid.utils.Types;
import com.keepassdroid.view.EntryEditSection;

public class EntryEditActivityV4 extends EntryEditActivity {
	
	private ScrollView scroll;
	private LayoutInflater inflater;

	protected static void putParentId(Intent i, String parentKey, PwGroupV4 parent) {
		PwGroupId id = parent.getId();
		PwGroupIdV4 id4 = (PwGroupIdV4) id;
		
		i.putExtra(parentKey, Types.UUIDtoBytes(id4.getId()));
		
	}

	@Override
	protected PwGroupId getParentGroupId(Intent i, String key) {
		byte[] buf = i.getByteArrayExtra(key);
		UUID id = Types.bytestoUUID(buf);
		
		return new PwGroupIdV4(id);
	}
	
	@Override
	protected void onCreate(Bundle savedInstanceState) {
		inflater = (LayoutInflater) this.getSystemService(Context.LAYOUT_INFLATER_SERVICE);
		
		super.onCreate(savedInstanceState);
		
		scroll = (ScrollView) findViewById(R.id.entry_scroll);
		
		View add = findViewById(R.id.add_advanced);
		add.setVisibility(View.VISIBLE);
		add.setOnClickListener(new View.OnClickListener() {
			
			@Override
			public void onClick(View v) {
				LinearLayout container = (LinearLayout) findViewById(R.id.advanced_container);
				
				EntryEditSection ees = (EntryEditSection) inflater.inflate(R.layout.entry_edit_section, container, false);
				ees.setData("", new ProtectedString(false, ""));
				container.addView(ees);
				
				// Scroll bottom
				scroll.post(new Runnable() {
					@Override
					public void run() {
						scroll.fullScroll(ScrollView.FOCUS_DOWN);
					}
				});
				
			}
		});
		
		View iconPicker = findViewById(R.id.icon_button);
		iconPicker.setVisibility(View.GONE);
		
		View divider = findViewById(R.id.divider_title);
		RelativeLayout.LayoutParams lp_div = (RelativeLayout.LayoutParams) divider.getLayoutParams();
		lp_div.addRule(RelativeLayout.BELOW, R.id.entry_title);
		
		View user_label = findViewById(R.id.entry_user_name_label);
		RelativeLayout.LayoutParams lp = (RelativeLayout.LayoutParams) user_label.getLayoutParams();
		lp.addRule(RelativeLayout.BELOW, R.id.divider_title);
	}

	@Override
	protected void fillData() {
		super.fillData();
		
		PwEntryV4 entry = (PwEntryV4) mEntry;
		
		LinearLayout container = (LinearLayout) findViewById(R.id.advanced_container);
		
		if (entry.strings.size() > 0) {
			for (Entry<String, ProtectedString> pair : entry.strings.entrySet()) {
				String key = pair.getKey();
				
				if (!PwEntryV4.IsStandardString(key)) {
					EntryEditSection ees = (EntryEditSection) inflater.inflate(R.layout.entry_edit_section, container, false);
					ees.setData(key, pair.getValue());
					container.addView(ees);
				}
			}
			
		}
		
	}

	@SuppressWarnings("unchecked")
	@Override
	protected PwEntry populateNewEntry() {
		PwEntryV4 newEntry = (PwEntryV4) mEntry.clone(true);
		newEntry.history = (ArrayList<PwEntryV4>) newEntry.history.clone();
		newEntry.createBackup((PwDatabaseV4)App.getDB().pm);
		
		newEntry = (PwEntryV4) super.populateNewEntry(newEntry);
		
		Map<String, ProtectedString> strings = newEntry.strings;
		
		// Delete all new standard strings
		Iterator<Entry<String, ProtectedString>> iter = strings.entrySet().iterator();
		while (iter.hasNext()) {
			Entry<String, ProtectedString> pair = iter.next();
			if (!PwEntryV4.IsStandardString(pair.getKey())) {
				iter.remove();
			}
		}
		
		LinearLayout container = (LinearLayout) findViewById(R.id.advanced_container);
		for (int i = 0; i < container.getChildCount(); i++) {
			View view = container.getChildAt(i);
			
			TextView keyView = (TextView)view.findViewById(R.id.title);
			String key = keyView.getText().toString();
			
			TextView valueView = (TextView)view.findViewById(R.id.value);
			String value = valueView.getText().toString();
			
			CheckBox cb = (CheckBox)view.findViewById(R.id.protection);
			boolean protect = cb.isChecked();
			
			strings.put(key, new ProtectedString(protect, value));
		}
		
		return newEntry;
	}
	
	public void deleteAdvancedString(View view) {
		EntryEditSection section = (EntryEditSection) view.getParent();
		LinearLayout container = (LinearLayout) findViewById(R.id.advanced_container);
		
		for (int i = 0; i < container.getChildCount(); i++) {
			EntryEditSection ees = (EntryEditSection) container.getChildAt(i);
			if (ees == section) {
				container.removeViewAt(i);
				container.invalidate();
				break;
			}
		}
	}

	@Override
	protected boolean validateBeforeSaving() {
		if(!super.validateBeforeSaving()) {
			return false;
		}
		
		LinearLayout container = (LinearLayout) findViewById(R.id.advanced_container);
		for (int i = 0; i < container.getChildCount(); i++) {
			EntryEditSection ees = (EntryEditSection) container.getChildAt(i);
			
			TextView keyView = (TextView) ees.findViewById(R.id.title);
			CharSequence key = keyView.getText();
			
			if (key == null || key.length() == 0) {
				Toast.makeText(this, R.string.error_string_key, Toast.LENGTH_LONG).show();
				return false;
			}
			
		}
		
		return true;
	}

}