diff options
-rw-r--r-- | model/postContext.go (renamed from model/replyContext.go) | 5 | ||||
-rw-r--r-- | model/session.go | 1 | ||||
-rw-r--r-- | model/settings.go | 1 | ||||
-rw-r--r-- | renderer/model.go | 46 | ||||
-rw-r--r-- | service/auth.go | 25 | ||||
-rw-r--r-- | service/logging.go | 8 | ||||
-rw-r--r-- | service/service.go | 38 | ||||
-rw-r--r-- | service/transport.go | 3 | ||||
-rw-r--r-- | static/main.css | 11 | ||||
-rw-r--r-- | templates/postform.tmpl | 20 | ||||
-rw-r--r-- | templates/status.tmpl | 32 | ||||
-rw-r--r-- | templates/thread.tmpl | 4 | ||||
-rw-r--r-- | templates/timeline.tmpl | 2 |
13 files changed, 126 insertions, 70 deletions
diff --git a/model/replyContext.go b/model/postContext.go index b17aacb..90e5771 100644 --- a/model/replyContext.go +++ b/model/postContext.go @@ -1,5 +1,10 @@ package model +type PostContext struct { + DefaultVisibility string + ReplyContext *ReplyContext +} + type ReplyContext struct { InReplyToID string InReplyToName string diff --git a/model/session.go b/model/session.go index 42c0aff..fce6173 100644 --- a/model/session.go +++ b/model/session.go @@ -17,7 +17,6 @@ type Session struct { type SessionRepository interface { Add(session Session) (err error) - Update(sessionID string, accessToken string) (err error) Get(sessionID string) (session Session, err error) } diff --git a/model/settings.go b/model/settings.go index ad7ec0f..7afe936 100644 --- a/model/settings.go +++ b/model/settings.go @@ -1,4 +1,5 @@ package model type Settings struct { + DefaultVisibility string `json:"default_visibility"` } diff --git a/renderer/model.go b/renderer/model.go index ce81e78..9380b7f 100644 --- a/renderer/model.go +++ b/renderer/model.go @@ -16,39 +16,41 @@ func NewNavbarTemplateData(notificationCount int) *NavbarTemplateData { } type TimelinePageTemplateData struct { - Statuses []*mastodon.Status - HasNext bool - NextLink string - HasPrev bool - PrevLink string - NavbarData *NavbarTemplateData + Statuses []*mastodon.Status + HasNext bool + NextLink string + HasPrev bool + PrevLink string + PostContext model.PostContext + NavbarData *NavbarTemplateData } func NewTimelinePageTemplateData(statuses []*mastodon.Status, hasNext bool, nextLink string, hasPrev bool, - prevLink string, navbarData *NavbarTemplateData) *TimelinePageTemplateData { + prevLink string, postContext model.PostContext, navbarData *NavbarTemplateData) *TimelinePageTemplateData { return &TimelinePageTemplateData{ - Statuses: statuses, - HasNext: hasNext, - NextLink: nextLink, - HasPrev: hasPrev, - PrevLink: prevLink, - NavbarData: navbarData, + Statuses: statuses, + HasNext: hasNext, + NextLink: nextLink, + HasPrev: hasPrev, + PrevLink: prevLink, + PostContext: postContext, + NavbarData: navbarData, } } type ThreadPageTemplateData struct { - Statuses []*mastodon.Status - ReplyContext *model.ReplyContext - ReplyMap map[string][]mastodon.ReplyInfo - NavbarData *NavbarTemplateData + Statuses []*mastodon.Status + PostContext model.PostContext + ReplyMap map[string][]mastodon.ReplyInfo + NavbarData *NavbarTemplateData } -func NewThreadPageTemplateData(statuses []*mastodon.Status, replyContext *model.ReplyContext, replyMap map[string][]mastodon.ReplyInfo, navbarData *NavbarTemplateData) *ThreadPageTemplateData { +func NewThreadPageTemplateData(statuses []*mastodon.Status, postContext model.PostContext, replyMap map[string][]mastodon.ReplyInfo, navbarData *NavbarTemplateData) *ThreadPageTemplateData { return &ThreadPageTemplateData{ - Statuses: statuses, - ReplyContext: replyContext, - ReplyMap: replyMap, - NavbarData: navbarData, + Statuses: statuses, + PostContext: postContext, + ReplyMap: replyMap, + NavbarData: navbarData, } } diff --git a/service/auth.go b/service/auth.go index 0209273..13e9c50 100644 --- a/service/auth.go +++ b/service/auth.go @@ -23,17 +23,9 @@ func NewAuthService(sessionRepo model.SessionRepository, appRepo model.AppReposi return &authService{sessionRepo, appRepo, s} } -func getSessionID(ctx context.Context) (sessionID string, err error) { +func (s *authService) getClient(ctx context.Context) (c *model.Client, err error) { sessionID, ok := ctx.Value("session_id").(string) if !ok || len(sessionID) < 1 { - return "", ErrInvalidSession - } - return sessionID, nil -} - -func (s *authService) getClient(ctx context.Context) (c *model.Client, err error) { - sessionID, err := getSessionID(ctx) - if err != nil { return nil, ErrInvalidSession } session, err := s.sessionRepo.Get(sessionID) @@ -50,7 +42,7 @@ func (s *authService) getClient(ctx context.Context) (c *model.Client, err error ClientSecret: client.ClientSecret, AccessToken: session.AccessToken, }) - c = &model.Client{Client: mc} + c = &model.Client{Client: mc, Session: session} return c, nil } @@ -61,21 +53,18 @@ func (s *authService) GetAuthUrl(ctx context.Context, instance string) ( func (s *authService) GetUserToken(ctx context.Context, sessionID string, c *model.Client, code string) (token string, err error) { - sessionID, err = getSessionID(ctx) - if err != nil { - return - } c, err = s.getClient(ctx) if err != nil { return } - token, err = s.Service.GetUserToken(ctx, sessionID, c, code) + token, err = s.Service.GetUserToken(ctx, c.Session.ID, c, code) if err != nil { return } - err = s.sessionRepo.Update(sessionID, token) + c.Session.AccessToken = token + err = s.sessionRepo.Add(c.Session) if err != nil { return } @@ -168,12 +157,12 @@ func (s *authService) UnRetweet(ctx context.Context, client io.Writer, c *model. return s.Service.UnRetweet(ctx, client, c, id) } -func (s *authService) PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, files []*multipart.FileHeader) (id string, err error) { +func (s *authService) PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, visibility string, files []*multipart.FileHeader) (id string, err error) { c, err = s.getClient(ctx) if err != nil { return } - return s.Service.PostTweet(ctx, client, c, content, replyToID, files) + return s.Service.PostTweet(ctx, client, c, content, replyToID, visibility, files) } func (s *authService) Follow(ctx context.Context, client io.Writer, c *model.Client, id string) (err error) { diff --git a/service/logging.go b/service/logging.go index f34bef5..ceba716 100644 --- a/service/logging.go +++ b/service/logging.go @@ -133,12 +133,12 @@ func (s *loggingService) UnRetweet(ctx context.Context, client io.Writer, c *mod return s.Service.UnRetweet(ctx, client, c, id) } -func (s *loggingService) PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, files []*multipart.FileHeader) (id string, err error) { +func (s *loggingService) PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, visibility string, files []*multipart.FileHeader) (id string, err error) { defer func(begin time.Time) { - s.logger.Printf("method=%v, content=%v, reply_to_id=%v, took=%v, err=%v\n", - "PostTweet", content, replyToID, time.Since(begin), err) + s.logger.Printf("method=%v, content=%v, reply_to_id=%v, visibility=%v, took=%v, err=%v\n", + "PostTweet", content, replyToID, visibility, time.Since(begin), err) }(time.Now()) - return s.Service.PostTweet(ctx, client, c, content, replyToID, files) + return s.Service.PostTweet(ctx, client, c, content, replyToID, visibility, files) } func (s *loggingService) Follow(ctx context.Context, client io.Writer, c *model.Client, id string) (err error) { diff --git a/service/service.go b/service/service.go index d844a6f..07a026a 100644 --- a/service/service.go +++ b/service/service.go @@ -38,7 +38,7 @@ type Service interface { UnLike(ctx context.Context, client io.Writer, c *model.Client, id string) (err error) Retweet(ctx context.Context, client io.Writer, c *model.Client, id string) (err error) UnRetweet(ctx context.Context, client io.Writer, c *model.Client, id string) (err error) - PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, files []*multipart.FileHeader) (id string, err error) + PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, visibility string, files []*multipart.FileHeader) (id string, err error) Follow(ctx context.Context, client io.Writer, c *model.Client, id string) (err error) UnFollow(ctx context.Context, client io.Writer, c *model.Client, id string) (err error) } @@ -251,12 +251,16 @@ func (svc *service) ServeTimelinePage(ctx context.Context, client io.Writer, nextLink = "/timeline?max_id=" + pg.MaxID } + postContext := model.PostContext{ + DefaultVisibility: c.Session.Settings.DefaultVisibility, + } + navbarData, err := svc.getNavbarTemplateData(ctx, client, c) if err != nil { return } - data := renderer.NewTimelinePageTemplateData(statuses, hasNext, nextLink, hasPrev, prevLink, navbarData) + data := renderer.NewTimelinePageTemplateData(statuses, hasNext, nextLink, hasPrev, prevLink, postContext, navbarData) err = svc.renderer.RenderTimelinePage(ctx, client, data) if err != nil { return @@ -276,7 +280,7 @@ func (svc *service) ServeThreadPage(ctx context.Context, client io.Writer, c *mo return } - var replyContext *model.ReplyContext + var postContext model.PostContext if reply { var content string if u.ID != status.Account.ID { @@ -287,10 +291,19 @@ func (svc *service) ServeThreadPage(ctx context.Context, client io.Writer, c *mo content += "@" + status.Mentions[i].Acct + " " } } - replyContext = &model.ReplyContext{ - InReplyToID: id, - InReplyToName: status.Account.Acct, - ReplyContent: content, + + s, err := c.GetStatus(ctx, id) + if err != nil { + return err + } + + postContext = model.PostContext{ + DefaultVisibility: s.Visibility, + ReplyContext: &model.ReplyContext{ + InReplyToID: id, + InReplyToName: status.Account.Acct, + ReplyContent: content, + }, } } @@ -314,7 +327,7 @@ func (svc *service) ServeThreadPage(ctx context.Context, client io.Writer, c *mo return } - data := renderer.NewThreadPageTemplateData(statuses, replyContext, replyMap, navbarData) + data := renderer.NewThreadPageTemplateData(statuses, postContext, replyMap, navbarData) err = svc.renderer.RenderThreadPage(ctx, client, data) if err != nil { return @@ -469,7 +482,7 @@ func (svc *service) UnRetweet(ctx context.Context, client io.Writer, c *model.Cl return } -func (svc *service) PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, files []*multipart.FileHeader) (id string, err error) { +func (svc *service) PostTweet(ctx context.Context, client io.Writer, c *model.Client, content string, replyToID string, visibility string, files []*multipart.FileHeader) (id string, err error) { var mediaIds []string for _, f := range files { a, err := c.UploadMediaFromMultipartFileHeader(ctx, f) @@ -479,10 +492,17 @@ func (svc *service) PostTweet(ctx context.Context, client io.Writer, c *model.Cl mediaIds = append(mediaIds, a.ID) } + // save visibility if it's a non-reply post + if len(replyToID) < 1 && visibility != c.Session.Settings.DefaultVisibility { + c.Session.Settings.DefaultVisibility = visibility + svc.sessionRepo.Add(c.Session) + } + tweet := &mastodon.Toot{ Status: content, InReplyToID: replyToID, MediaIDs: mediaIds, + Visibility: visibility, } s, err := c.PostStatus(ctx, tweet) diff --git a/service/transport.go b/service/transport.go index 2b75662..4e852f2 100644 --- a/service/transport.go +++ b/service/transport.go @@ -155,9 +155,10 @@ func NewHandler(s Service, staticDir string) http.Handler { content := getMultipartFormValue(req.MultipartForm, "content") replyToID := getMultipartFormValue(req.MultipartForm, "reply_to_id") + visibility := getMultipartFormValue(req.MultipartForm, "visibility") files := req.MultipartForm.File["attachments"] - id, err := s.PostTweet(ctx, w, nil, content, replyToID, files) + id, err := s.PostTweet(ctx, w, nil, content, replyToID, visibility, files) if err != nil { s.ServeErrorPage(ctx, w, err) return diff --git a/static/main.css b/static/main.css index 4f8d156..e37cef4 100644 --- a/static/main.css +++ b/static/main.css @@ -246,3 +246,14 @@ font-family: inherit; font-size: inherit; } + +.status-visibility { + margin: 0 4px; + display: inline-block; + color: #444444; +} + +.status-visibility span { + font-size: 9pt; + vertical-align: bottom; +} diff --git a/templates/postform.tmpl b/templates/postform.tmpl index 3f6eeaa..137232f 100644 --- a/templates/postform.tmpl +++ b/templates/postform.tmpl @@ -1,15 +1,25 @@ <form class="post-form" action="/post" method="POST" enctype="multipart/form-data"> - {{if .}} - <input type="hidden" name="reply_to_id" value="{{.InReplyToID}}" /> - <label for="post-content"> Reply to {{.InReplyToName}} </label> + {{if .ReplyContext}} + <input type="hidden" name="reply_to_id" value="{{.ReplyContext.InReplyToID}}" /> + <label for="post-content"> Reply to {{.ReplyContext.InReplyToName}} </label> {{else}} <label for="post-content"> New post </label> {{end}} <div class="post-form-content-container"> - <textarea id="post-content" name="content" class="post-content" cols="50" rows="5">{{if .}}{{.ReplyContent}}{{end}}</textarea> + <textarea id="post-content" name="content" class="post-content" cols="50" rows="5">{{if .ReplyContext}}{{.ReplyContext.ReplyContent}}{{end}}</textarea> </div> <div> - Attachments <input id="post-file-picker" type="file" name="attachments" multiple> + <label for ="post-visilibity"> Visibility </label> + <select id="post-visilibity" type="file" name="visibility"> + <option value="public" {{if eq .DefaultVisibility "public"}}selected{{end}}>Public</option> + <option value="unlisted" {{if eq .DefaultVisibility "unlisted"}}selected{{end}}>Unlisted</option> + <option value="private" {{if eq .DefaultVisibility "private"}}selected{{end}}>Private</option> + <option value="direct" {{if eq .DefaultVisibility "direct"}}selected{{end}}>Direct</option> + </select> + </div> + <div> + <label for ="post-file-picker"> Attachments </label> + <input id="post-file-picker" type="file" name="attachments" multiple> </div> <button type="submit"> Post </button> </form> diff --git a/templates/status.tmpl b/templates/status.tmpl index 0e6f0c0..a89fc78 100644 --- a/templates/status.tmpl +++ b/templates/status.tmpl @@ -26,6 +26,17 @@ <a href="/user/{{.Account.ID}}" > <span class="status-uname"> {{.Account.Acct}} </span> </a> + <a class="status-visibility" href="{{.URL}}" target="_blank"> + {{if eq .Visibility "public"}} + <span class="icon dripicons-web" title="Public"></span> + {{else if eq .Visibility "unlisted"}} + <span class="icon dripicons-lock-open" title="Unlisted"></span> + {{else if eq .Visibility "private"}} + <span class="icon dripicons-lock" title="Private"></span> + {{else if eq .Visibility "direct"}} + <span class="icon dripicons-mail" title="Direct"></span> + {{end}} + </a> </div> {{end}} <div class="status-reply-container"> @@ -68,16 +79,23 @@ <span class="icon dripicons-reply"></span> <span> {{DisplayInteractionCount .RepliesCount}} </span> </a> - {{if .Reblogged}} - <a class="status-retweet" href="/unretweet/{{.ID}}" title="undo repost"> - <span class="icon dripicons-retweet retweeted"></span> - <span> {{DisplayInteractionCount .ReblogsCount}} </span> - </a> - {{else}} - <a class="status-retweet" href="/retweet/{{.ID}}" title="repost"> + {{if or (eq .Visibility "private") (eq .Visibility "direct")}} + <a class="status-retweet" title="this status cannot be retweeted"> <span class="icon dripicons-retweet"></span> <span> {{DisplayInteractionCount .ReblogsCount}} </span> </a> + {{else}} + {{if .Reblogged}} + <a class="status-retweet" href="/unretweet/{{.ID}}" title="undo retweet"> + <span class="icon dripicons-retweet retweeted"></span> + <span> {{DisplayInteractionCount .ReblogsCount}} </span> + </a> + {{else}} + <a class="status-retweet" href="/retweet/{{.ID}}" title="retweet"> + <span class="icon dripicons-retweet"></span> + <span> {{DisplayInteractionCount .ReblogsCount}} </span> + </a> + {{end}} {{end}} {{if .Favourited}} <a class="status-like" href="/unlike/{{.ID}}" title="unlike"> diff --git a/templates/thread.tmpl b/templates/thread.tmpl index 9d8f650..afb307b 100644 --- a/templates/thread.tmpl +++ b/templates/thread.tmpl @@ -5,8 +5,8 @@ {{range .Statuses}} {{template "status.tmpl" .}} -{{if $.ReplyContext}}{{if eq .ID $.ReplyContext.InReplyToID}} -{{template "postform.tmpl" $.ReplyContext}} +{{if $.PostContext.ReplyContext}}{{if eq .ID $.PostContext.ReplyContext.InReplyToID}} +{{template "postform.tmpl" $.PostContext}} {{end}}{{end}} {{end}} diff --git a/templates/timeline.tmpl b/templates/timeline.tmpl index 09717c1..6280e65 100644 --- a/templates/timeline.tmpl +++ b/templates/timeline.tmpl @@ -3,7 +3,7 @@ <div class="page-title"> Timeline </div> -{{template "postform.tmpl" }} +{{template "postform.tmpl" .PostContext}} {{range .Statuses}} {{template "status.tmpl" .}} |