package com.jetbrains.edu.learning.ui;
import com.intellij.facet.ui.FacetValidatorsManager;
import com.intellij.facet.ui.ValidationResult;
import com.intellij.icons.AllIcons;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.fileChooser.FileChooser;
import com.intellij.openapi.fileChooser.FileChooserDescriptor;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.project.DefaultProjectFactory;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.project.ProjectManager;
import com.intellij.openapi.ui.ComboBox;
import com.intellij.openapi.ui.FixedSizeButton;
import com.intellij.openapi.ui.LabeledComponent;
import com.intellij.openapi.ui.VerticalFlowLayout;
import com.intellij.openapi.ui.popup.JBPopupFactory;
import com.intellij.openapi.ui.popup.ListPopup;
import com.intellij.openapi.ui.popup.PopupStep;
import com.intellij.openapi.ui.popup.util.BaseListPopupStep;
import com.intellij.openapi.util.SystemInfo;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.ui.AncestorListenerAdapter;
import com.intellij.ui.PanelWithAnchor;
import com.intellij.ui.components.JBScrollPane;
import com.intellij.util.ui.UIUtil;
import com.jetbrains.edu.learning.StudySettings;
import com.jetbrains.edu.learning.StudyUtils;
import com.jetbrains.edu.learning.courseFormat.Course;
import com.jetbrains.edu.learning.courseGeneration.StudyProjectGenerator;
import com.jetbrains.edu.learning.stepic.EduAdaptiveStepicConnector;
import com.jetbrains.edu.learning.stepic.EduStepicConnector;
import com.jetbrains.edu.learning.stepic.OAuthDialog;
import com.jetbrains.edu.learning.stepic.StepicUser;
import icons.EducationalCoreIcons;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;
import javax.swing.*;
import javax.swing.event.AncestorEvent;
import java.awt.*;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.ArrayList;
import java.util.List;
/**
* author: liana
* data: 7/31/14.
*/
public class StudyNewProjectPanel extends JPanel implements PanelWithAnchor {
private List<Course> myAvailableCourses = new ArrayList<>();
private JButton myBrowseButton;
private ComboBox<Course> myCoursesComboBox;
private JButton myRefreshButton;
private JLabel myAuthorLabel;
private JPanel myInfoPanel;
private JTextPane myDescriptionPane;
private JComponent myAnchor;
private final StudyProjectGenerator myGenerator;
private boolean isLocal = false;
private static final String CONNECTION_ERROR = "<html>Failed to download courses.<br>Check your Internet connection.</html>";
private static final String INVALID_COURSE = "Selected course is invalid";
private FacetValidatorsManager myValidationManager;
private boolean isComboboxInitialized;
private static final String LOGIN_TO_STEPIC_MESSAGE = "<html><u>Login to Stepik</u> to open the adaptive course </html>";
private static final String LOGIN_TO_STEPIC = "Login to Stepik";
public StudyNewProjectPanel(@NotNull final StudyProjectGenerator generator, boolean isLocal) {
super(new VerticalFlowLayout(true, true));
myGenerator = generator;
this.isLocal = isLocal;
layoutPanel();
initListeners();
}
private void layoutPanel() {
myCoursesComboBox = new ComboBox<>();
final LabeledComponent<ComboBox> coursesCombo = LabeledComponent.create(myCoursesComboBox, "Courses:", BorderLayout.WEST);
myRefreshButton = new FixedSizeButton(coursesCombo);
if (SystemInfo.isMac && !UIUtil.isUnderDarcula())
myRefreshButton.putClientProperty("JButton.buttonType", null);
myRefreshButton.setIcon(AllIcons.Actions.Refresh);
myBrowseButton = new FixedSizeButton(coursesCombo);
final JPanel comboPanel = new JPanel(new BorderLayout());
comboPanel.add(coursesCombo, BorderLayout.CENTER);
comboPanel.add(myRefreshButton, BorderLayout.EAST);
final JPanel coursesPanel = new JPanel(new BorderLayout());
coursesPanel.add(comboPanel, BorderLayout.CENTER);
coursesPanel.add(myBrowseButton, BorderLayout.EAST);
add(coursesPanel);
myAnchor = coursesCombo;
final JPanel panel = new JPanel(new BorderLayout());
final JLabel invisibleLabel = new JLabel();
invisibleLabel.setPreferredSize(new JLabel("Location: ").getPreferredSize());
panel.add(invisibleLabel, BorderLayout.WEST);
myInfoPanel = new JPanel(new VerticalFlowLayout());
myAuthorLabel = new JLabel();
myDescriptionPane = new JTextPane();
myDescriptionPane.setEditable(true);
myDescriptionPane.setEnabled(true);
myAuthorLabel.setEnabled(true);
myDescriptionPane.setPreferredSize(new Dimension(150, 200));
myDescriptionPane.setFont(coursesCombo.getFont());
myInfoPanel.add(myAuthorLabel);
myInfoPanel.add(new JBScrollPane(myDescriptionPane));
panel.add(myInfoPanel, BorderLayout.CENTER);
add(panel);
}
private void initCoursesCombobox() {
myAvailableCourses =
myGenerator.getCoursesUnderProgress(!isLocal, "Getting Available Courses", ProjectManager.getInstance().getDefaultProject());
if (myAvailableCourses.contains(Course.INVALID_COURSE)) {
setError(CONNECTION_ERROR);
}
else {
addCoursesToCombobox(myAvailableCourses);
final Course selectedCourse = StudyUtils.getFirst(myAvailableCourses);
if (selectedCourse == null) return;
setAuthors(selectedCourse);
myDescriptionPane.setText(selectedCourse.getDescription());
myDescriptionPane.setEditable(false);
//setting the first course in list as selected
myGenerator.setSelectedCourse(selectedCourse);
if (myGenerator.getSelectedCourse() != null) {
myCoursesComboBox.setSelectedItem(myGenerator.getSelectedCourse());
}
if (selectedCourse.isAdaptive() && !myGenerator.isLoggedIn()) {
setError(LOGIN_TO_STEPIC_MESSAGE);
}
else {
setOK();
}
}
}
private void setupBrowseButton() {
if (SystemInfo.isMac && !UIUtil.isUnderDarcula())
myBrowseButton.putClientProperty("JButton.buttonType", null);
myBrowseButton.setIcon(EducationalCoreIcons.InterpreterGear);
final FileChooserDescriptor fileChooser = new FileChooserDescriptor(true, false, false, true, false, false) {
@Override
public boolean isFileVisible(VirtualFile file, boolean showHiddenFiles) {
return file.isDirectory() || StudyUtils.isZip(file.getName());
}
@Override
public boolean isFileSelectable(VirtualFile file) {
return StudyUtils.isZip(file.getName());
}
};
myBrowseButton.addActionListener(new ActionListener() {
public void actionPerformed(ActionEvent e) {
final BaseListPopupStep<String> popupStep = new BaseListPopupStep<String>("", "Add local course", LOGIN_TO_STEPIC) {
@Override
public PopupStep onChosen(final String selectedValue, boolean finalChoice) {
return doFinalStep(() -> {
if ("Add local course".equals(selectedValue)) {
Project[] projects = ProjectManager.getInstance().getOpenProjects();
FileChooser.chooseFile(fileChooser, null, projects.length == 0 ? null : projects[0].getBaseDir(),
file -> {
String fileName = file.getPath();
int oldSize = myAvailableCourses.size();
Course courseInfo = myGenerator.addLocalCourse(fileName);
if (courseInfo != null) {
if (oldSize != myAvailableCourses.size()) {
myCoursesComboBox.addItem(courseInfo);
}
myCoursesComboBox.setSelectedItem(courseInfo);
setOK();
}
else {
setError(INVALID_COURSE);
myCoursesComboBox.removeAllItems();
myCoursesComboBox.addItem(Course.INVALID_COURSE);
for (Course course : myAvailableCourses) {
myCoursesComboBox.addItem(course);
}
myCoursesComboBox.setSelectedItem(Course.INVALID_COURSE);
}
});
}
else if (LOGIN_TO_STEPIC.equals(selectedValue)) {
EduStepicConnector.doAuthorize(() -> showLoginDialog());
StepicUser stepicUser = StudySettings.getInstance().getUser();
if (stepicUser != null) {
ProgressManager.getInstance().runProcessWithProgressSynchronously(
() -> myGenerator.setEnrolledCoursesIds(EduAdaptiveStepicConnector.getEnrolledCoursesIds(stepicUser)),
"Getting Enrolled Courses", true, DefaultProjectFactory.getInstance().getDefaultProject());
final List<Course> courses = myGenerator.getCourses(true);
if (courses != null) {
ApplicationManager.getApplication().invokeLater(() -> refreshCoursesList(courses));
}
}
}
});
}
};
final ListPopup popup = JBPopupFactory.getInstance().createListPopup(popupStep);
popup.showInScreenCoordinates(myBrowseButton, myBrowseButton.getLocationOnScreen());
}
});
}
public void showLoginDialog() {
OAuthDialog dialog = new OAuthDialog();
if (dialog.showAndGet()) {
StepicUser stepicUser = dialog.getStepicUser();
StudySettings.getInstance().setUser(stepicUser);
setOK();
}
}
private void initListeners() {
myRefreshButton.addActionListener(new RefreshActionListener());
myCoursesComboBox.addActionListener(new CourseSelectedListener());
setupBrowseButton();
addAncestorListener(new AncestorListenerAdapter() {
@Override
public void ancestorMoved(AncestorEvent event) {
if (!isComboboxInitialized && isVisible()) {
isComboboxInitialized = true;
initCoursesCombobox();
}
Course selectedCourse = (Course)myCoursesComboBox.getSelectedItem();
if (selectedCourse == null || selectedCourse.equals(Course.INVALID_COURSE)) {
setError(CONNECTION_ERROR);
}
}
});
}
private void setError(@NotNull final String errorMessage) {
myGenerator.fireStateChanged(new ValidationResult(errorMessage));
if (myValidationManager != null) {
myValidationManager.validate();
}
}
public void setOK() {
myGenerator.fireStateChanged(ValidationResult.OK);
if (myValidationManager != null) {
myValidationManager.validate();
}
}
public void registerValidators(final FacetValidatorsManager manager) {
myValidationManager = manager;
}
@Override
public JComponent getAnchor() {
return myAnchor;
}
@Override
public void setAnchor(@Nullable JComponent anchor) {
myAnchor = anchor;
}
public void updateInfoPanel(@NotNull Course selectedCourse) {
setAuthors(selectedCourse);
myDescriptionPane.setText(selectedCourse.getDescription());
}
/**
* Handles refreshing courses
* Old courses added to new courses only if their
* meta file still exists in local file system
*/
private class RefreshActionListener implements ActionListener {
@Override
public void actionPerformed(ActionEvent e) {
final List<Course> courses =
myGenerator.getCoursesUnderProgress(true, "Refreshing Course List", DefaultProjectFactory.getInstance().getDefaultProject());
if (!courses.contains(Course.INVALID_COURSE)) {
refreshCoursesList(courses);
}
}
}
private void refreshCoursesList(@NotNull final List<Course> courses) {
if (courses.isEmpty()) {
setError(CONNECTION_ERROR);
return;
}
myCoursesComboBox.removeAllItems();
addCoursesToCombobox(courses);
final Course selectedCourse = StudyUtils.getFirst(courses);
if (selectedCourse == null) return;
myGenerator.setSelectedCourse(selectedCourse);
myGenerator.setCourses(courses);
myAvailableCourses = courses;
}
private void addCoursesToCombobox(@NotNull List<Course> courses) {
for (Course courseInfo : courses) {
myCoursesComboBox.addItem(courseInfo);
}
}
/**
* Handles selecting course in combo box
* Sets selected course in combo box as selected in
* {@link StudyNewProjectPanel#myGenerator}
*/
private class CourseSelectedListener implements ActionListener {
@Override
public void actionPerformed(ActionEvent e) {
JComboBox cb = (JComboBox)e.getSource();
Course selectedCourse = (Course)cb.getSelectedItem();
if (selectedCourse == null || selectedCourse.equals(Course.INVALID_COURSE)) {
myAuthorLabel.setText("");
myDescriptionPane.setText("");
setError(INVALID_COURSE);
return;
}
updateInfoPanel(selectedCourse);
myCoursesComboBox.removeItem(Course.INVALID_COURSE);
myGenerator.setSelectedCourse(selectedCourse);
setOK();
if (selectedCourse.isAdaptive()) {
if(!myGenerator.isLoggedIn()) {
setError(LOGIN_TO_STEPIC_MESSAGE);
}
}
}
}
private void setAuthors(Course selectedCourse) {
final String authorsString = Course.getAuthorsString(selectedCourse.getAuthors());
myAuthorLabel.setText(!StringUtil.isEmptyOrSpaces(authorsString) ? "Author: " + authorsString : "");
}
public JPanel getInfoPanel() {
return myInfoPanel;
}
}