src/main/java/com/gitblit/wicket/pages/FilestorePage.java - gitblit - Git at Google

 /*
  * Copyright 2015 gitblit.com.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
  * You may obtain a copy of the License at
  *
  *     http://www.apache.org/licenses/LICENSE-2.0
  *
  * Unless required by applicable law or agreed to in writing, software
  * distributed under the License is distributed on an "AS IS" BASIS,
  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  * See the License for the specific language governing permissions and
  * limitations under the License.
  */
 package com.gitblit.wicket.pages;

 import java.text.DateFormat;
 import java.text.SimpleDateFormat;
 import java.util.ArrayList;
 import java.util.List;

 import org.apache.commons.io.FileUtils;
 import org.apache.wicket.PageParameters;
 import org.apache.wicket.markup.html.basic.Label;
 import org.apache.wicket.markup.html.link.BookmarkablePageLink;
 import org.apache.wicket.markup.repeater.Item;
 import org.apache.wicket.markup.repeater.data.DataView;
 import org.apache.wicket.markup.repeater.data.ListDataProvider;

 import com.gitblit.Constants;
 import com.gitblit.Keys;
 import com.gitblit.models.FilestoreModel;
 import com.gitblit.models.FilestoreModel.Status;
 import com.gitblit.models.UserModel;
 import com.gitblit.wicket.CacheControl;
 import com.gitblit.wicket.FilestoreUI;
 import com.gitblit.wicket.GitBlitWebSession;
 import com.gitblit.wicket.WicketUtils;
 import com.gitblit.wicket.CacheControl.LastModified;

 /**
  * Page to display the current status of the filestore.
  * Certain errors also displayed to aid in fault finding
  *
  * @author Paul Martin
  */
 @CacheControl(LastModified.ACTIVITY)
 public class FilestorePage extends RootPage {

 	public FilestorePage(PageParameters params) {
 		super(params);
 		setupPage("", "");

 		int itemsPerPage = app().settings().getInteger(Keys.web.itemsPerPage, 20);
 		if (itemsPerPage <= 1) {
 			itemsPerPage = 20;
 		}

 		final int pageNumber = WicketUtils.getPage(params);
 		final String filter = WicketUtils.getSearchString(params);

 		int prevPage = Math.max(0, pageNumber - 1);
 		int nextPage = pageNumber + 1;
 		boolean hasMore = false;

 		final UserModel user = (GitBlitWebSession.get().getUser() == null) ? UserModel.ANONYMOUS : GitBlitWebSession.get().getUser();
 		final long nBytesUsed = app().filestore().getFilestoreUsedByteCount();
 		final long nBytesAvailable = app().filestore().getFilestoreAvailableByteCount();
 		List<FilestoreModel> files = app().filestore().getAllObjects(app().repositories().getRepositoryModels(user));

 		if (files == null) {
 			files = new ArrayList<FilestoreModel>();
 		}

 		long nOk = 0;
 		long nPending = 0;
 		long nInprogress = 0;
 		long nError = 0;
 		long nDeleted = 0;

 		for (FilestoreModel file : files) {
 			switch (file.getStatus()) {
 			case Available: { nOk++;} break;
 			case Upload_Pending: { nPending++; } break;
 			case Upload_In_Progress: { nInprogress++; } break;
 			case Deleted: { nDeleted++; } break;
 			default: { nError++; } break;
 			}
 		}


 		BookmarkablePageLink<Void> itemOk = new BookmarkablePageLink<Void>("filterByOk", FilestorePage.class,
 				WicketUtils.newFilestorePageParameter(prevPage, SortBy.ok.name()));

 		BookmarkablePageLink<Void> itemPending = new BookmarkablePageLink<Void>("filterByPending", FilestorePage.class,
 				WicketUtils.newFilestorePageParameter(prevPage, SortBy.pending.name()));

 		BookmarkablePageLink<Void> itemInprogress = new BookmarkablePageLink<Void>("filterByInprogress", FilestorePage.class,
 				WicketUtils.newFilestorePageParameter(prevPage, SortBy.inprogress.name()));

 		BookmarkablePageLink<Void> itemError = new BookmarkablePageLink<Void>("filterByError", FilestorePage.class,
 				WicketUtils.newFilestorePageParameter(prevPage, SortBy.error.name()));

 		BookmarkablePageLink<Void> itemDeleted = new BookmarkablePageLink<Void>("filterByDeleted", FilestorePage.class,
 				WicketUtils.newFilestorePageParameter(prevPage, SortBy.deleted.name()));


 		List<FilestoreModel> filteredResults = new ArrayList<FilestoreModel>(files.size());

 		if (filter == null) {
 			filteredResults = files;
 		} else if (filter.equals(SortBy.ok.name())) {
 			WicketUtils.setCssClass(itemOk, "filter-on");

 			for (FilestoreModel item : files) {
 				if (item.getStatus() == Status.Available) {
 					filteredResults.add(item);
 				}
 			}
 		} else if (filter.equals(SortBy.pending.name())) {
 			WicketUtils.setCssClass(itemPending, "filter-on");

 			for (FilestoreModel item : files) {
 				if (item.getStatus() == Status.Upload_Pending) {
 					filteredResults.add(item);
 				}
 			}
 		} else if (filter.equals(SortBy.inprogress.name())) {
 			WicketUtils.setCssClass(itemInprogress, "filter-on");

 			for (FilestoreModel item : files) {
 				if (item.getStatus() == Status.Upload_In_Progress) {
 					filteredResults.add(item);
 				}
 			}
 		} else if (filter.equals(SortBy.error.name())) {
 			WicketUtils.setCssClass(itemError, "filter-on");

 			for (FilestoreModel item : files) {
 				if (item.isInErrorState()) {
 					filteredResults.add(item);
 				}
 			}
 		} else if (filter.equals(SortBy.deleted.name())) {
 			WicketUtils.setCssClass(itemDeleted, "filter-on");

 			for (FilestoreModel item : files) {
 				if (item.getStatus() == Status.Deleted) {
 					filteredResults.add(item);
 				}
 			}
 		}

 		DataView<FilestoreModel> filesView = new DataView<FilestoreModel>("fileRow",
 				new ListDataProvider<FilestoreModel>(filteredResults) , itemsPerPage) {
 			private static final long serialVersionUID = 1L;
 			private int counter;

 			@Override
 			protected void onBeforeRender() {
 				super.onBeforeRender();
 				counter = 0;
 			}

 			@Override
 			public void populateItem(final Item<FilestoreModel> item) {
 				final FilestoreModel entry = item.getModelObject();

 				DateFormat dateFormater = new SimpleDateFormat(Constants.ISO8601);

 				UserModel user = app().users().getUserModel(entry.getChangedBy());
 				user = user == null ? UserModel.ANONYMOUS : user;

 				Label icon = FilestoreUI.getStatusIcon("status", entry);
 				item.add(icon);
 				item.add(new Label("on", dateFormater.format(entry.getChangedOn())));
 				item.add(new Label("by", user.getDisplayName()));

 				item.add(new Label("oid", entry.oid));
 				item.add(new Label("size", FileUtils.byteCountToDisplaySize(entry.getSize())));

 				WicketUtils.setAlternatingBackground(item, counter);
 				counter++;
 			}

 		};


 		if (filteredResults.size() < itemsPerPage) {
 			filesView.setCurrentPage(0);
 			hasMore = false;
 		} else {
 			filesView.setCurrentPage(pageNumber - 1);
 			hasMore = true;
 		}


 		add(filesView);


 		add(new BookmarkablePageLink<Void>("firstPageBottom", FilestorePage.class).setEnabled(pageNumber > 1));
 		add(new BookmarkablePageLink<Void>("prevPageBottom", FilestorePage.class,
 				WicketUtils.newFilestorePageParameter(prevPage, filter)).setEnabled(pageNumber > 1));
 		add(new BookmarkablePageLink<Void>("nextPageBottom", FilestorePage.class,
 				WicketUtils.newFilestorePageParameter(nextPage, filter)).setEnabled(hasMore));


 		itemOk.add(FilestoreUI.getStatusIcon("statusOkIcon", FilestoreModel.Status.Available));
 		itemPending.add(FilestoreUI.getStatusIcon("statusPendingIcon", FilestoreModel.Status.Upload_Pending));
 		itemInprogress.add(FilestoreUI.getStatusIcon("statusInprogressIcon", FilestoreModel.Status.Upload_In_Progress));
 		itemError.add(FilestoreUI.getStatusIcon("statusErrorIcon", FilestoreModel.Status.Error_Unknown));
 		itemDeleted.add(FilestoreUI.getStatusIcon("statusDeletedIcon", FilestoreModel.Status.Deleted));

 		itemOk.add(new Label("statusOkCount", String.valueOf(nOk)));
 		itemPending.add(new Label("statusPendingCount", String.valueOf(nPending)));
 		itemInprogress.add(new Label("statusInprogressCount", String.valueOf(nInprogress)));
 		itemError.add(new Label("statusErrorCount", String.valueOf(nError)));
 		itemDeleted.add(new Label("statusDeletedCount", String.valueOf(nDeleted)));

 		add(itemOk);
 		add(itemPending);
 		add(itemInprogress);
 		add(itemError);
 		add(itemDeleted);

 		add(new Label("spaceAvailable", String.format("%s / %s",
 				FileUtils.byteCountToDisplaySize(nBytesUsed),
 				FileUtils.byteCountToDisplaySize(nBytesAvailable))));

 		BookmarkablePageLink<Void> helpLink = new BookmarkablePageLink<Void>("filestoreHelp", FilestoreUsage.class);
 		helpLink.add(new Label("helpMessage", getString("gb.filestoreHelp")));
 		add(helpLink);

 	}

 	protected enum SortBy {
 		ok, pending, inprogress, error, deleted;
 	}

 }
	/*
	* Copyright 2015 gitblit.com.
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/
	package com.gitblit.wicket.pages;

	import java.text.DateFormat;
	import java.text.SimpleDateFormat;
	import java.util.ArrayList;
	import java.util.List;

	import org.apache.commons.io.FileUtils;
	import org.apache.wicket.PageParameters;
	import org.apache.wicket.markup.html.basic.Label;
	import org.apache.wicket.markup.html.link.BookmarkablePageLink;
	import org.apache.wicket.markup.repeater.Item;
	import org.apache.wicket.markup.repeater.data.DataView;
	import org.apache.wicket.markup.repeater.data.ListDataProvider;

	import com.gitblit.Constants;
	import com.gitblit.Keys;
	import com.gitblit.models.FilestoreModel;
	import com.gitblit.models.FilestoreModel.Status;
	import com.gitblit.models.UserModel;
	import com.gitblit.wicket.CacheControl;
	import com.gitblit.wicket.FilestoreUI;
	import com.gitblit.wicket.GitBlitWebSession;
	import com.gitblit.wicket.WicketUtils;
	import com.gitblit.wicket.CacheControl.LastModified;

	/**
	* Page to display the current status of the filestore.
	* Certain errors also displayed to aid in fault finding
	*
	* @author Paul Martin
	*/
	@CacheControl(LastModified.ACTIVITY)
	public class FilestorePage extends RootPage {

	public FilestorePage(PageParameters params) {
	super(params);
	setupPage("", "");

	int itemsPerPage = app().settings().getInteger(Keys.web.itemsPerPage, 20);
	if (itemsPerPage <= 1) {
	itemsPerPage = 20;
	}

	final int pageNumber = WicketUtils.getPage(params);
	final String filter = WicketUtils.getSearchString(params);

	int prevPage = Math.max(0, pageNumber - 1);
	int nextPage = pageNumber + 1;
	boolean hasMore = false;

	final UserModel user = (GitBlitWebSession.get().getUser() == null) ? UserModel.ANONYMOUS : GitBlitWebSession.get().getUser();
	final long nBytesUsed = app().filestore().getFilestoreUsedByteCount();
	final long nBytesAvailable = app().filestore().getFilestoreAvailableByteCount();
	List<FilestoreModel> files = app().filestore().getAllObjects(app().repositories().getRepositoryModels(user));

	if (files == null) {
	files = new ArrayList<FilestoreModel>();
	}

	long nOk = 0;
	long nPending = 0;
	long nInprogress = 0;
	long nError = 0;
	long nDeleted = 0;

	for (FilestoreModel file : files) {
	switch (file.getStatus()) {
	case Available: { nOk++;} break;
	case Upload_Pending: { nPending++; } break;
	case Upload_In_Progress: { nInprogress++; } break;
	case Deleted: { nDeleted++; } break;
	default: { nError++; } break;
	}
	}


	BookmarkablePageLink<Void> itemOk = new BookmarkablePageLink<Void>("filterByOk", FilestorePage.class,
	WicketUtils.newFilestorePageParameter(prevPage, SortBy.ok.name()));

	BookmarkablePageLink<Void> itemPending = new BookmarkablePageLink<Void>("filterByPending", FilestorePage.class,
	WicketUtils.newFilestorePageParameter(prevPage, SortBy.pending.name()));

	BookmarkablePageLink<Void> itemInprogress = new BookmarkablePageLink<Void>("filterByInprogress", FilestorePage.class,
	WicketUtils.newFilestorePageParameter(prevPage, SortBy.inprogress.name()));

	BookmarkablePageLink<Void> itemError = new BookmarkablePageLink<Void>("filterByError", FilestorePage.class,
	WicketUtils.newFilestorePageParameter(prevPage, SortBy.error.name()));

	BookmarkablePageLink<Void> itemDeleted = new BookmarkablePageLink<Void>("filterByDeleted", FilestorePage.class,
	WicketUtils.newFilestorePageParameter(prevPage, SortBy.deleted.name()));


	List<FilestoreModel> filteredResults = new ArrayList<FilestoreModel>(files.size());

	if (filter == null) {
	filteredResults = files;
	} else if (filter.equals(SortBy.ok.name())) {
	WicketUtils.setCssClass(itemOk, "filter-on");

	for (FilestoreModel item : files) {
	if (item.getStatus() == Status.Available) {
	filteredResults.add(item);
	}
	}
	} else if (filter.equals(SortBy.pending.name())) {
	WicketUtils.setCssClass(itemPending, "filter-on");

	for (FilestoreModel item : files) {
	if (item.getStatus() == Status.Upload_Pending) {
	filteredResults.add(item);
	}
	}
	} else if (filter.equals(SortBy.inprogress.name())) {
	WicketUtils.setCssClass(itemInprogress, "filter-on");

	for (FilestoreModel item : files) {
	if (item.getStatus() == Status.Upload_In_Progress) {
	filteredResults.add(item);
	}
	}
	} else if (filter.equals(SortBy.error.name())) {
	WicketUtils.setCssClass(itemError, "filter-on");

	for (FilestoreModel item : files) {
	if (item.isInErrorState()) {
	filteredResults.add(item);
	}
	}
	} else if (filter.equals(SortBy.deleted.name())) {
	WicketUtils.setCssClass(itemDeleted, "filter-on");

	for (FilestoreModel item : files) {
	if (item.getStatus() == Status.Deleted) {
	filteredResults.add(item);
	}
	}
	}

	DataView<FilestoreModel> filesView = new DataView<FilestoreModel>("fileRow",
	new ListDataProvider<FilestoreModel>(filteredResults) , itemsPerPage) {
	private static final long serialVersionUID = 1L;
	private int counter;

	@Override
	protected void onBeforeRender() {
	super.onBeforeRender();
	counter = 0;
	}

	@Override
	public void populateItem(final Item<FilestoreModel> item) {
	final FilestoreModel entry = item.getModelObject();

	DateFormat dateFormater = new SimpleDateFormat(Constants.ISO8601);

	UserModel user = app().users().getUserModel(entry.getChangedBy());
	user = user == null ? UserModel.ANONYMOUS : user;

	Label icon = FilestoreUI.getStatusIcon("status", entry);
	item.add(icon);
	item.add(new Label("on", dateFormater.format(entry.getChangedOn())));
	item.add(new Label("by", user.getDisplayName()));

	item.add(new Label("oid", entry.oid));
	item.add(new Label("size", FileUtils.byteCountToDisplaySize(entry.getSize())));

	WicketUtils.setAlternatingBackground(item, counter);
	counter++;
	}

	};


	if (filteredResults.size() < itemsPerPage) {
	filesView.setCurrentPage(0);
	hasMore = false;
	} else {
	filesView.setCurrentPage(pageNumber - 1);
	hasMore = true;
	}


	add(filesView);


	add(new BookmarkablePageLink<Void>("firstPageBottom", FilestorePage.class).setEnabled(pageNumber > 1));
	add(new BookmarkablePageLink<Void>("prevPageBottom", FilestorePage.class,
	WicketUtils.newFilestorePageParameter(prevPage, filter)).setEnabled(pageNumber > 1));
	add(new BookmarkablePageLink<Void>("nextPageBottom", FilestorePage.class,
	WicketUtils.newFilestorePageParameter(nextPage, filter)).setEnabled(hasMore));


	itemOk.add(FilestoreUI.getStatusIcon("statusOkIcon", FilestoreModel.Status.Available));
	itemPending.add(FilestoreUI.getStatusIcon("statusPendingIcon", FilestoreModel.Status.Upload_Pending));
	itemInprogress.add(FilestoreUI.getStatusIcon("statusInprogressIcon", FilestoreModel.Status.Upload_In_Progress));
	itemError.add(FilestoreUI.getStatusIcon("statusErrorIcon", FilestoreModel.Status.Error_Unknown));
	itemDeleted.add(FilestoreUI.getStatusIcon("statusDeletedIcon", FilestoreModel.Status.Deleted));

	itemOk.add(new Label("statusOkCount", String.valueOf(nOk)));
	itemPending.add(new Label("statusPendingCount", String.valueOf(nPending)));
	itemInprogress.add(new Label("statusInprogressCount", String.valueOf(nInprogress)));
	itemError.add(new Label("statusErrorCount", String.valueOf(nError)));
	itemDeleted.add(new Label("statusDeletedCount", String.valueOf(nDeleted)));

	add(itemOk);
	add(itemPending);
	add(itemInprogress);
	add(itemError);
	add(itemDeleted);

	add(new Label("spaceAvailable", String.format("%s / %s",
	FileUtils.byteCountToDisplaySize(nBytesUsed),
	FileUtils.byteCountToDisplaySize(nBytesAvailable))));

	BookmarkablePageLink<Void> helpLink = new BookmarkablePageLink<Void>("filestoreHelp", FilestoreUsage.class);
	helpLink.add(new Label("helpMessage", getString("gb.filestoreHelp")));
	add(helpLink);

	}

	protected enum SortBy {
	ok, pending, inprogress, error, deleted;
	}

	}