b6e1e806ce41637deb27a2715df7f6a60090c02c
[idea/community.git] / python / educational-core / course-creator / src / com / jetbrains / edu / coursecreator / actions / CCShowPreview.java
1 /*
2  * Copyright 2000-2014 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.jetbrains.edu.coursecreator.actions;
17
18 import com.intellij.openapi.Disposable;
19 import com.intellij.openapi.actionSystem.AnActionEvent;
20 import com.intellij.openapi.actionSystem.CommonDataKeys;
21 import com.intellij.openapi.actionSystem.LangDataKeys;
22 import com.intellij.openapi.actionSystem.Presentation;
23 import com.intellij.openapi.application.ApplicationManager;
24 import com.intellij.openapi.diff.impl.util.LabeledEditor;
25 import com.intellij.openapi.editor.Document;
26 import com.intellij.openapi.editor.EditorFactory;
27 import com.intellij.openapi.editor.ex.EditorEx;
28 import com.intellij.openapi.fileEditor.FileDocumentManager;
29 import com.intellij.openapi.module.Module;
30 import com.intellij.openapi.project.DumbAwareAction;
31 import com.intellij.openapi.project.Project;
32 import com.intellij.openapi.ui.FrameWrapper;
33 import com.intellij.openapi.ui.Messages;
34 import com.intellij.openapi.util.Disposer;
35 import com.intellij.openapi.util.Pair;
36 import com.intellij.openapi.vfs.VirtualFile;
37 import com.intellij.psi.PsiDirectory;
38 import com.intellij.psi.PsiFile;
39 import com.intellij.ui.JBColor;
40 import com.jetbrains.edu.coursecreator.CCUtils;
41 import com.jetbrains.edu.learning.StudyTaskManager;
42 import com.jetbrains.edu.learning.StudyUtils;
43 import com.jetbrains.edu.learning.core.EduAnswerPlaceholderPainter;
44 import com.jetbrains.edu.learning.core.EduUtils;
45 import com.jetbrains.edu.learning.courseFormat.AnswerPlaceholder;
46 import com.jetbrains.edu.learning.courseFormat.Course;
47 import com.jetbrains.edu.learning.courseFormat.TaskFile;
48 import org.jetbrains.annotations.NotNull;
49
50 import javax.swing.*;
51 import javax.swing.border.EmptyBorder;
52 import java.awt.*;
53 import java.text.SimpleDateFormat;
54 import java.util.Calendar;
55 import java.util.Collections;
56
57 public class CCShowPreview extends DumbAwareAction {
58   public static final String SHOW_PREVIEW = "Show Preview";
59   public static final String NO_PREVIEW_MESSAGE = "Preview is available for task files with answer placeholders only";
60
61   public CCShowPreview() {
62     super(SHOW_PREVIEW, SHOW_PREVIEW, null);
63   }
64
65   @Override
66   public void update(@NotNull AnActionEvent e) {
67     Presentation presentation = e.getPresentation();
68     presentation.setEnabledAndVisible(false);
69     Project project = e.getProject();
70     if (project == null) {
71       return;
72     }
73     if (!CCUtils.isCourseCreator(project)) {
74       return;
75     }
76     final PsiFile file = CommonDataKeys.PSI_FILE.getData(e.getDataContext());
77     if (file != null && StudyUtils.getTaskFile(project, file.getVirtualFile()) != null) {
78       presentation.setEnabledAndVisible(true);
79     }
80   }
81
82   @Override
83   public void actionPerformed(@NotNull AnActionEvent e) {
84     final Project project = e.getProject();
85     Module module = LangDataKeys.MODULE.getData(e.getDataContext());
86     if (project == null || module == null) {
87       return;
88     }
89     final PsiFile file = CommonDataKeys.PSI_FILE.getData(e.getDataContext());
90     if (file == null) {
91       return;
92     }
93     Course course = StudyTaskManager.getInstance(project).getCourse();
94     if (course == null) {
95       return;
96     }
97     VirtualFile virtualFile = file.getVirtualFile();
98     TaskFile taskFile = StudyUtils.getTaskFile(project, virtualFile);
99     if (taskFile == null) {
100       return;
101     }
102     final PsiDirectory taskDir = file.getContainingDirectory();
103     if (taskDir == null) {
104       return;
105     }
106     PsiDirectory lessonDir = taskDir.getParentDirectory();
107     if (lessonDir == null) {
108       return;
109     }
110
111
112     if (taskFile.getActivePlaceholders().isEmpty()) {
113       Messages.showInfoMessage("Preview is available for task files with answer placeholders only", "No Preview for This File");
114       return;
115     }
116
117     VirtualFile generatedFilesFolder = CCUtils.getGeneratedFilesFolder(project, module);
118
119     if (generatedFilesFolder == null) {
120       return;
121     }
122
123     ApplicationManager.getApplication().runWriteAction(new Runnable() {
124       @Override
125       public void run() {
126         Pair<VirtualFile, TaskFile> pair =
127           EduUtils.createStudentFile(this, project, virtualFile, generatedFilesFolder, null, taskFile.getTask().getActiveSubtaskIndex());
128         if (pair != null) {
129           showPreviewDialog(project, pair.getFirst(), pair.getSecond());
130         }
131       }
132     });
133   }
134
135   private static void showPreviewDialog(@NotNull Project project, @NotNull VirtualFile userFile, @NotNull TaskFile taskFile) {
136     final FrameWrapper showPreviewFrame = new FrameWrapper(project);
137     showPreviewFrame.setTitle(userFile.getName());
138     LabeledEditor labeledEditor = new LabeledEditor(null);
139     final EditorFactory factory = EditorFactory.getInstance();
140     Document document = FileDocumentManager.getInstance().getDocument(userFile);
141     if (document == null) {
142       return;
143     }
144     final EditorEx createdEditor = (EditorEx)factory.createEditor(document, project, userFile, true);
145     Disposer.register(project, new Disposable() {
146       public void dispose() {
147         factory.releaseEditor(createdEditor);
148       }
149     });
150     for (AnswerPlaceholder answerPlaceholder : taskFile.getActivePlaceholders()) {
151       Integer minIndex = Collections.min(answerPlaceholder.getSubtaskInfos().keySet());
152       answerPlaceholder.setUseLength(minIndex >= answerPlaceholder.getActiveSubtaskIndex());
153       EduAnswerPlaceholderPainter.drawAnswerPlaceholder(createdEditor, answerPlaceholder, JBColor.BLUE);
154     }
155     JPanel header = new JPanel();
156     header.setLayout(new BoxLayout(header, BoxLayout.Y_AXIS));
157     header.setBorder(new EmptyBorder(10, 10, 10, 10));
158     header.add(new JLabel("Read-only preview."));
159     String timeStamp = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss").format(Calendar.getInstance().getTime());
160     header.add(new JLabel(String.format("Created %s.", timeStamp)));
161     JComponent editorComponent = createdEditor.getComponent();
162     labeledEditor.setComponent(editorComponent, header);
163     createdEditor.setCaretVisible(false);
164     createdEditor.setCaretEnabled(false);
165     showPreviewFrame.setComponent(labeledEditor);
166     showPreviewFrame.setSize(new Dimension(500, 500));
167     if (!ApplicationManager.getApplication().isUnitTestMode()) {
168       showPreviewFrame.show();
169     }
170   }
171 }