diff --git a/db/db_queries.go b/db/db_queries.go index 43e648e..9ca0498 100644 --- a/db/db_queries.go +++ b/db/db_queries.go @@ -23,7 +23,8 @@ const GetBookDataById = ` const GetNotesHighlightsById = ` select a.ZAEANNOTATION.ZANNOTATIONSELECTEDTEXT, - a.ZAEANNOTATION.ZANNOTATIONNOTE + a.ZAEANNOTATION.ZANNOTATIONNOTE, + a.ZAEANNOTATION.ZANNOTATIONSTYLE from a.ZAEANNOTATION where diff --git a/db/structs.go b/db/structs.go index fd223cb..397ff62 100644 --- a/db/structs.go +++ b/db/structs.go @@ -12,6 +12,7 @@ type SingleBook struct { type SingleHighlightNote struct { HightLight string Note sql.NullString + Style int } type SingleBookInList struct { diff --git a/main.go b/main.go index 8094dd3..d40cbb0 100644 --- a/main.go +++ b/main.go @@ -178,12 +178,12 @@ func exportNotesAndHighlights(cCtx *cli.Context) error { var singleHightLightNote dbThings.SingleHighlightNote for rows.Next() { - err := rows.Scan(&singleHightLightNote.HightLight, &singleHightLightNote.Note) + err := rows.Scan(&singleHightLightNote.HightLight, &singleHightLightNote.Note, &singleHightLightNote.Style) if err != nil { log.Fatal(err) } - fmt.Println(fmt.Sprintf("> %s", strings.Replace(singleHightLightNote.HightLight, "\n", "", -1))) + fmt.Println(fmt.Sprintf("> %s",styleToColor(singleHightLightNote.Style),strings.Replace(singleHightLightNote.HightLight, "\n", "", -1))) if singleHightLightNote.Note.Valid { fmt.Println(fmt.Sprintf("\n%s", strings.Replace(singleHightLightNote.Note.String, "\n", "", -1))) @@ -195,3 +195,25 @@ func exportNotesAndHighlights(cCtx *cli.Context) error { return nil } + +func styleToColor(style int) string { + switch style { + case 1: + // green + return "#a8e196" + case 2: + // blue + return "#a5c3ff" + case 3: + // yellow + return "#fde15c" + case 4: + // pink + return "#ffaabf" + case 5: + // purple + return "#cdbbfb" + default: + return "" + } +} \ No newline at end of file