vrr/ProxyAI
2
0
Fork 0
mirror of https://github.com/carlrobertoh/ProxyAI.git synced 2026-04-29 12:11:26 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Chat/Conversations toolbar actions (#40)

Browse source
This commit is contained in:
Carl-Robert Linnupuu 2023-03-17 10:15:17 +00:00
parent 3317fb8603
commit 4e80043c3f
15 changed files with 361 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

63
src/main/java/ee/carlrobert/codegpt/ide/conversations/ConversationsState.java
View file

@ -14,7 +14,11 @@ import ee.carlrobert.codegpt.ide.conversations.converter.ConversationsConverter;
import ee.carlrobert.codegpt.ide.settings.SettingsState;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.Comparator;
import java.util.List;
import java.util.Optional;
import java.util.UUID;
import java.util.stream.Collectors;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
@ -53,6 +57,16 @@ public class ConversationsState implements PersistentStateComponent<Conversation
return getInstance().currentConversation;
}
public List<Conversation> getSortedConversations() {
return conversationsContainer
.getConversationsMapping()
.values()
.stream()
.flatMap(List::stream)
.sorted(Comparator.comparing(Conversation::getUpdatedOn).reversed())
.collect(Collectors.toList());
}
public Conversation createConversation(ClientCode clientCode) {
var settings = SettingsState.getInstance();
var conversation = new Conversation();
@ -101,4 +115,53 @@ public class ConversationsState implements PersistentStateComponent<Conversation
addConversation(conversation);
return conversation;
}
public void clearAll() {
conversationsContainer.getConversationsMapping().clear();
currentConversation = null;
}
public Optional<Conversation> getNextConversation() {
return tryGetNextOrPreviousConversation(true);
}
public Optional<Conversation> getPreviousConversation() {
return tryGetNextOrPreviousConversation(false);
}
private Optional<Conversation> tryGetNextOrPreviousConversation(boolean isNext) {
if (currentConversation != null) {
var sortedConversations = getSortedConversations();
for (int i = 0; i < sortedConversations.size(); i++) {
var conversation = sortedConversations.get(i);
if (conversation != null && conversation.getId().equals(currentConversation.getId())) {
var nextIndex = isNext ? i + 1 : i - 1;
if (isNext ? nextIndex < sortedConversations.size() : nextIndex != -1) {
return Optional.of(sortedConversations.get(nextIndex));
}
}
}
}
return Optional.empty();
}
public void deleteSelectedConversation() {
var nextConversation = getNextConversation();
if (nextConversation.isEmpty()) {
nextConversation = getPreviousConversation();
}
var iterator = conversationsContainer.getConversationsMapping()
.get(currentConversation.getClientCode())
.listIterator();
while (iterator.hasNext()) {
var next = iterator.next();
if (next.getId().equals(currentConversation.getId())) {
iterator.remove();
break;
}
}
nextConversation.ifPresent(this::setCurrentConversation);
}
}