mirror of https://github.com/kurisufriend/Clover
Floens
10 years ago
parent
797832c9e0
commit
05275eb7bf
@ -0,0 +1,144 @@ |
||||
/* |
||||
* Clover - 4chan browser https://github.com/Floens/Clover/
|
||||
* Copyright (C) 2014 Floens |
||||
* |
||||
* This program 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 3 of the License, or |
||||
* (at your option) any later version. |
||||
* |
||||
* This program 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 this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/ |
||||
package org.floens.chan.ui.adapter; |
||||
|
||||
import android.text.TextUtils; |
||||
|
||||
import org.floens.chan.core.model.Post; |
||||
|
||||
import java.util.ArrayList; |
||||
import java.util.Collections; |
||||
import java.util.Comparator; |
||||
import java.util.Iterator; |
||||
import java.util.List; |
||||
import java.util.Locale; |
||||
|
||||
public class PostFilter { |
||||
public static final Comparator<Post> IMAGE_COMPARATOR = new Comparator<Post>() { |
||||
@Override |
||||
public int compare(Post lhs, Post rhs) { |
||||
return rhs.images - lhs.images; |
||||
} |
||||
}; |
||||
|
||||
public static final Comparator<Post> REPLY_COMPARATOR = new Comparator<Post>() { |
||||
@Override |
||||
public int compare(Post lhs, Post rhs) { |
||||
return rhs.replies - lhs.replies; |
||||
} |
||||
}; |
||||
|
||||
public static final Comparator<Post> NEWEST_COMPARATOR = new Comparator<Post>() { |
||||
@Override |
||||
public int compare(Post lhs, Post rhs) { |
||||
return (int) (rhs.time - lhs.time); |
||||
} |
||||
}; |
||||
|
||||
public static final Comparator<Post> OLDEST_COMPARATOR = new Comparator<Post>() { |
||||
@Override |
||||
public int compare(Post lhs, Post rhs) { |
||||
return (int) (lhs.time - rhs.time); |
||||
} |
||||
}; |
||||
|
||||
private Order order; |
||||
private String query; |
||||
|
||||
public PostFilter(Order order, String query) { |
||||
this.order = order; |
||||
this.query = query; |
||||
} |
||||
|
||||
/** |
||||
* Creates a copy of {@code original} and applies any sorting or filtering to it. |
||||
* |
||||
* @param original List of {@link Post}s to filter. |
||||
* @return a new filtered List |
||||
*/ |
||||
public List<Post> apply(List<Post> original) { |
||||
List<Post> posts = new ArrayList<>(original); |
||||
|
||||
// Process order
|
||||
if (order != PostFilter.Order.BUMP) { |
||||
switch (order) { |
||||
case IMAGE: |
||||
Collections.sort(posts, IMAGE_COMPARATOR); |
||||
break; |
||||
case REPLY: |
||||
Collections.sort(posts, REPLY_COMPARATOR); |
||||
break; |
||||
case NEWEST: |
||||
Collections.sort(posts, NEWEST_COMPARATOR); |
||||
break; |
||||
case OLDEST: |
||||
Collections.sort(posts, OLDEST_COMPARATOR); |
||||
break; |
||||
} |
||||
} |
||||
|
||||
// Process search
|
||||
if (!TextUtils.isEmpty(query)) { |
||||
String lowerQuery = query.toLowerCase(Locale.ENGLISH); |
||||
|
||||
boolean add; |
||||
Iterator<Post> i = posts.iterator(); |
||||
while (i.hasNext()) { |
||||
Post item = i.next(); |
||||
add = false; |
||||
if (item.comment.toString().toLowerCase(Locale.ENGLISH).contains(lowerQuery)) { |
||||
add = true; |
||||
} else if (item.subject.toLowerCase(Locale.ENGLISH).contains(lowerQuery)) { |
||||
add = true; |
||||
} else if (item.name.toLowerCase(Locale.ENGLISH).contains(lowerQuery)) { |
||||
add = true; |
||||
} else if (item.filename != null && item.filename.toLowerCase(Locale.ENGLISH).contains(lowerQuery)) { |
||||
add = true; |
||||
} |
||||
if (!add) { |
||||
i.remove(); |
||||
} |
||||
} |
||||
} |
||||
|
||||
return posts; |
||||
} |
||||
|
||||
public enum Order { |
||||
BUMP("bump"), |
||||
REPLY("reply"), |
||||
IMAGE("image"), |
||||
NEWEST("newest"), |
||||
OLDEST("oldest"); |
||||
|
||||
public String name; |
||||
|
||||
Order(String storeName) { |
||||
this.name = storeName; |
||||
} |
||||
|
||||
public static Order find(String name) { |
||||
for (Order mode : Order.values()) { |
||||
if (mode.name.equals(name)) { |
||||
return mode; |
||||
} |
||||
} |
||||
return null; |
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue