ef6b7a6c960c81b12b04e613eb3a28cbeb056dd5
[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
56 public class CCShowPreview extends DumbAwareAction {
57   public static final String SHOW_PREVIEW = "Show Preview";
58   public static final String NO_PREVIEW_MESSAGE = "Preview is available for task files with answer placeholders only";
59
60   public CCShowPreview() {
61     super(SHOW_PREVIEW, SHOW_PREVIEW, null);
62   }
63
64   @Override
65   public void update(@NotNull AnActionEvent e) {
66     Presentation presentation = e.getPresentation();
67     presentation.setEnabledAndVisible(false);
68     Project project = e.getProject();
69     if (project == null) {
70       return;
71     }
72     if (!CCUtils.isCourseCreator(project)) {
73       return;
74     }
75     final PsiFile file = CommonDataKeys.PSI_FILE.getData(e.getDataContext());
76     if (file != null && StudyUtils.getTaskFile(project, file.getVirtualFile()) != null) {
77       presentation.setEnabledAndVisible(true);
78     }
79   }
80
81   @Override
82   public void actionPerformed(@NotNull AnActionEvent e) {
83     final Project project = e.getProject();
84     Module module = LangDataKeys.MODULE.getData(e.getDataContext());
85     if (project == null || module == null) {
86       return;
87     }
88     final PsiFile file = CommonDataKeys.PSI_FILE.getData(e.getDataContext());
89     if (file == null) {
90       return;
91     }
92     Course course = StudyTaskManager.getInstance(project).getCourse();
93     if (course == null) {
94       return;
95     }
96     VirtualFile virtualFile = file.getVirtualFile();
97     TaskFile taskFile = StudyUtils.getTaskFile(project, virtualFile);
98     if (taskFile == null) {
99       return;
100     }
101     final PsiDirectory taskDir = file.getContainingDirectory();
102     if (taskDir == null) {
103       return;
104     }
105     PsiDirectory lessonDir = taskDir.getParentDirectory();
106     if (lessonDir == null) {
107       return;
108     }
109
110
111     if (taskFile.getActivePlaceholders().isEmpty()) {
112       Messages.showInfoMessage("Preview is available for task files with answer placeholders only", "No Preview for This File");
113       return;
114     }
115
116     VirtualFile generatedFilesFolder = CCUtils.getGeneratedFilesFolder(project, module);
117
118     if (generatedFilesFolder == null) {
119       return;
120     }
121
122     ApplicationManager.getApplication().runWriteAction(new Runnable() {
123       @Override
124       public void run() {
125         Pair<VirtualFile, TaskFile> pair =
126           EduUtils.createStudentFile(this, project, virtualFile, generatedFilesFolder, null);
127         if (pair != null) {
128           showPreviewDialog(project, pair.getFirst(), pair.getSecond());
129         }
130       }
131     });
132   }
133
134   private static void showPreviewDialog(@NotNull Project project, @NotNull VirtualFile userFile, @NotNull TaskFile taskFile) {
135     final FrameWrapper showPreviewFrame = new FrameWrapper(project);
136     showPreviewFrame.setTitle(userFile.getName());
137     LabeledEditor labeledEditor = new LabeledEditor(null);
138     final EditorFactory factory = EditorFactory.getInstance();
139     Document document = FileDocumentManager.getInstance().getDocument(userFile);
140     if (document == null) {
141       return;
142     }
143     final EditorEx createdEditor = (EditorEx)factory.createEditor(document, project, userFile, true);
144     Disposer.register(project, new Disposable() {
145       public void dispose() {
146         factory.releaseEditor(createdEditor);
147       }
148     });
149     for (AnswerPlaceholder answerPlaceholder : taskFile.getActivePlaceholders()) {
150       answerPlaceholder.setUseLength(true);
151       EduAnswerPlaceholderPainter.drawAnswerPlaceholder(createdEditor, answerPlaceholder, JBColor.BLUE);
152     }
153     JPanel header = new JPanel();
154     header.setLayout(new BoxLayout(header, BoxLayout.Y_AXIS));
155     header.setBorder(new EmptyBorder(10, 10, 10, 10));
156     header.add(new JLabel("Read-only preview."));
157     String timeStamp = new SimpleDateFormat("yyyy/MM/dd HH:mm:ss").format(Calendar.getInstance().getTime());
158     header.add(new JLabel(String.format("Created %s.", timeStamp)));
159     JComponent editorComponent = createdEditor.getComponent();
160     labeledEditor.setComponent(editorComponent, header);
161     createdEditor.setCaretVisible(false);
162     createdEditor.setCaretEnabled(false);
163     showPreviewFrame.setComponent(labeledEditor);
164     showPreviewFrame.setSize(new Dimension(500, 500));
165     if (!ApplicationManager.getApplication().isUnitTestMode()) {
166       showPreviewFrame.show();
167     }
168   }
169 }