remove usages of redundant CCProjectService
[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.diagnostic.Logger;
25 import com.intellij.openapi.diff.impl.util.LabeledEditor;
26 import com.intellij.openapi.editor.Document;
27 import com.intellij.openapi.editor.EditorFactory;
28 import com.intellij.openapi.editor.ex.EditorEx;
29 import com.intellij.openapi.fileEditor.FileDocumentManager;
30 import com.intellij.openapi.module.Module;
31 import com.intellij.openapi.project.DumbAwareAction;
32 import com.intellij.openapi.project.Project;
33 import com.intellij.openapi.ui.FrameWrapper;
34 import com.intellij.openapi.ui.Messages;
35 import com.intellij.openapi.util.Disposer;
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.learning.StudyTaskManager;
41 import com.jetbrains.edu.learning.StudyUtils;
42 import com.jetbrains.edu.learning.core.EduAnswerPlaceholderPainter;
43 import com.jetbrains.edu.learning.core.EduUtils;
44 import com.jetbrains.edu.learning.courseFormat.AnswerPlaceholder;
45 import com.jetbrains.edu.learning.courseFormat.Course;
46 import com.jetbrains.edu.learning.courseFormat.TaskFile;
47 import com.jetbrains.edu.coursecreator.CCUtils;
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   private static final Logger LOG = Logger.getInstance(CCShowPreview.class.getName());
58
59   public CCShowPreview() {
60     super("Show Preview", "Show Preview", null);
61   }
62
63   @Override
64   public void update(@NotNull AnActionEvent e) {
65     Presentation presentation = e.getPresentation();
66     presentation.setEnabledAndVisible(false);
67     Project project = e.getProject();
68     if (project == null) {
69       return;
70     }
71     if (!CCUtils.isCourseCreator(project)) {
72       return;
73     }
74     final PsiFile file = CommonDataKeys.PSI_FILE.getData(e.getDataContext());
75     if (file != null && StudyUtils.getTaskFile(project, file.getVirtualFile()) != null) {
76       presentation.setEnabledAndVisible(true);
77     }
78   }
79
80   @Override
81   public void actionPerformed(@NotNull AnActionEvent e) {
82     final Project project = e.getProject();
83     Module module = LangDataKeys.MODULE.getData(e.getDataContext());
84     if (project == null || module == null) {
85       return;
86     }
87     final PsiFile file = CommonDataKeys.PSI_FILE.getData(e.getDataContext());
88     if (file == null) {
89       return;
90     }
91     Course course = StudyTaskManager.getInstance(project).getCourse();
92     if (course == null) {
93       return;
94     }
95     VirtualFile virtualFile = file.getVirtualFile();
96     TaskFile taskFile = StudyUtils.getTaskFile(project, virtualFile);
97     if (taskFile == null) {
98       return;
99     }
100     final PsiDirectory taskDir = file.getContainingDirectory();
101     if (taskDir == null) {
102       return;
103     }
104     PsiDirectory lessonDir = taskDir.getParentDirectory();
105     if (lessonDir == null) {
106       return;
107     }
108
109
110     if (taskFile.getAnswerPlaceholders().isEmpty()) {
111       Messages.showInfoMessage("Preview is available for task files with answer placeholders only", "No Preview for This File");
112       return;
113     }
114     final TaskFile taskFileCopy = new TaskFile();
115     TaskFile.copy(taskFile, taskFileCopy);
116
117
118     VirtualFile generatedFilesFolder = CCUtils.getGeneratedFilesFolder(project, module);
119
120     if (generatedFilesFolder == null) {
121       return;
122     }
123
124     ApplicationManager.getApplication().runWriteAction(new Runnable() {
125       @Override
126       public void run() {
127         EduUtils.createStudentFileFromAnswer(project, generatedFilesFolder, taskDir.getVirtualFile(), virtualFile.getName(), taskFileCopy);
128       }
129     });
130
131     VirtualFile userFile = generatedFilesFolder.findChild(virtualFile.getName());
132     if (userFile == null) {
133       LOG.info("Generated file " + virtualFile.getName() + "was not found");
134       return;
135     }
136     final FrameWrapper showPreviewFrame = new FrameWrapper(project);
137     showPreviewFrame.setTitle(virtualFile.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 : taskFileCopy.getAnswerPlaceholders()) {
151       EduAnswerPlaceholderPainter.drawAnswerPlaceholder(createdEditor, answerPlaceholder, true, 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     showPreviewFrame.show();
166   }
167 }