mirror of https://github.com/cloudreve/Cloudreve
69 lines
1.9 KiB
Go
69 lines
1.9 KiB
Go
package wopi
|
|
|
|
import (
|
|
"encoding/xml"
|
|
"fmt"
|
|
"github.com/cloudreve/Cloudreve/v4/inventory/types"
|
|
"github.com/gofrs/uuid"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
type ActonType string
|
|
|
|
var (
|
|
ActionPreview = ActonType("embedview")
|
|
ActionPreviewFallback = ActonType("view")
|
|
ActionEdit = ActonType("edit")
|
|
)
|
|
|
|
func DiscoveryXmlToViewerGroup(xmlStr string) (*types.ViewerGroup, error) {
|
|
var discovery WopiDiscovery
|
|
if err := xml.Unmarshal([]byte(xmlStr), &discovery); err != nil {
|
|
return nil, fmt.Errorf("failed to parse WOPI discovery XML: %w", err)
|
|
}
|
|
|
|
group := &types.ViewerGroup{
|
|
Viewers: make([]types.Viewer, 0, len(discovery.NetZone.App)),
|
|
}
|
|
|
|
for _, app := range discovery.NetZone.App {
|
|
viewer := types.Viewer{
|
|
ID: uuid.Must(uuid.NewV4()).String(),
|
|
DisplayName: app.Name,
|
|
Type: types.ViewerTypeWopi,
|
|
Icon: app.FavIconUrl,
|
|
WopiActions: make(map[string]map[types.ViewerAction]string),
|
|
}
|
|
|
|
for _, action := range app.Action {
|
|
if action.Ext == "" {
|
|
continue
|
|
}
|
|
|
|
if _, ok := viewer.WopiActions[action.Ext]; !ok {
|
|
viewer.WopiActions[action.Ext] = make(map[types.ViewerAction]string)
|
|
}
|
|
|
|
if action.Name == string(ActionPreview) {
|
|
viewer.WopiActions[action.Ext][types.ViewerActionView] = action.Urlsrc
|
|
} else if action.Name == string(ActionPreviewFallback) {
|
|
viewer.WopiActions[action.Ext][types.ViewerActionView] = action.Urlsrc
|
|
} else if action.Name == string(ActionEdit) {
|
|
viewer.WopiActions[action.Ext][types.ViewerActionEdit] = action.Urlsrc
|
|
} else if len(viewer.WopiActions[action.Ext]) == 0 {
|
|
delete(viewer.WopiActions, action.Ext)
|
|
}
|
|
}
|
|
|
|
viewer.Exts = lo.MapToSlice(viewer.WopiActions, func(key string, value map[types.ViewerAction]string) string {
|
|
return key
|
|
})
|
|
|
|
if len(viewer.WopiActions) > 0 {
|
|
group.Viewers = append(group.Viewers, viewer)
|
|
}
|
|
}
|
|
|
|
return group, nil
|
|
}
|