1package eu.siacs.conversations.ui.adapter;
 2
 3import android.content.Context;
 4import android.widget.ArrayAdapter;
 5import android.widget.Filter;
 6
 7import androidx.annotation.NonNull;
 8
 9import com.google.common.collect.ImmutableList;
10import com.google.common.collect.Ordering;
11
12import eu.siacs.conversations.Config;
13
14import java.util.ArrayList;
15import java.util.Collection;
16import java.util.List;
17import java.util.Locale;
18import java.util.regex.Pattern;
19
20public class KnownHostsAdapter extends ArrayAdapter<String> {
21
22    private static final Pattern E164_PATTERN = Pattern.compile("^\\+[1-9]\\d{1,14}$");
23
24    private List<String> domains;
25    private final Filter domainFilter = new Filter() {
26
27        @Override
28        protected FilterResults performFiltering(final CharSequence constraint) {
29            final ImmutableList.Builder<String> builder = new ImmutableList.Builder<>();
30            final String[] split = constraint == null ? new String[0] : constraint.toString().split("@");
31            if (split.length == 1) {
32                final String local = split[0].toLowerCase(Locale.ENGLISH);
33                if (Config.QUICKSY_DOMAIN != null && E164_PATTERN.matcher(local).matches()) {
34                    builder.add(local + '@' + Config.QUICKSY_DOMAIN.toEscapedString());
35                } else {
36                    for (String domain : domains) {
37                        builder.add(local + '@' + domain);
38                    }
39                }
40            } else if (split.length == 2) {
41                final String localPart = split[0].toLowerCase(Locale.ENGLISH);
42                final String domainPart = split[1].toLowerCase(Locale.ENGLISH);
43                if (domains.contains(domainPart)) {
44                    return new FilterResults();
45                }
46                for (final String domain : domains) {
47                    if (domain.contains(domainPart)) {
48                        builder.add(localPart + "@" + domain);
49                    }
50                }
51            } else {
52                return new FilterResults();
53            }
54            final var suggestions = builder.build();
55            final FilterResults filterResults = new FilterResults();
56            filterResults.values = suggestions;
57            filterResults.count = suggestions.size();
58            return filterResults;
59        }
60
61        @Override
62        protected void publishResults(final CharSequence constraint, final FilterResults results) {
63            final ImmutableList.Builder<String> suggestions = new ImmutableList.Builder<>();
64            if (results.values instanceof Collection<?> collection) {
65                for(final Object item : collection) {
66                    if (item instanceof String string) {
67                        suggestions.add(string);
68                    }
69                }
70            }
71            clear();
72            addAll(suggestions.build());
73            notifyDataSetChanged();
74        }
75    };
76
77    public KnownHostsAdapter(final Context context, final int viewResourceId, final Collection<String> knownHosts) {
78        super(context, viewResourceId, new ArrayList<>());
79        domains =  Ordering.natural().sortedCopy(knownHosts);
80    }
81
82    public KnownHostsAdapter(final Context context, final int viewResourceId) {
83        super(context, viewResourceId, new ArrayList<>());
84        domains = ImmutableList.of();
85    }
86
87    public void refresh(final Collection<String> knownHosts) {
88        this.domains = Ordering.natural().sortedCopy(knownHosts);
89        notifyDataSetChanged();
90    }
91
92    @Override
93    @NonNull
94    public Filter getFilter() {
95        return domainFilter;
96    }
97}