diff --git a/store/file/file.go b/store/file/file.go index 3f7df36..5ddb722 100644 --- a/store/file/file.go +++ b/store/file/file.go @@ -19,6 +19,7 @@ type FileProboCollectorStore struct { Dir string memoryStore *memory.MemoryProboCollectorStore + paths map[string]string } func NewFileProboCollectorStore(dirname string) (*FileProboCollectorStore, error) { @@ -29,6 +30,8 @@ func NewFileProboCollectorStore(dirname string) (*FileProboCollectorStore, error return nil, err } + s.paths = make(map[string]string) + markdownFiles := make([]fs.FileInfo, 0) for _, file := range files { @@ -71,13 +74,23 @@ func NewFileProboCollectorStore(dirname string) (*FileProboCollectorStore, error func (s *FileProboCollectorStore) ReadAllQuizzes() ([]*models.Quiz, error) { return s.memoryStore.ReadAllQuizzes() } + func (s *FileProboCollectorStore) CreateQuiz(r *client.CreateUpdateQuizRequest) (*models.Quiz, error) { quiz, err := s.memoryStore.CreateQuiz(r) if err != nil { return nil, err } - err = s.writeMarkdownFile(quiz) + err = s.createOrUpdateMarkdownFile(quiz) + if err != nil { + return nil, err + } + + return quiz, nil +} + +func (s *FileProboCollectorStore) UpdateQuiz(r *client.CreateUpdateQuizRequest, id string) (*models.Quiz, error) { + quiz, err := s.memoryStore.UpdateQuiz(r, id) if err != nil { return nil, err } @@ -146,17 +159,42 @@ func QuizFromMarkdown(markdown string) (*client.Quiz, error) { return quiz, nil } -func (s *FileProboCollectorStore) writeMarkdownFile(quiz *models.Quiz) error { +func (s *FileProboCollectorStore) createOrUpdateMarkdownFile(quiz *models.Quiz) error { + var filename string + markdown, err := MarkdownFromQuiz(quiz) if err != nil { return err } + fn := s.paths[quiz.ID] + if fn == "" { + filename = filepath.Join(s.Dir, quiz.Hash+".md") + } + file, err := os.Create(filename) + if err != nil { + return err + } + defer file.Close() - filename := filepath.Join(s.Dir, quiz.Hash+".md") - err = ioutil.WriteFile(filename, []byte(markdown), 0644) + _, err = file.Write([]byte(markdown)) if err != nil { return err } return nil } + +// func (s *FileProboCollectorStore) writeMarkdownFile(quiz *models.Quiz) error { +// markdown, err := MarkdownFromQuiz(quiz) +// if err != nil { +// return err +// } + +// filename := filepath.Join(s.Dir, quiz.Hash+".md") +// err = ioutil.WriteFile(filename, []byte(markdown), 0644) +// if err != nil { +// return err +// } + +// return nil +// } diff --git a/store/file/file_test.go b/store/file/file_test.go index 9fd202d..0f890c0 100644 --- a/store/file/file_test.go +++ b/store/file/file_test.go @@ -109,6 +109,48 @@ func (t *testSuite) TestCreateQuiz() { } } +func (t *testSuite) TestUpdateQuiz() { + dirname := "./test/quizzes" + store, err := NewFileProboCollectorStore(dirname) + t.True(err == nil, fmt.Sprintf("A file store should be initialized without problems but an error occurred: %v", err)) + + if !t.Failed() { + quiz, err := createQuizOnDisk(store, &client.CreateUpdateQuizRequest{ + Quiz: &client.Quiz{ + Question: &client.Question{Text: "Newly created question text."}, + Answers: []*client.Answer{ + {Text: "Answer 1", Correct: true}, + {Text: "Answer 2", Correct: false}, + {Text: "Answer 3", Correct: false}, + {Text: "Answer 4", Correct: false}, + }, + }, + }) + + t.Nil(err, "The quiz to update should be created without issue") + + if !t.Failed() { + _, err = store.UpdateQuiz( + &client.CreateUpdateQuizRequest{ + Quiz: &client.Quiz{ + Question: &client.Question{Text: "Newly created question text."}, + Answers: []*client.Answer{ + {Text: "Answer 1", Correct: true}, + {Text: "Answer 2", Correct: false}, + {Text: "Answer 3", Correct: false}, + {Text: "Answer 4", Correct: false}, + }, + }, + }, quiz.ID) + } + } +} + +func createQuizOnDisk(store *FileProboCollectorStore, req *client.CreateUpdateQuizRequest) (*models.Quiz, error) { + return store.CreateQuiz(req) + +} + func testsAreEqual(got, want []*models.Quiz) bool { return reflect.DeepEqual(got, want) } diff --git a/store/file/test/quizzes/94ed4e9cdf8e0a75a2c5ce925cb791ebc5977ce1801e12059f58ce4d66c0c7f6.md b/store/file/test/quizzes/94ed4e9cdf8e0a75a2c5ce925cb791ebc5977ce1801e12059f58ce4d66c0c7f6.md new file mode 100644 index 0000000..356d066 --- /dev/null +++ b/store/file/test/quizzes/94ed4e9cdf8e0a75a2c5ce925cb791ebc5977ce1801e12059f58ce4d66c0c7f6.md @@ -0,0 +1,7 @@ +Newly created question text. + +* Answer 1 +* Answer 1 +* Answer 2 +* Answer 3 +* Answer 4